github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-01 08:11:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent):fix agent overrid bug (#2235)

Browse Source 
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
明天 2024-12-21 12:47:33 +08:00 committed by GitHub
parent 1bdf695a1f
commit 78c0368463
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/serve/agent/agents/expand/app_start_assisant_agent.py
View File

@ -55,6 +55,7 @@ class StartAppAssistantAgent(ConversableAgent):
received_message: Optional[AgentMessage], received_message: Optional[AgentMessage],
sender: Agent, sender: Agent,
rely_messages: Optional[List[AgentMessage]] = None, rely_messages: Optional[List[AgentMessage]] = None,
**kwargs,
) -> Dict[str, Any]: ) -> Dict[str, Any]:
return { return {
"user_input": received_message.content, "user_input": received_message.content,
@ -67,9 +68,10 @@ class StartAppAssistantAgent(ConversableAgent):
received_message: AgentMessage, received_message: AgentMessage,
sender: Agent, sender: Agent,
rely_messages: Optional[List[AgentMessage]] = None, rely_messages: Optional[List[AgentMessage]] = None,
historical_dialogues: Optional[List[AgentMessage]] = None,
context: Optional[Dict[str, Any]] = None, context: Optional[Dict[str, Any]] = None,
is_retry_chat: Optional[bool] = False, is_retry_chat: bool = False,
) -> tuple[List[AgentMessage], Optional[Dict]]: ) -> Tuple[List[AgentMessage], Optional[Dict]]:
if rely_messages and len(rely_messages) > 0: if rely_messages and len(rely_messages) > 0:
return rely_messages[-1:], None return rely_messages[-1:], None
else: else: