aient 1.1.39__tar.gz → 1.1.40__tar.gz

{aient-1.1.39/src/aient.egg-info → aient-1.1.40}/PKG-INFO

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

{aient-1.1.39 → aient-1.1.40}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.1.39",
+    version="1.1.40",
     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.39 → aient-1.1.40}/src/aient/core/request.py

@@ -240,6 +240,10 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
                 val = int(m.group(1))
                 if val < 0:
                     val = 0
+                elif 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 > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
@@ -532,6 +536,10 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
                 val = int(m.group(1))
                 if val < 0:
                     val = 0
+                elif 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 > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val

{aient-1.1.39 → aient-1.1.40}/src/aient/core/response.py

@@ -568,6 +568,7 @@ async def fetch_response(client, url, headers, payload, engine, model):
 
     elif engine == "gemini" or engine == "vertex-gemini" or engine == "aws":
         response_json = response.json()
+        # print("response_json", json.dumps(response_json, indent=4, ensure_ascii=False))
 
         if isinstance(response_json, str):
             import ast
@@ -579,7 +580,7 @@ async def fetch_response(client, url, headers, payload, engine, model):
         else:
             logger.error(f"error fetch_response: Unknown response_json type: {type(response_json)}")
             parsed_data = response_json
-        # print("parsed_data", json.dumps(parsed_data, indent=4, ensure_ascii=False))
+
         content = ""
         reasoning_content = ""
         image_base64 = ""
@@ -596,15 +597,6 @@ async def fetch_response(client, url, headers, payload, engine, model):
                     reasoning_content += chunk
                 else:
                     content += chunk
-        # for item in parsed_data:
-        #     chunk = safe_get(item, "candidates", 0, "content", "parts", 0, "text")
-        #     is_think = safe_get(item, "candidates", 0, "content", "parts", 0, "thought", default=False)
-        #     # logger.info(f"chunk: {repr(chunk)}")
-        #     if chunk:
-        #         if is_think:
-        #             reasoning_content += chunk
-        #         else:
-        #             content += chunk
 
         usage_metadata = safe_get(parsed_data, -1, "usageMetadata")
         prompt_tokens = safe_get(usage_metadata, "promptTokenCount", default=0)
@@ -618,8 +610,13 @@ async def fetch_response(client, url, headers, payload, engine, model):
             logger.error(f"Unknown role: {role}, parsed_data: {parsed_data}")
             role = "assistant"
 
-        function_call_name = safe_get(parsed_data, -1, "candidates", 0, "content", "parts", 0, "functionCall", "name", default=None)
-        function_call_content = safe_get(parsed_data, -1, "candidates", 0, "content", "parts", 0, "functionCall", "args", default=None)
+        has_think = safe_get(parsed_data, 0, "candidates", 0, "content", "parts", 0, "thought", default=False)
+        if has_think:
+            function_message_parts_index = -1
+        else:
+            function_message_parts_index = 0
+        function_call_name = safe_get(parsed_data, -1, "candidates", 0, "content", "parts", function_message_parts_index, "functionCall", "name", default=None)
+        function_call_content = safe_get(parsed_data, -1, "candidates", 0, "content", "parts", function_message_parts_index, "functionCall", "args", default=None)
 
         timestamp = int(datetime.timestamp(datetime.now()))
         yield await generate_no_stream_response(timestamp, model, content=content, tools_id=None, function_call_name=function_call_name, function_call_content=function_call_content, role=role, total_tokens=total_tokens, prompt_tokens=prompt_tokens, completion_tokens=candidates_tokens, reasoning_content=reasoning_content, image_base64=image_base64)

{aient-1.1.39 → aient-1.1.40/src/aient.egg-info}/PKG-INFO

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