aient 1.1.0__tar.gz → 1.1.2__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.
Files changed (79) hide show
  1. {aient-1.1.0/src/aient.egg-info → aient-1.1.2}/PKG-INFO +1 -1
  2. {aient-1.1.0 → aient-1.1.2}/setup.py +1 -1
  3. {aient-1.1.0 → aient-1.1.2}/src/aient/core/request.py +1 -4
  4. {aient-1.1.0 → aient-1.1.2}/src/aient/core/response.py +1 -1
  5. {aient-1.1.0 → aient-1.1.2/src/aient.egg-info}/PKG-INFO +1 -1
  6. {aient-1.1.0 → aient-1.1.2}/LICENSE +0 -0
  7. {aient-1.1.0 → aient-1.1.2}/MANIFEST.in +0 -0
  8. {aient-1.1.0 → aient-1.1.2}/README.md +0 -0
  9. {aient-1.1.0 → aient-1.1.2}/setup.cfg +0 -0
  10. {aient-1.1.0 → aient-1.1.2}/src/aient/__init__.py +0 -0
  11. {aient-1.1.0 → aient-1.1.2}/src/aient/core/.git +0 -0
  12. {aient-1.1.0 → aient-1.1.2}/src/aient/core/.gitignore +0 -0
  13. {aient-1.1.0 → aient-1.1.2}/src/aient/core/__init__.py +0 -0
  14. {aient-1.1.0 → aient-1.1.2}/src/aient/core/log_config.py +0 -0
  15. {aient-1.1.0 → aient-1.1.2}/src/aient/core/models.py +0 -0
  16. {aient-1.1.0 → aient-1.1.2}/src/aient/core/test/test_base_api.py +0 -0
  17. {aient-1.1.0 → aient-1.1.2}/src/aient/core/test/test_geminimask.py +0 -0
  18. {aient-1.1.0 → aient-1.1.2}/src/aient/core/test/test_image.py +0 -0
  19. {aient-1.1.0 → aient-1.1.2}/src/aient/core/test/test_payload.py +0 -0
  20. {aient-1.1.0 → aient-1.1.2}/src/aient/core/utils.py +0 -0
  21. {aient-1.1.0 → aient-1.1.2}/src/aient/models/__init__.py +0 -0
  22. {aient-1.1.0 → aient-1.1.2}/src/aient/models/audio.py +0 -0
  23. {aient-1.1.0 → aient-1.1.2}/src/aient/models/base.py +0 -0
  24. {aient-1.1.0 → aient-1.1.2}/src/aient/models/chatgpt.py +0 -0
  25. {aient-1.1.0 → aient-1.1.2}/src/aient/models/claude.py +0 -0
  26. {aient-1.1.0 → aient-1.1.2}/src/aient/models/duckduckgo.py +0 -0
  27. {aient-1.1.0 → aient-1.1.2}/src/aient/models/gemini.py +0 -0
  28. {aient-1.1.0 → aient-1.1.2}/src/aient/models/groq.py +0 -0
  29. {aient-1.1.0 → aient-1.1.2}/src/aient/models/vertex.py +0 -0
  30. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/__init__.py +0 -0
  31. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/arXiv.py +0 -0
  32. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/config.py +0 -0
  33. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/excute_command.py +0 -0
  34. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/get_time.py +0 -0
  35. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/image.py +0 -0
  36. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/list_directory.py +0 -0
  37. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/read_file.py +0 -0
  38. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/registry.py +0 -0
  39. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/run_python.py +0 -0
  40. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/websearch.py +0 -0
  41. {aient-1.1.0 → aient-1.1.2}/src/aient/plugins/write_file.py +0 -0
  42. {aient-1.1.0 → aient-1.1.2}/src/aient/prompt/__init__.py +0 -0
  43. {aient-1.1.0 → aient-1.1.2}/src/aient/prompt/agent.py +0 -0
  44. {aient-1.1.0 → aient-1.1.2}/src/aient/utils/__init__.py +0 -0
  45. {aient-1.1.0 → aient-1.1.2}/src/aient/utils/prompt.py +0 -0
  46. {aient-1.1.0 → aient-1.1.2}/src/aient/utils/scripts.py +0 -0
  47. {aient-1.1.0 → aient-1.1.2}/src/aient.egg-info/SOURCES.txt +0 -0
  48. {aient-1.1.0 → aient-1.1.2}/src/aient.egg-info/dependency_links.txt +0 -0
  49. {aient-1.1.0 → aient-1.1.2}/src/aient.egg-info/requires.txt +0 -0
  50. {aient-1.1.0 → aient-1.1.2}/src/aient.egg-info/top_level.txt +0 -0
  51. {aient-1.1.0 → aient-1.1.2}/test/test.py +0 -0
  52. {aient-1.1.0 → aient-1.1.2}/test/test_API.py +0 -0
  53. {aient-1.1.0 → aient-1.1.2}/test/test_Deepbricks.py +0 -0
  54. {aient-1.1.0 → aient-1.1.2}/test/test_Web_crawler.py +0 -0
  55. {aient-1.1.0 → aient-1.1.2}/test/test_aiwaves.py +0 -0
  56. {aient-1.1.0 → aient-1.1.2}/test/test_aiwaves_arxiv.py +0 -0
  57. {aient-1.1.0 → aient-1.1.2}/test/test_ask_gemini.py +0 -0
  58. {aient-1.1.0 → aient-1.1.2}/test/test_class.py +0 -0
  59. {aient-1.1.0 → aient-1.1.2}/test/test_claude.py +0 -0
  60. {aient-1.1.0 → aient-1.1.2}/test/test_claude_zh_char.py +0 -0
  61. {aient-1.1.0 → aient-1.1.2}/test/test_ddg_search.py +0 -0
  62. {aient-1.1.0 → aient-1.1.2}/test/test_download_pdf.py +0 -0
  63. {aient-1.1.0 → aient-1.1.2}/test/test_gemini.py +0 -0
  64. {aient-1.1.0 → aient-1.1.2}/test/test_get_token_dict.py +0 -0
  65. {aient-1.1.0 → aient-1.1.2}/test/test_google_search.py +0 -0
  66. {aient-1.1.0 → aient-1.1.2}/test/test_jieba.py +0 -0
  67. {aient-1.1.0 → aient-1.1.2}/test/test_json.py +0 -0
  68. {aient-1.1.0 → aient-1.1.2}/test/test_logging.py +0 -0
  69. {aient-1.1.0 → aient-1.1.2}/test/test_ollama.py +0 -0
  70. {aient-1.1.0 → aient-1.1.2}/test/test_plugin.py +0 -0
  71. {aient-1.1.0 → aient-1.1.2}/test/test_py_run.py +0 -0
  72. {aient-1.1.0 → aient-1.1.2}/test/test_requests.py +0 -0
  73. {aient-1.1.0 → aient-1.1.2}/test/test_search.py +0 -0
  74. {aient-1.1.0 → aient-1.1.2}/test/test_tikitoken.py +0 -0
  75. {aient-1.1.0 → aient-1.1.2}/test/test_token.py +0 -0
  76. {aient-1.1.0 → aient-1.1.2}/test/test_url.py +0 -0
  77. {aient-1.1.0 → aient-1.1.2}/test/test_whisper.py +0 -0
  78. {aient-1.1.0 → aient-1.1.2}/test/test_wildcard.py +0 -0
  79. {aient-1.1.0 → aient-1.1.2}/test/test_yjh.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aient
