github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-07 12:00:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Gpts app v0.4 (#1170)

Browse Source
This commit is contained in:
明天
2024-02-20 14:28:58 +08:00
committed by GitHub
parent 63ab612e75
commit c78bd22fda
18 changed files with 65 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/serve/agent/db/gpts_messages_db.py
View File

@@ -39,7 +39,7 @@ class GptsMessagesEntity(Model):
content = Column(
Text(length=2**31 - 1), nullable=True, comment="Content of the speech"
)
current_gogal = Column(
current_goal = Column(
Text, nullable=True, comment="The target corresponding to the current message"
)
context = Column(Text, nullable=True, comment="Current conversation context")
@@ -78,7 +78,7 @@ class GptsMessagesDao(BaseDao):
model_name=entity.get("model_name", None),
context=entity.get("context", None),
rounds=entity.get("rounds", None),
current_gogal=entity.get("current_gogal", None),
current_goal=entity.get("current_goal", None),
review_info=entity.get("review_info", None),
action_report=entity.get("action_report", None),
)
@@ -120,7 +120,7 @@ class GptsMessagesDao(BaseDao):
conv_id: str,
agent1: str,
agent2: str,
current_gogal: Optional[str] = None,
current_goal: Optional[str] = None,
) -> Optional[List[GptsMessagesEntity]]:
session = self.get_raw_session()
gpts_messages = session.query(GptsMessagesEntity)
@@ -139,9 +139,9 @@ class GptsMessagesDao(BaseDao):
),
)
)
if current_gogal:
if current_goal:
gpts_messages = gpts_messages.filter(
GptsMessagesEntity.current_gogal == current_gogal
GptsMessagesEntity.current_goal == current_goal
)
result = gpts_messages.order_by(GptsMessagesEntity.rounds).all()
session.close()