aient 1.1.41__tar.gz → 1.1.43__tar.gz

{aient-1.1.41/src/aient.egg-info → aient-1.1.43}/PKG-INFO

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

{aient-1.1.41 → aient-1.1.43}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.1.41",
+    version="1.1.43",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{aient-1.1.41 → aient-1.1.43}/src/aient/core/request.py

@@ -239,15 +239,17 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
         if m:
             try:
                 val = int(m.group(1))
-                if val < 0:
-                    val = 0
-                elif val > 32768 and "gemini-2.5-pro" in original_model:
+                if val > 32768 and "gemini-2.5-pro" in original_model:
                     val = 32768
                 elif val < 128 and "gemini-2.5-pro" in original_model:
                     val = 128
+                elif val <= 0:
+                    val = 0
                 elif val > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
+                if val == 0:
+                    payload["generationConfig"].pop("thinkingConfig", None)
             except ValueError:
                 # 如果转换为整数失败，忽略思考预算设置
                 pass
@@ -535,15 +537,17 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
         if m:
             try:
                 val = int(m.group(1))
-                if val < 0:
-                    val = 0
-                elif val > 32768 and "gemini-2.5-pro" in original_model:
+                if val > 32768 and "gemini-2.5-pro" in original_model:
                     val = 32768
                 elif val < 128 and "gemini-2.5-pro" in original_model:
                     val = 128
+                elif val <= 0:
+                    val = 0
                 elif val > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
+                if val == 0:
+                    payload["generationConfig"].pop("thinkingConfig", None)
             except ValueError:
                 # 如果转换为整数失败，忽略思考预算设置
                 pass

{aient-1.1.41 → aient-1.1.43}/src/aient/plugins/websearch.py

@@ -383,7 +383,7 @@ if __name__ == "__main__":
     # result = asyncio.run(docQA("https://yym68686.top", "说一下HSTL pipeline"))
     # result = asyncio.run(docQA("https://wiki.yym68686.top", "PyTorch to MindSpore翻译思路是什么？"))
     # print(result['answer'])
-    # result = asyncio.run(pdfQA("https://api.telegram.org/file/bot5569497961:AAHobhUuydAwD8SPkXZiVFybvZJOmGrST_w/documents/file_1.pdf", "HSTL的pipeline详细讲一下"))
+    # result = asyncio.run(pdfQA("https://api.telegram.org/file/bot/documents/file_1.pdf", "HSTL的pipeline详细讲一下"))
     # print(result)
     # source_url = set([i.metadata['source'] for i in result["source_documents"]])
     # source_url = "\n".join(source_url)

{aient-1.1.41 → aient-1.1.43/src/aient.egg-info}/PKG-INFO

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