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

feat: (0.6)New UI (#1855)

Browse Source 
Co-authored-by: 夏姜 <wenfengjiang.jwf@digital-engine.com>
Co-authored-by: aries_ckt <916701291@qq.com>
Co-authored-by: wb-lh513319 <wb-lh513319@alibaba-inc.com>
Co-authored-by: csunny <cfqsunny@163.com>
This commit is contained in:
明天
2024-08-21 17:37:45 +08:00
committed by GitHub
parent 3fc82693ba
commit b124ecc10b
824 changed files with 93371 additions and 2515 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
dbgpt/serve/agent/agents/db_gpts_memory.py
View File

@@ -1,9 +1,9 @@
from typing import List, Optional
from dbgpt.agent.core.memory.gpts.base import GptsPlan
from dbgpt.agent.core.memory.gpts.gpts_memory import (
from dbgpt.agent.core.memory.gpts import (
GptsMessage,
GptsMessageMemory,
GptsPlan,
GptsPlansMemory,
)
@@ -56,9 +56,9 @@ class MetaDbGptsPlansMemory(GptsPlansMemory):
task_num: int,
state: str,
retry_times: int,
agent: Optional[str] = None,
model: Optional[str] = None,
result: Optional[str] = None,
agent: str = None,
model: str = None,
result: str = None,
):
self.gpts_plan.update_task(
conv_id=conv_id,
@@ -95,7 +95,7 @@ class MetaDbGptsMessageMemory(GptsMessageMemory):
agent1: str,
agent2: str,
current_goal: Optional[str] = None,
) -> List[GptsMessage]:
) -> Optional[List[GptsMessage]]:
db_results = self.gpts_message.get_between_agents(
conv_id, agent1, agent2, current_goal
)
@@ -105,7 +105,7 @@ class MetaDbGptsMessageMemory(GptsMessageMemory):
results.append(GptsMessage.from_dict(item.__dict__))
return results
def get_by_conv_id(self, conv_id: str) -> List[GptsMessage]:
def get_by_conv_id(self, conv_id: str) -> Optional[List[GptsMessage]]:
db_results = self.gpts_message.get_by_conv_id(conv_id)
results = []