opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-08 02:43:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(api): clean up AssistantPromptMessage typing in CotChatAgentRunner (#34681)

Browse Source
This commit is contained in:
Pulakesh 2026-04-07 22:53:14 +00:00 committed by GitHub
parent f8f7b0ec1a
commit 2575a3a3ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
api/core/agent/cot_chat_agent_runner.py
View File

@ -79,21 +79,18 @@ class CotChatAgentRunner(CotAgentRunner):
if not agent_scratchpad: if not agent_scratchpad:
assistant_messages = [] assistant_messages = []
else: else:
assistant_message = AssistantPromptMessage(content="") content = ""
assistant_message.content = "" # FIXME: type check tell mypy that assistant_message.content is str
for unit in agent_scratchpad: for unit in agent_scratchpad:
if unit.is_final(): if unit.is_final():
assert isinstance(assistant_message.content, str) content += f"Final Answer: {unit.agent_response}"
assistant_message.content += f"Final Answer: {unit.agent_response}"
else: else:
assert isinstance(assistant_message.content, str) content += f"Thought: {unit.thought}\n\n"
assistant_message.content += f"Thought: {unit.thought}\n\n"
if unit.action_str: if unit.action_str:
assistant_message.content += f"Action: {unit.action_str}\n\n" content += f"Action: {unit.action_str}\n\n"
if unit.observation: if unit.observation:
assistant_message.content += f"Observation: {unit.observation}\n\n" content += f"Observation: {unit.observation}\n\n"
assistant_messages = [assistant_message] assistant_messages = [AssistantPromptMessage(content=content)]
# query messages # query messages
query_messages = self._organize_user_query(self._query, []) query_messages = self._organize_user_query(self._query, [])