mirror of
https://github.com/Pythagora-io/gpt-pilot.git
synced 2026-01-08 12:53:50 -05:00
5b474ccc1f
This is a complete rewrite of the GPT Pilot core, from the ground up, making the agentic architecture front and center, and also fixing some long-standing problems with the database architecture that weren't feasible to solve without breaking compatibility. As the database structure and config file syntax have changed, we have automatic imports for projects and current configs, see the README.md file for details. This also relicenses the project to FSL-1.1-MIT license.
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
from datetime import datetime
|
|
from typing import TYPE_CHECKING, Optional, Union
|
|
from uuid import UUID, uuid4
|
|
|
|
from sqlalchemy import ForeignKey, inspect, select
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from sqlalchemy.sql import func
|
|
|
|
from core.db.models import Base
|
|
|
|
if TYPE_CHECKING:
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from core.db.models import ExecLog, LLMRequest, Project, ProjectState, UserInput
|
|
|
|
|
|
class Branch(Base):
|
|
__tablename__ = "branches"
|
|
|
|
DEFAULT = "main"
|
|
|
|
# ID and parent FKs
|
|
id: Mapped[UUID] = mapped_column(primary_key=True, default=uuid4)
|
|
project_id: Mapped[UUID] = mapped_column(ForeignKey("projects.id", ondelete="CASCADE"))
|
|
|
|
# Attributes
|
|
created_at: Mapped[datetime] = mapped_column(server_default=func.now())
|
|
name: Mapped[str] = mapped_column(default=DEFAULT)
|
|
|
|
# Relationships
|
|
project: Mapped["Project"] = relationship(back_populates="branches", lazy="selectin")
|
|
states: Mapped[list["ProjectState"]] = relationship(back_populates="branch", cascade="all")
|
|
llm_requests: Mapped[list["LLMRequest"]] = relationship(back_populates="branch", cascade="all")
|
|
user_inputs: Mapped[list["UserInput"]] = relationship(back_populates="branch", cascade="all")
|
|
exec_logs: Mapped[list["ExecLog"]] = relationship(back_populates="branch", cascade="all")
|
|
|
|
@staticmethod
|
|
async def get_by_id(session: "AsyncSession", branch_id: Union[str, UUID]) -> Optional["Branch"]:
|
|
"""
|
|
Get a project by ID.
|
|
|
|
:param session: The SQLAlchemy session.
|
|
:param project_id: The branch ID (as str or UUID value).
|
|
:return: The Branch object if found, None otherwise.
|
|
"""
|
|
if not isinstance(branch_id, UUID):
|
|
branch_id = UUID(branch_id)
|
|
|
|
result = await session.execute(select(Branch).where(Branch.id == branch_id))
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_last_state(self) -> Optional["ProjectState"]:
|
|
"""
|
|
Get the last project state of the branch.
|
|
|
|
:return: The last step of the branch, or None if there are no steps.
|
|
"""
|
|
|
|
from core.db.models import ProjectState
|
|
|
|
session = inspect(self).async_session
|
|
if session is None:
|
|
raise ValueError("Branch instance not associated with a DB session.")
|
|
|
|
result = await session.execute(
|
|
select(ProjectState)
|
|
.where(ProjectState.branch_id == self.id)
|
|
.order_by(ProjectState.step_index.desc())
|
|
.limit(1)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_state_at_step(self, step_index: int) -> Optional["ProjectState"]:
|
|
"""
|
|
Get the project state at the given step index for the branch.
|
|
|
|
:return: The indicated step within the branch, or None if there's no such step.
|
|
"""
|
|
|
|
from core.db.models import ProjectState
|
|
|
|
session = inspect(self).async_session
|
|
if session is None:
|
|
raise ValueError("Branch instance not associated with a DB session.")
|
|
|
|
result = await session.execute(
|
|
select(ProjectState).where((ProjectState.branch_id == self.id) & (ProjectState.step_index == step_index))
|
|
)
|
|
return result.scalar_one_or_none()
|