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

aient/models/chatgpt.py

@@ -60,6 +60,7 @@ class chatgpt(BaseLLM):
         use_plugins: bool = True,
         print_log: bool = False,
         tools: Optional[Union[list, str, Callable]] = [],
+        function_call_max_loop: int = 3,
     ) -> None:
         """
         Initialize Chatbot with API key (from https://platform.openai.com/account/api-keys)
@@ -74,7 +75,7 @@ class chatgpt(BaseLLM):
             ],
         }
         self.function_calls_counter = {}
-        self.function_call_max_loop = 3
+        self.function_call_max_loop = function_call_max_loop
 
 
         # 注册和处理传入的工具

aient/plugins/config.py

@@ -3,19 +3,13 @@ import json
 import inspect
 
 from .registry import registry
-from ..utils.scripts import cut_message, safe_get
+from ..utils.scripts import cut_message
 from ..utils.prompt import search_key_word_prompt, arxiv_doc_user_prompt
 
 async def get_tools_result_async(function_call_name, function_full_response, function_call_max_tokens, engine, robot, api_key, api_url, use_plugins, model, add_message, convo_id, language):
     function_response = ""
     if function_call_name in registry.tools:
         function_to_call = registry.tools[function_call_name]
-        function_args = registry.tools_info[function_call_name].args
-        # required_args = registry.tools_info[function_call_name].required
-        if function_args:
-            arg = function_args[0]
-        else:
-            arg = None
     if function_call_name == "get_search_results":
         prompt = json.loads(function_full_response)["query"]
         yield "message_search_stage_1"
@@ -47,18 +41,12 @@ async def get_tools_result_async(function_call_name, function_full_response, fun
             function_response = "无法找到相关信息，停止使用 tools"
         # user_prompt = f"You need to response the following question: {prompt}. Search results is provided inside <Search_results></Search_results> XML tags. Your task is to think about the question step by step and then answer the above question in {config.language} based on the Search results provided. Please response in {config.language} and adopt a style that is logical, in-depth, and detailed. Note: In order to make the answer appear highly professional, you should be an expert in textual analysis, aiming to make the answer precise and comprehensive. Directly response markdown format, without using markdown code blocks"
         # self.add_to_conversation(user_prompt, "user", convo_id=convo_id)
-    elif arg: # generate_image get_url_content run_python_script
-        prompt = safe_get(json.loads(function_full_response), arg, default=".")
-        if inspect.iscoroutinefunction(function_to_call):
-            function_response = await function_to_call(prompt)
-        else:
-            function_response = function_to_call(prompt)
-        function_response, text_len = cut_message(function_response, function_call_max_tokens, engine)
-    else: # get_date_time_weekday
+    else:
+        prompt = json.loads(function_full_response)
         if inspect.iscoroutinefunction(function_to_call):
-            function_response = await function_to_call()
+            function_response = await function_to_call(**prompt)
         else:
-            function_response = function_to_call()
+            function_response = function_to_call(**prompt)
         function_response, text_len = cut_message(function_response, function_call_max_tokens, engine)
 
     if function_call_name == "download_read_arxiv_pdf":

{aient-1.0.33.dist-info → aient-1.0.35.dist-info}/METADATA

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

{aient-1.0.33.dist-info → aient-1.0.35.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=d7ZGc2WTrpwevXCFzIX4HU0zZ0T_mwfYkrStZ8yW5v0,37063
+aient/models/chatgpt.py,sha256=22BynRXOtWZeQzbJHYabPYNf7iabWq1JMG91lNoAUOM,37125
 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
@@ -20,7 +20,7 @@ aient/models/groq.py,sha256=2JCB0QE1htOprJHI5fZ11R2RtOhsHlsTjbmFyzc8oSM,10084
 aient/models/vertex.py,sha256=qVD5l1Q538xXUPulxG4nmDjXE1VoV4yuAkTCpIeJVw0,16795
 aient/plugins/__init__.py,sha256=KrCM6kFD1NB96hfhwUZIG8vJcdZVnfpACMew5YOWxSo,956
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
-aient/plugins/config.py,sha256=S7UHlo5e-czjssSyDQXVbR9bUlB21Fj934OfOKW5Vug,8191
+aient/plugins/config.py,sha256=J1x1newErZ44-IzUtm8gT9Rsx0SRrQxIY__p911dJZM,7543
 aient/plugins/image.py,sha256=ZElCIaZznE06TN9xW3DrSukS7U3A5_cjk1Jge4NzPxw,2072
 aient/plugins/registry.py,sha256=YknzhieU_8nQ3oKlUSSWDB4X7t2Jx0JnqT2Jd9Xsvfk,3574
 aient/plugins/run_python.py,sha256=dgcUwBunMuDkaSKR5bToudVzSdrXVewktDDFUz_iIOQ,4589
@@ -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.33.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.0.33.dist-info/METADATA,sha256=KCS4VPiA0DuFlqX_ulQvFaMDWzC04fV5F_U9jXQrsWw,4986
-aient-1.0.33.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-aient-1.0.33.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.0.33.dist-info/RECORD,,
+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,,