aient 1.2.20__tar.gz → 1.2.22__tar.gz

{aient-1.2.20 → aient-1.2.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.2.20
+Version: 1.2.22
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.2.20 → aient-1.2.22}/aient/models/chatgpt.py

@@ -188,9 +188,9 @@ class chatgpt(BaseLLM):
                             ])
                     self.conversation[convo_id].append(AssistantMessage(convert_functions_to_xml(function_arguments)))
                     if image_message_list:
-                        self.conversation[convo_id].append(UserMessage(message + image_message_list))
+                        self.conversation[convo_id].append(UserMessage(message + image_message_list, Texts("\n\nYour message **must** end with [done] to signify the end of your output.", name="done", visible=self.check_done)))
                     else:
-                        self.conversation[convo_id].append(UserMessage(message))
+                        self.conversation[convo_id].append(UserMessage(message, Texts("\n\nYour message **must** end with [done] to signify the end of your output.", name="done", visible=self.check_done)))
                 else:
                     self.conversation[convo_id].append(AssistantMessage("我已经执行过这个工具了，接下来我需要做什么？"))
         else:
@@ -586,7 +586,7 @@ class chatgpt(BaseLLM):
                 if tool_name == "read_file" and "<tool_error>" not in tool_response:
                     self.conversation[convo_id].provider("files").update(tool_info['parameter']["file_path"], tool_response)
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\nRead file successfully! The file content has been updated in the tag <latest_file_content>.")
-                if tool_name == "get_knowledge_graph_tree" and "<tool_error>" not in tool_response:
+                elif tool_name == "get_knowledge_graph_tree" and "<tool_error>" not in tool_response:
                     self.conversation[convo_id].provider("knowledge_graph").visible = True
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\nGet knowledge graph tree successfully! The knowledge graph tree has been updated in the tag <knowledge_graph_tree>.")
                 elif tool_name == "write_to_file" and "<tool_error>" not in tool_response:
@@ -600,9 +600,6 @@ class chatgpt(BaseLLM):
                 else:
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\n{tool_response}")
 
-            if self.check_done:
-                all_responses.append("Your message **must** end with [done] to signify the end of your output.")
-
             # 合并所有工具响应
             function_response = "\n\n".join(all_responses).strip()
             if missing_required_params:

{aient-1.2.20 → aient-1.2.22}/aient.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.2.20
+Version: 1.2.22
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.2.20 → aient-1.2.22}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aient"
-version = "1.2.20"
+version = "1.2.22"
 description = "Aient: The Awakening of Agent."
 readme = "README.md"
 requires-python = ">=3.11"