3
- Version: 1.1.0
3
+ Version: 1.1.2
4
4
  Summary: Aient: The Awakening of Agent.
5
5
  Description-Content-Type: text/markdown
6
6
  License-File: LICENSE
@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
4
4
 
5
5
  setup(
6
6
  name="aient",
7
- version="1.1.0",
7
+ version="1.1.2",
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",
@@ -459,10 +459,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
459
459
  payload["generationConfig"]["max_output_tokens"] = 8192
460
460
 
461
461
  if request.model.endswith("-search"):
462
- if "tools" not in payload:
463
- payload["tools"] = [search_tool]
464
- else:
465
- payload["tools"].append(search_tool)
462
+ payload["tools"] = [search_tool]
466
463
 
467
464
  return url, headers, payload
468
465
 
@@ -67,7 +67,7 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
67
67
  if (line and '"parts": [' in line or parts_json != "") and is_finish == False:
68
68
  parts_json += line
69
69
  if parts_json != "" and line and '],' == line.strip():
70
- parts_json = "{" + parts_json.strip().rstrip(",} ]}") + "}]}"
70
+ parts_json = "{" + parts_json.strip().replace(', "groundingMetadata": {}', "").rstrip(",} ]}").lstrip("{") + "}]}"
71
71
  try:
72
72
  json_data = json.loads(parts_json)
73
73
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aient
3
- Version: 1.1.0
3
+ Version: 1.1.2
4
4
  Summary: Aient: The Awakening of Agent.
5
5
  Description-Content-Type: text/markdown
6
6
  License-File: LICENSE
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