aient 1.2.36__tar.gz → 1.2.38__tar.gz

{aient-1.2.36 → aient-1.2.38}/PKG-INFO

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

{aient-1.2.36 → aient-1.2.38}/aient/core/request.py

@@ -196,7 +196,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
 
     for field, value in request.model_dump(exclude_unset=True).items():
         if field not in miss_fields and value is not None:
-            if field == "tools" and "gemini-2.0-flash-thinking" in original_model:
+            if field == "tools" and ("gemini-2.0-flash-thinking" in original_model or "gemini-2.5-flash-image" in original_model):
                 continue
             if field == "tools":
                 # 处理每个工具的 function 定义
@@ -221,6 +221,8 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
                         }
                     })
             elif field == "temperature":
+                if "gemini-2.5-flash-image" in original_model:
+                    value = 1
                 generation_config["temperature"] = value
             elif field == "max_tokens":
                 if value > 65536:
@@ -244,7 +246,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
                 "Image",
             ]
 
-    if "gemini-2.5" in original_model:
+    if "gemini-2.5" in original_model and "gemini-2.5-flash-image" not in original_model:
         # 从请求模型名中检测思考预算设置
         m = re.match(r".*-think-(-?\d+)", request.model)
         if m:

{aient-1.2.36 → aient-1.2.38}/aient/core/response.py

@@ -78,7 +78,7 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model, tim
                     try:
                         response_json = await asyncio.to_thread(json.loads, parts_json)
                     except json.JSONDecodeError:
-                        logger.error(f"JSON decode error: {parts_json}")
+                        # logger.error(f"JSON decode error: {parts_json}")
                         continue
                 else:
                     parts_json += line
@@ -99,7 +99,12 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model, tim
                     yield sse_string
 
                 if image_base64:
-                    yield await generate_no_stream_response(timestamp, model, content=content, tools_id=None, function_call_name=None, function_call_content=None, role=None, total_tokens=totalTokenCount, prompt_tokens=promptTokenCount, completion_tokens=candidatesTokenCount, image_base64=image_base64)
+                    if "gemini-2.5-flash-image" not in model:
+                        yield await generate_no_stream_response(timestamp, model, content=content, tools_id=None, function_call_name=None, function_call_content=None, role=None, total_tokens=totalTokenCount, prompt_tokens=promptTokenCount, completion_tokens=candidatesTokenCount, image_base64=image_base64)
+                    else:
+                        image_url = await upload_image_to_0x0st("data:image/png;base64," + image_base64)
+                        sse_string = await generate_sse_response(timestamp, model, content=f"\n\n![image]({image_url})")
+                        yield sse_string
 
                 if function_call_name:
                     sse_string = await generate_sse_response(timestamp, model, content=None, tools_id="chatcmpl-9inWv0yEtgn873CxMBzHeCeiHctTV", function_call_name=function_call_name)

{aient-1.2.36 → aient-1.2.38}/aient/models/chatgpt.py

@@ -108,6 +108,7 @@ class chatgpt(BaseLLM):
         cache_messages: list = None,
         logger: logging.Logger = None,
         check_done: bool = False,
+        retry_count: int = 999999,
     ) -> None:
         """
         Initialize Chatbot with API key (from https://platform.openai.com/account/api-keys)
@@ -119,7 +120,7 @@ class chatgpt(BaseLLM):
         self.function_calls_counter = {}
         self.function_call_max_loop = function_call_max_loop
         self.check_done = check_done
-
+        self.retry_count = retry_count
         if logger:
             self.logger = logger
         else:
@@ -708,7 +709,8 @@ class chatgpt(BaseLLM):
         # 发送请求并处理响应
         retry_times = 0
         error_to_raise = None
-        while True:
+        while retry_times < self.retry_count:
+            retry_times += 1
             tmp_post_json = copy.deepcopy(json_post)
             if need_done_prompt:
                 tmp_post_json["messages"].extend(need_done_prompt)
@@ -821,8 +823,7 @@ class chatgpt(BaseLLM):
                     error_message = "您输入了无效的API URL，请使用正确的URL并使用`/start`命令重新设置API URL。具体错误如下：\n\n" + str(e)
                     raise ConfigurationError(error_message)
                 # 最后一次重试失败，向上抛出异常
-                retry_times += 1
-                if retry_times == 9:
+                if retry_times == self.retry_count:
                     raise RetryFailedError(str(e))
 
         if error_to_raise:

{aient-1.2.36 → aient-1.2.38}/aient.egg-info/PKG-INFO

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

{aient-1.2.36 → aient-1.2.38}/pyproject.toml

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