aient 1.0.35__py3-none-any.whl → 1.0.36__py3-none-any.whl

aient/models/chatgpt.py

@@ -61,6 +61,7 @@ class chatgpt(BaseLLM):
         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)
@@ -76,6 +77,7 @@ class chatgpt(BaseLLM):
         }
         self.function_calls_counter = {}
         self.function_call_max_loop = function_call_max_loop
+        self.cut_history_by_function_name = cut_history_by_function_name
 
 
         # 注册和处理传入的工具
@@ -124,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.35.dist-info → aient-1.0.36.dist-info}/METADATA

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

{aient-1.0.35.dist-info → aient-1.0.36.dist-info}/RECORD

@@ -12,7 +12,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
 aient/models/__init__.py,sha256=ouNDNvoBBpIFrLsk09Q_sq23HR0GbLAKfGLIFmfEuXE,219
 aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 aient/models/base.py,sha256=Loyt2F2WrDMBbK-sdmTtgkLVtdUXxK5tg4qoI6nc0Xo,7527
-aient/models/chatgpt.py,sha256=22BynRXOtWZeQzbJHYabPYNf7iabWq1JMG91lNoAUOM,37125
+aient/models/chatgpt.py,sha256=RIybztUTsxnGCM08zWVeyrBpIxUCSX5ApS-2IU9vC2Y,37748
 aient/models/claude.py,sha256=thK9P8qkaaoUN3OOJ9Shw4KDs-pAGKPoX4FOPGFXva8,28597
 aient/models/duckduckgo.py,sha256=1l7vYCs9SG5SWPCbcl7q6pCcB5AUF_r-a4l9frz3Ogo,8115
 aient/models/gemini.py,sha256=chGLc-8G_DAOxr10HPoOhvVFW1RvMgHd6mt--VyAW98,14730
@@ -29,8 +29,8 @@ aient/plugins/websearch.py,sha256=yiBzqXK5X220ibR-zko3VDsn4QOnLu1k6E2YOygCeTQ,15
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 aient/utils/scripts.py,sha256=O-0IXN3mezPauFs6fw83WDDgklpXTDvcbJBNTDrsIG0,8201
-aient-1.0.35.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.0.35.dist-info/METADATA,sha256=tMX7Wfg1VKnZsKP9mglhGK_bTYh32EhPpuKoQzTteF0,4986
-aient-1.0.35.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-aient-1.0.35.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.0.35.dist-info/RECORD,,
+aient-1.0.36.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.0.36.dist-info/METADATA,sha256=b70xJDgX9xQ4ga67B8NXYelG21cn9HDh8GanBhOR1R4,4986
+aient-1.0.36.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
+aient-1.0.36.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.0.36.dist-info/RECORD,,

{aient-1.0.35.dist-info → aient-1.0.36.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.3)
+Generator: setuptools (78.0.2)
 Root-Is-Purelib: true
 Tag: py3-none-any