aient 1.0.75__tar.gz → 1.0.77__tar.gz
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-1.0.75/src/aient.egg-info → aient-1.0.77}/PKG-INFO +1 -1
- {aient-1.0.75 → aient-1.0.77}/setup.py +1 -1
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/request.py +1 -1
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/chatgpt.py +13 -1
- {aient-1.0.75 → aient-1.0.77/src/aient.egg-info}/PKG-INFO +1 -1
- {aient-1.0.75 → aient-1.0.77}/LICENSE +0 -0
- {aient-1.0.75 → aient-1.0.77}/MANIFEST.in +0 -0
- {aient-1.0.75 → aient-1.0.77}/README.md +0 -0
- {aient-1.0.75 → aient-1.0.77}/setup.cfg +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/__init__.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/.git +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/.gitignore +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/__init__.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/log_config.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/models.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/response.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/test/test_base_api.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/test/test_image.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/test/test_payload.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/core/utils.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/__init__.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/audio.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/base.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/claude.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/duckduckgo.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/gemini.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/groq.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/models/vertex.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/__init__.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/arXiv.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/config.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/excute_command.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/get_time.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/image.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/list_directory.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/read_file.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/registry.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/run_python.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/websearch.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/plugins/write_file.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/prompt/__init__.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/prompt/agent.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/utils/__init__.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/utils/prompt.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient/utils/scripts.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient.egg-info/SOURCES.txt +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient.egg-info/dependency_links.txt +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient.egg-info/requires.txt +0 -0
- {aient-1.0.75 → aient-1.0.77}/src/aient.egg-info/top_level.txt +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_API.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_Deepbricks.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_Web_crawler.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_aiwaves.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_aiwaves_arxiv.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_ask_gemini.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_class.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_claude.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_claude_zh_char.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_ddg_search.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_download_pdf.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_gemini.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_get_token_dict.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_google_search.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_jieba.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_json.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_langchain_search_old.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_logging.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_ollama.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_plugin.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_py_run.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_requests.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_search.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_tikitoken.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_token.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_url.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_whisper.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_wildcard.py +0 -0
- {aient-1.0.75 → aient-1.0.77}/test/test_yjh.py +0 -0
|
@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
|
|
|
4
4
|
|
|
5
5
|
setup(
|
|
6
6
|
name="aient",
|
|
7
|
-
version="1.0.
|
|
7
|
+
version="1.0.77",
|
|
8
8
|
description="Aient: The Awakening of Agent.",
|
|
9
9
|
long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
|
|
10
10
|
long_description_content_type="text/markdown",
|
|
@@ -300,7 +300,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
|
|
|
300
300
|
original_model = model_dict[request.model]
|
|
301
301
|
search_tool = None
|
|
302
302
|
|
|
303
|
-
pro_models = ["gemini-2.5
|
|
303
|
+
pro_models = ["gemini-2.5", "gemini-2.0", "gemini-exp"]
|
|
304
304
|
if any(pro_model in original_model for pro_model in pro_models):
|
|
305
305
|
location = gemini2
|
|
306
306
|
search_tool = {"googleSearch": {}}
|
|
@@ -409,7 +409,19 @@ class chatgpt(BaseLLM):
|
|
|
409
409
|
|
|
410
410
|
function_parameter = parse_function_xml(full_response)
|
|
411
411
|
if function_parameter:
|
|
412
|
-
|
|
412
|
+
for tool_dict in function_parameter:
|
|
413
|
+
if tool_dict.get("function_name", "") not in self.plugins.keys():
|
|
414
|
+
function_parameter.remove(tool_dict)
|
|
415
|
+
if function_parameter:
|
|
416
|
+
need_function_call = True
|
|
417
|
+
else:
|
|
418
|
+
if self.print_log:
|
|
419
|
+
print("Failed to parse function_parameter", function_parameter)
|
|
420
|
+
print("full_response", full_response)
|
|
421
|
+
full_response = (
|
|
422
|
+
f"{full_response}"
|
|
423
|
+
"\n\nFailed to parse function parameter, I need to try again."
|
|
424
|
+
)
|
|
413
425
|
|
|
414
426
|
# 处理函数调用
|
|
415
427
|
if need_function_call and self.use_plugins != False:
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|