aient 1.2.42__tar.gz → 1.2.44__tar.gz

{aient-1.2.42 → aient-1.2.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.2.42
+Version: 1.2.44
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.2.42 → aient-1.2.44}/aient/core/request.py

@@ -295,7 +295,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude", "deepseek"]) and "-" not in key:
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -591,7 +591,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1015,9 +1015,16 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
             for item in msg.content:
                 if item.type == "text":
                     text_message = await get_text_message(item.text, engine)
+                    if "v1/responses" in url:
+                        text_message["type"] = "input_text"
                     content.append(text_message)
                 elif item.type == "image_url" and provider.get("image", True) and "o1-mini" not in original_model:
                     image_message = await get_image_message(item.image_url.url, engine)
+                    if "v1/responses" in url:
+                        image_message = {
+                            "type": "input_image",
+                            "image_url": image_message["image_url"]["url"]
+                        }
                     content.append(image_message)
         else:
             content = msg.content
@@ -1049,10 +1056,16 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
         system_msg = messages.pop(0)
         messages[0]["content"] = system_msg["content"] + messages[0]["content"]
 
-    payload = {
-        "model": original_model,
-        "messages": messages,
-    }
+    if "v1/responses" in url:
+        payload = {
+            "model": original_model,
+            "input": messages,
+        }
+    else:
+        payload = {
+            "model": original_model,
+            "messages": messages,
+        }
 
     miss_fields = [
         'model',
@@ -1085,11 +1098,20 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
     "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"
+            if "v1/responses" in url:
+                payload["reasoning"] = {"effort": "high"}
+            else:
+                payload["reasoning_effort"] = "high"
         elif request.model.endswith("low"):
-            payload["reasoning_effort"] = "low"
+            if "v1/responses" in url:
+                payload["reasoning"] = {"effort": "low"}
+            else:
+                payload["reasoning_effort"] = "low"
         else:
-            payload["reasoning_effort"] = "medium"
+            if "v1/responses" in url:
+                payload["reasoning"] = {"effort": "medium"}
+            else:
+                payload["reasoning_effort"] = "medium"
 
         if "temperature" in payload:
             payload.pop("temperature")
@@ -1127,7 +1149,7 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1225,7 +1247,7 @@ async def get_azure_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1345,7 +1367,7 @@ async def get_azure_databricks_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1432,7 +1454,7 @@ async def get_openrouter_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload
@@ -1798,7 +1820,7 @@ async def get_claude_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in model_dict.keys()) and "-" not in key:
                 payload[key] = value
 
     return url, headers, payload

{aient-1.2.42 → aient-1.2.44}/aient/core/response.py

@@ -213,6 +213,10 @@ async def fetch_gpt_response_stream(client, url, headers, payload, timeout):
 
         buffer = ""
         enter_buffer = ""
+
+        input_tokens = 0
+        output_tokens = 0
+
         async for chunk in response.aiter_text():
             buffer += chunk
             while "\n" in buffer:
@@ -221,12 +225,32 @@ async def fetch_gpt_response_stream(client, url, headers, payload, timeout):
                 if line.startswith(": keepalive"):
                     yield line + end_of_line
                     continue
-                if line and not line.startswith(":") and (result:=line.lstrip("data: ").strip()):
+                if line and not line.startswith(":") and (result:=line.lstrip("data: ").strip()) and not line.startswith("event: "):
                     if result.strip() == "[DONE]":
                         break
                     line = await asyncio.to_thread(json.loads, result)
                     line['id'] = f"chatcmpl-{random_str}"
 
+                    # v1/responses
+                    if line.get("type") == "response.reasoning_summary_text.delta" and line.get("delta"):
+                        sse_string = await generate_sse_response(timestamp, payload["model"], reasoning_content=line.get("delta"))
+                        yield sse_string
+                        continue
+                    elif line.get("type") == "response.output_text.delta" and line.get("delta"):
+                        sse_string = await generate_sse_response(timestamp, payload["model"], content=line.get("delta"))
+                        yield sse_string
+                        continue
+                    elif line.get("type") == "response.output_text.done":
+                        sse_string = await generate_sse_response(timestamp, payload["model"], stop="stop")
+                        yield sse_string
+                        continue
+                    elif line.get("type") == "response.completed":
+                        input_tokens = safe_get(line, "response", "usage", "input_tokens", default=0)
+                        output_tokens = safe_get(line, "response", "usage", "output_tokens", default=0)
+                        continue
+                    elif line.get("type", "").startswith("response."):
+                        continue
+
                     # 处理 <think> 标签
                     content = safe_get(line, "choices", 0, "delta", "content", default="")
                     if "<think>" in content:
