aient 1.0.34__tar.gz → 1.0.36__tar.gz

{aient-1.0.34/src/aient.egg-info → aient-1.0.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.0.34
+Version: 1.0.36
 Summary: Aient: The Awakening of Agent.
 Description-Content-Type: text/markdown
 License-File: LICENSE

{aient-1.0.34 → aient-1.0.36}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.0.34",
+    version="1.0.36",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{aient-1.0.34 → aient-1.0.36}/src/aient/models/chatgpt.py

@@ -60,6 +60,8 @@ class chatgpt(BaseLLM):
         use_plugins: bool = True,
         print_log: bool = False,
         tools: Optional[Union[list, str, Callable]] = [],
+        function_call_max_loop: int = 3,
+        cut_history_by_function_name: str = "",
     ) -> None:
         """
         Initialize Chatbot with API key (from https://platform.openai.com/account/api-keys)
@@ -74,7 +76,8 @@ class chatgpt(BaseLLM):
             ],
         }
         self.function_calls_counter = {}
-        self.function_call_max_loop = 3
+        self.function_call_max_loop = function_call_max_loop
+        self.cut_history_by_function_name = cut_history_by_function_name
 
 
         # 注册和处理传入的工具
@@ -123,6 +126,11 @@ class chatgpt(BaseLLM):
         if function_name == "" and message and message != None:
             self.conversation[convo_id].append({"role": role, "content": message})
         elif function_name != "" and message and message != None:
+            # 删除从 cut_history_by_function_name 以后的所有历史记录
+            if function_name == self.cut_history_by_function_name:
+                matching_message = next(filter(lambda x: safe_get(x, "tool_calls", 0, "function", "name", default="") == 'get_next_pdf', self.conversation[convo_id]), None)
+                if matching_message is not None:
+                    self.conversation[convo_id] = self.conversation[convo_id][:self.conversation[convo_id].index(matching_message)]
             self.conversation[convo_id].append({
                 "role": "assistant",
                 "tool_calls": [

{aient-1.0.34 → aient-1.0.36/src/aient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.0.34
+Version: 1.0.36
 Summary: Aient: The Awakening of Agent.
 Description-Content-Type: text/markdown
 License-File: LICENSE