aient 1.1.54__tar.gz → 1.1.55__tar.gz

{aient-1.1.54/src/aient.egg-info → aient-1.1.55}/PKG-INFO

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

{aient-1.1.54 → aient-1.1.55}/setup.py

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

@@ -1071,7 +1071,9 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
         elif request.model.endswith("low"):
             payload["reasoning_effort"] = "low"
 
-    if "o1" in original_model or "o3" in original_model or "o4" in original_model:
+    if "o1" in original_model or \
+    "o3" in original_model or "o4" in original_model or \
+    "gpt-oss" in original_model or "gpt-5" in original_model:
         if request.model.endswith("high"):
             payload["reasoning_effort"] = "high"
         elif request.model.endswith("low"):

{aient-1.1.54 → aient-1.1.55}/src/aient/core/response.py

@@ -34,12 +34,14 @@ def gemini_json_poccess(response_str):
         candidatesTokenCount = safe_get(response_json, "usageMetadata", "candidatesTokenCount", default=0)
         totalTokenCount = safe_get(response_json, "usageMetadata", "totalTokenCount", default=0)
 
-    content = safe_get(json_data, "parts", 0, "text", default="")
+    content = reasoning_content = safe_get(json_data, "parts", 0, "text", default="")
     b64_json = safe_get(json_data, "parts", 0, "inlineData", "data", default="")
     if b64_json:
         image_base64 = b64_json
 
     is_thinking = safe_get(json_data, "parts", 0, "thought", default=False)
+    if is_thinking:
+        content = safe_get(json_data, "parts", 1, "text", default="")
 
     function_call_name = safe_get(json_data, "functionCall", "name", default=None)
     function_full_response = safe_get(json_data, "functionCall", "args", default="")
@@ -47,7 +49,7 @@ def gemini_json_poccess(response_str):
 
     blockReason = safe_get(json_data, 0, "promptFeedback", "blockReason", default=None)
 
-    return is_thinking, content, image_base64, function_call_name, function_full_response, finishReason, blockReason, promptTokenCount, candidatesTokenCount, totalTokenCount
+    return is_thinking, reasoning_content, content, image_base64, function_call_name, function_full_response, finishReason, blockReason, promptTokenCount, candidatesTokenCount, totalTokenCount
 
 async def fetch_gemini_response_stream(client, url, headers, payload, model):
     timestamp = int(datetime.timestamp(datetime.now()))
@@ -63,11 +65,18 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
         parts_json = ""
         async for chunk in response.aiter_text():
             buffer += chunk
+            if buffer and "\n" not in buffer:
+                buffer += "\n"
 
             while "\n" in buffer:
                 line, buffer = buffer.split("\n", 1)
                 if line.startswith("data: "):
                     parts_json = line.lstrip("data: ").strip()
+                    try:
+                        json.loads(parts_json)
+                    except json.JSONDecodeError:
+                        logger.error(f"JSON decode error: {parts_json}")
+                        continue
                 else:
                     parts_json += line
                     parts_json = parts_json.lstrip("[,")
@@ -77,12 +86,12 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
                         continue
 
                 # https://ai.google.dev/api/generate-content?hl=zh-cn#FinishReason
-                is_thinking, content, image_base64, function_call_name, function_full_response, finishReason, blockReason, promptTokenCount, candidatesTokenCount, totalTokenCount = gemini_json_poccess(parts_json)
+                is_thinking, reasoning_content, content, image_base64, function_call_name, function_full_response, finishReason, blockReason, promptTokenCount, candidatesTokenCount, totalTokenCount = gemini_json_poccess(parts_json)
 
                 if is_thinking:
-                    sse_string = await generate_sse_response(timestamp, model, reasoning_content=content)
+                    sse_string = await generate_sse_response(timestamp, model, reasoning_content=reasoning_content)
                     yield sse_string
-                elif not image_base64 and content:
+                if not image_base64 and content:
                     sse_string = await generate_sse_response(timestamp, model, content=content)
                     yield sse_string
 

{aient-1.1.54 → aient-1.1.55/src/aient.egg-info}/PKG-INFO

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