@@ -322,6 +346,11 @@ async def fetch_gpt_response_stream(client, url, headers, payload, timeout):
                             del line["choices"][0]["message"]
                         json_line = await asyncio.to_thread(json.dumps, line)
                         yield "data: " + json_line.strip() + end_of_line
+
+    if input_tokens and output_tokens:
+        sse_string = await generate_sse_response(timestamp, payload["model"], None, None, None, None, None, total_tokens=input_tokens + output_tokens, prompt_tokens=input_tokens, completion_tokens=output_tokens)
+        yield sse_string
+
     yield "data: [DONE]" + end_of_line
 
 async def fetch_azure_response_stream(client, url, headers, payload, timeout):

{aient-1.2.42 → aient-1.2.44}/aient/core/utils.py

@@ -50,7 +50,11 @@ class BaseAPI:
             self.v1_models: str = urlunparse(parsed_url[:2] + ("v1/models",) + ("",) * 3)
         else:
             self.v1_models: str = urlunparse(parsed_url[:2] + (before_v1 + "models",) + ("",) * 3)
-        self.chat_url: str = urlunparse(parsed_url[:2] + (before_v1 + "chat/completions",) + ("",) * 3)
+
+        if "v1/responses" in parsed_url.path:
+            self.chat_url: str = urlunparse(parsed_url[:2] + ("v1/responses",) + ("",) * 3)
+        else:
+            self.chat_url: str = urlunparse(parsed_url[:2] + (before_v1 + "chat/completions",) + ("",) * 3)
         self.image_url: str = urlunparse(parsed_url[:2] + (before_v1 + "images/generations",) + ("",) * 3)
         if parsed_url.hostname == "dashscope.aliyuncs.com":
             self.audio_transcriptions: str = urlunparse(parsed_url[:2] + ("/api/v1/services/aigc/multimodal-generation/generation",) + ("",) * 3)

{aient-1.2.42 → aient-1.2.44}/aient/models/chatgpt.py

@@ -12,7 +12,7 @@ from typing import Union, Optional, Callable
 from .base import BaseLLM
 from ..plugins.registry import registry
 from ..plugins import PLUGINS, get_tools_result_async, function_call_list, update_tools_config
-from ..utils.scripts import safe_get, async_generator_to_sync, parse_function_xml, parse_continuous_json, convert_functions_to_xml, remove_xml_tags_and_content, find_most_frequent_phrase
+from ..utils.scripts import safe_get, async_generator_to_sync, parse_function_xml, parse_continuous_json, convert_functions_to_xml, remove_xml_tags_and_content
 from ..core.request import prepare_request_payload
 from ..core.response import fetch_response_stream, fetch_response
 from ..architext.architext import Messages, SystemMessage, UserMessage, AssistantMessage, ToolCalls, ToolResults, Texts, RoleMessage, Images, Files
@@ -80,7 +80,7 @@ class TaskComplete(Exception):
         self.completion_message = message
         super().__init__(f"Task completed with message: {message}")
 
-
+# 结尾重复响应错误
 class RepetitiveResponseError(Exception):
     """Custom exception for detecting repetitive and meaningless generated strings."""
     def __init__(self, message, phrase, count):
@@ -446,13 +446,6 @@ class chatgpt(BaseLLM):
 
         if not full_response.strip() and not need_function_call:
             raise EmptyResponseError("Response is empty")
-        most_frequent_phrase, most_frequent_phrase_count = find_most_frequent_phrase(full_response)
-        if most_frequent_phrase_count > 100:
-            raise RepetitiveResponseError(
-                f"Detected repetitive and meaningless content. The phrase '{most_frequent_phrase}' appeared {most_frequent_phrase_count} times.",
-                most_frequent_phrase,
-                most_frequent_phrase_count
-            )
 
         if self.print_log:
             self.logger.info(f"total_tokens: {total_tokens}")

{aient-1.2.42 → aient-1.2.44}/aient.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.2.42
+Version: 1.2.44
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.2.42 → aient-1.2.44}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aient"
-version = "1.2.42"
+version = "1.2.44"
 description = "Aient: The Awakening of Agent."
 readme = "README.md"
 requires-python = ">=3.11"