aient 1.0.90__py3-none-any.whl → 1.0.91__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
aient/models/chatgpt.py CHANGED
@@ -125,9 +125,9 @@ class chatgpt(BaseLLM):
125
125
  # print("role", role, "function_name", function_name, "message", message)
126
126
  if convo_id not in self.conversation:
127
127
  self.reset(convo_id=convo_id)
128
- if function_name == "" and message and message != None:
128
+ if function_name == "" and message:
129
129
  self.conversation[convo_id].append({"role": role, "content": message})
130
- elif function_name != "" and message and message != None:
130
+ elif function_name != "" and message:
131
131
  # 删除从 cut_history_by_function_name 以后的所有历史记录
132
132
  if function_name == self.cut_history_by_function_name:
133
133
  matching_message = next(filter(lambda x: safe_get(x, "tool_calls", 0, "function", "name", default="") == 'get_next_pdf', self.conversation[convo_id]), None)
@@ -429,26 +429,27 @@ class chatgpt(BaseLLM):
429
429
  if response_role is None:
430
430
  response_role = "assistant"
431
431
 
432
- function_parameter = parse_function_xml(full_response)
433
- if function_parameter:
434
- invalid_tools = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") not in self.plugins.keys()]
435
- function_parameter = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") in self.plugins.keys()]
436
- for tool_dict in invalid_tools:
437
- full_response = full_response + f"\n\nFunction: {tool_dict.get('function_name', '')} does not exist! I must use existing functions. I need to try again."
438
- if self.print_log and invalid_tools:
439
- print("invalid_tools", invalid_tools)
440
- print("function_parameter", function_parameter)
441
- print("full_response", full_response)
432
+ if self.use_plugins == True:
433
+ function_parameter = parse_function_xml(full_response)
442
434
  if function_parameter:
443
- need_function_call = True
444
- else:
445
- need_function_call = False
446
- if self.print_log:
447
- print("Failed to parse function_parameter full_response", full_response)
448
- full_response = ""
435
+ invalid_tools = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") not in self.plugins.keys()]
436
+ function_parameter = [tool_dict for tool_dict in function_parameter if tool_dict.get("function_name", "") in self.plugins.keys()]
437
+ for tool_dict in invalid_tools:
438
+ full_response = full_response + f"\n\nFunction: {tool_dict.get('function_name', '')} does not exist! I must use existing functions. I need to try again."
439
+ if self.print_log and invalid_tools:
440
+ print("invalid_tools", invalid_tools)
441
+ print("function_parameter", function_parameter)
442
+ print("full_response", full_response)
443
+ if function_parameter:
444
+ need_function_call = True
445
+ else:
446
+ need_function_call = False
447
+ if self.print_log:
448
+ print("Failed to parse function_parameter full_response", full_response)
449
+ full_response = ""
449
450
 
450
451
  # 处理函数调用
451
- if need_function_call and self.use_plugins != False:
452
+ if need_function_call and self.use_plugins == True:
452
453
  if self.print_log:
453
454
  print("function_parameter", function_parameter)
454
455
  print("function_full_response", function_full_response)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aient
3
- Version: 1.0.90
3
+ Version: 1.0.91
4
4
  Summary: Aient: The Awakening of Agent.
5
5
  Description-Content-Type: text/markdown
6
6
  License-File: LICENSE
@@ -13,7 +13,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
13
13
  aient/models/__init__.py,sha256=ouNDNvoBBpIFrLsk09Q_sq23HR0GbLAKfGLIFmfEuXE,219
14
14
  aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
15
15
  aient/models/base.py,sha256=z-Z0pJfTN2x0cuwfvu0BdMRY9O-RmLwHEnBIJN1x4Fg,6719
16
- aient/models/chatgpt.py,sha256=Y99EIUE9m7QIo3G-6MQWQ_HShCv5tDQHYM-nQyV32DM,45002
16
+ aient/models/chatgpt.py,sha256=-NWkkKxTCyraPYT0YN37NA2rUfOaDNXtvFSQmIE5tS8,45066
17
17
  aient/models/claude.py,sha256=thK9P8qkaaoUN3OOJ9Shw4KDs-pAGKPoX4FOPGFXva8,28597
18
18
  aient/models/duckduckgo.py,sha256=1l7vYCs9SG5SWPCbcl7q6pCcB5AUF_r-a4l9frz3Ogo,8115
19
19
  aient/models/gemini.py,sha256=chGLc-8G_DAOxr10HPoOhvVFW1RvMgHd6mt--VyAW98,14730
@@ -36,8 +36,8 @@ aient/prompt/agent.py,sha256=WLs0KiNZs29FJ5w7N3kQZYLVEeS7K8vxIOUw06LcXVE,23825
36
36
  aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
37
37
  aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
38
38
  aient/utils/scripts.py,sha256=PPwaJEigPkpciJHUXOag483iq1GjvaLReHDHjkinv6c,26780
39
- aient-1.0.90.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
40
- aient-1.0.90.dist-info/METADATA,sha256=r6fU6TadrmIOJb-YQgLT5nuH1IcipmCK--zNMS_-Ub4,5000
41
- aient-1.0.90.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
42
- aient-1.0.90.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
43
- aient-1.0.90.dist-info/RECORD,,
39
+ aient-1.0.91.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
40
+ aient-1.0.91.dist-info/METADATA,sha256=6QuaVz9_62x2b2ynwHT1dQULl-FJCTBMdBYOOh-3MZw,5000
41
+ aient-1.0.91.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
42
+ aient-1.0.91.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
43
+ aient-1.0.91.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.0.0)
2
+ Generator: setuptools (80.0.1)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5