added chat history
This commit is contained in:
@@ -54,6 +54,9 @@ class Project(Base):
|
||||
designator_mappings: Mapped[list[DesignatorMapping]] = relationship(
|
||||
back_populates="project", cascade="all, delete-orphan"
|
||||
)
|
||||
llm_chats: Mapped[list[LlmChat]] = relationship(
|
||||
back_populates="project", cascade="all, delete-orphan"
|
||||
)
|
||||
llm_messages: Mapped[list[LlmMessage]] = relationship(
|
||||
back_populates="project", cascade="all, delete-orphan"
|
||||
)
|
||||
@@ -304,11 +307,32 @@ class SimpleListRow(Base):
|
||||
is_empty: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||
|
||||
|
||||
class LlmChat(Base):
|
||||
__tablename__ = "llm_chats"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
project_id: Mapped[int] = mapped_column(ForeignKey("projects.id", ondelete="CASCADE"), index=True)
|
||||
title: Mapped[str] = mapped_column(String(255), default="Новый чат")
|
||||
table_type: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
|
||||
)
|
||||
|
||||
project: Mapped[Project] = relationship(back_populates="llm_chats")
|
||||
messages: Mapped[list[LlmMessage]] = relationship(
|
||||
back_populates="chat", cascade="all, delete-orphan", order_by="LlmMessage.id"
|
||||
)
|
||||
|
||||
|
||||
class LlmMessage(Base):
|
||||
__tablename__ = "llm_messages"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
project_id: Mapped[int] = mapped_column(ForeignKey("projects.id", ondelete="CASCADE"), index=True)
|
||||
chat_id: Mapped[Optional[int]] = mapped_column(
|
||||
ForeignKey("llm_chats.id", ondelete="CASCADE"), index=True, nullable=True
|
||||
)
|
||||
role: Mapped[str] = mapped_column(String(32), nullable=False)
|
||||
content: Mapped[str] = mapped_column(Text, default="")
|
||||
table_type: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
||||
@@ -316,3 +340,4 @@ class LlmMessage(Base):
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
||||
|
||||
project: Mapped[Project] = relationship(back_populates="llm_messages")
|
||||
chat: Mapped[Optional[LlmChat]] = relationship(back_populates="messages")
|
||||
|
||||
Reference in New Issue
Block a user