aient 1.1.51__tar.gz → 1.1.52__tar.gz

{aient-1.1.51/src/aient.egg-info → aient-1.1.52}/PKG-INFO

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

{aient-1.1.51 → aient-1.1.52}/setup.py

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

@@ -3,6 +3,8 @@ import json
 import httpx
 import base64
 import urllib.parse
+from io import IOBase
+from typing import Tuple
 
 from .models import RequestModel, Message
 from .utils import (
@@ -239,7 +241,6 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
             ]
 
     if "gemini-2.5" in original_model:
-        generation_config = payload.get("generationConfig", {})
         # 从请求模型名中检测思考预算设置
         m = re.match(r".*-think-(-?\d+)", request.model)
         if m:
@@ -254,7 +255,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
                         budget = 32768
                     else: # 128 <= val <= 32768
                         budget = val
-                
+
                 # gemini-2.5-flash-lite: [0] or [512, 24576]
                 elif "gemini-2.5-flash-lite" in original_model:
                     if val > 0 and val < 512:
@@ -270,9 +271,9 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
                         budget = 24576
                     else: # Includes 0 and valid range, and clamps invalid negatives
                         budget = val if val >= 0 else 0
-                
+
                 payload["generationConfig"]["thinkingConfig"] = {
-                    "includeThoughts": True,
+                    "includeThoughts": True if budget else False,
                     "thinkingBudget": budget
                 }
             except ValueError:
@@ -282,11 +283,6 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
             payload["generationConfig"]["thinkingConfig"] = {
                 "includeThoughts": True,
             }
-        payload["generationConfig"] = generation_config
-
-    # # 检测search标签
-    # if request.model.endswith("-search"):
-    #     payload["tools"] = [{"googleSearch": {}}]
 
     if safe_get(provider, "preferences", "post_body_parameter_overrides", default=None):
         for key, value in safe_get(provider, "preferences", "post_body_parameter_overrides", default={}).items():
@@ -372,16 +368,12 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
     # search_tool = None
 
     # https://cloud.google.com/vertex-ai/generative-ai/docs/models/gemini/2-0-flash?hl=zh-cn
-    pro_models = ["gemini-2.5", "gemini-2.0"]
+    pro_models = ["gemini-2.5"]
     if any(pro_model in original_model for pro_model in pro_models):
-        location = gemini2
+        location = gemini2_5_pro_exp
     else:
         location = gemini1
 
-    if "gemini-2.5-flash-lite-preview-06-17" == original_model or \
-    "gemini-2.5-pro-preview-06-05" == original_model:
-        location = gemini2_5_pro_exp
-
     if "google-vertex-ai" in provider.get("base_url", ""):
         url = provider.get("base_url").rstrip('/') + "/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/google/models/{MODEL_ID}:{stream}".format(
             LOCATION=await location.next(),
@@ -390,24 +382,8 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
             stream=gemini_stream
         )
     elif api_key is not None and api_key[2] == ".":
-        if provider.get("project_id") and "gemini-2.5-pro-preview-06-05" == original_model:
-            if isinstance(provider.get("project_id"), list):
-                api_key_index = provider.get("api").index(api_key)
-                project_id = provider.get("project_id")[api_key_index]
-            else:
-                project_id = provider.get("project_id")
-            url = f"https://aiplatform.googleapis.com/v1/projects/{project_id}/locations/global/publishers/google/models/{original_model}:{gemini_stream}?key={api_key}"
-        else:
-            url = f"https://aiplatform.googleapis.com/v1/publishers/google/models/{original_model}:{gemini_stream}?key={api_key}"
+        url = f"https://aiplatform.googleapis.com/v1/publishers/google/models/{original_model}:{gemini_stream}?key={api_key}"
         headers.pop("Authorization", None)
-    elif "gemini-2.5-flash-lite-preview-06-17" == original_model or \
-    "gemini-2.5-pro-preview-06-05" == original_model:
-        url = "https://aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/google/models/{MODEL_ID}:{stream}".format(
-            LOCATION=await location.next(),
-            PROJECT_ID=project_id,
-            MODEL_ID=original_model,
-            stream=gemini_stream
-        )
     else:
         url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/google/models/{MODEL_ID}:{stream}".format(
             LOCATION=await location.next(),
@@ -573,7 +549,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
                         budget = 32768
                     else: # 128 <= val <= 32768
                         budget = val
-                
+
                 # gemini-2.5-flash-lite: [0] or [512, 24576]
                 elif "gemini-2.5-flash-lite" in original_model:
                     if val > 0 and val < 512:
@@ -589,9 +565,9 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
                         budget = 24576
                     else: # Includes 0 and valid range, and clamps invalid negatives
                         budget = val if val >= 0 else 0
-                
+
                 payload["generationConfig"]["thinkingConfig"] = {
-                    "includeThoughts": True,
+                    "includeThoughts": True if budget else False,
                     "thinkingBudget": budget
                 }
             except ValueError:
@@ -602,9 +578,6 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
                 "includeThoughts": True,
             }
 
-    # if request.model.endswith("-search"):
-    #     payload["tools"] = [search_tool]
-
     if safe_get(provider, "preferences", "post_body_parameter_overrides", default=None):
         for key, value in safe_get(provider, "preferences", "post_body_parameter_overrides", default={}).items():
             if key == request.model:
@@ -1822,21 +1795,98 @@ async def get_dalle_payload(request, engine, provider, api_key=None):
 
     return url, headers, payload
 
+async def get_upload_certificate(client: httpx.AsyncClient, api_key: str, model: str) -> dict:
+    """第一步：获取文件上传凭证"""
+    # print("步骤 1: 正在获取上传凭证...")
+    headers = {"Authorization": f"Bearer {api_key}"}
+    params = {"action": "getPolicy", "model": model}
+    try:
+        response = await client.get("https://dashscope.aliyuncs.com/api/v1/uploads", headers=headers, params=params)
+        response.raise_for_status()  # 如果请求失败则抛出异常
+        cert_data = response.json()
+        # print("凭证获取成功。")
+        return cert_data.get("data")
+    except httpx.HTTPStatusError as e:
+        print(f"获取凭证失败: HTTP {e.response.status_code}")
+        print(f"响应内容: {e.response.text}")
+        return None
+    except Exception as e:
+        print(f"获取凭证时发生未知错误: {e}")
+        return None
+
+from mimetypes import guess_type
+
+async def upload_file_to_oss(client: httpx.AsyncClient, certificate: dict, file: Tuple[str, IOBase, str]) -> str:
+    """第二步：使用凭证将文件内容上传到OSS"""
+    upload_host = certificate.get("upload_host")
+    upload_dir = certificate.get("upload_dir")
+    object_key = f"{upload_dir}/{file[0]}"
+
+    form_data = {
+        "key": object_key,
+        "policy": certificate.get("policy"),
+        "OSSAccessKeyId": certificate.get("oss_access_key_id"),
+        "signature": certificate.get("signature"),
+        "success_action_status": "200",
+        "x-oss-object-acl": certificate.get("x_oss_object_acl"),
+        "x-oss-forbid-overwrite": certificate.get("x_oss_forbid_overwrite"),
+    }
+
+    files = {"file": file}
+
+    try:
+        response = await client.post(upload_host, data=form_data, files=files, timeout=3600)
+        response.raise_for_status()
+        # print("文件上传成功！")
+        oss_url = f"oss://{object_key}"
+        # print(f"文件OSS URL: {oss_url}")
+        return oss_url
+    except httpx.HTTPStatusError as e:
+        print(f"上传文件失败: HTTP {e.response.status_code}")
+        print(f"响应内容: {e.response.text}")
+        return None
+    except Exception as e:
+        print(f"上传文件时发生未知错误: {e}")
+        return None
+
 async def get_whisper_payload(request, engine, provider, api_key=None):
     model_dict = get_model_dict(provider)
     original_model = model_dict[request.model]
-    headers = {
-        # "Content-Type": "multipart/form-data",
-    }
+    headers = {}
     if api_key:
         headers['Authorization'] = f"Bearer {api_key}"
     url = provider['base_url']
     url = BaseAPI(url).audio_transcriptions
 
-    payload = {
-        "model": original_model,
-        "file": request.file,
-    }
+    if "dashscope.aliyuncs.com" in url:
+        client = httpx.AsyncClient()
+        certificate = await get_upload_certificate(client, api_key, original_model)
+        if not certificate:
+            return
+
+        # 步骤 2: 上传文件
+        oss_url = await upload_file_to_oss(client, certificate, request.file)
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+            "X-DashScope-OssResourceResolve": "enable"
+        }
+        payload = {
+            "model": original_model,
+            "input": {
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": [{"audio": oss_url}]
+                    }
+                ]
+            }
+        }
+    else:
+        payload = {
+            "model": original_model,
+            "file": request.file,
+        }
 
     if request.prompt:
         payload["prompt"] = request.prompt
@@ -1906,11 +1956,20 @@ async def get_tts_payload(request, engine, provider, api_key=None):
     url = provider['base_url']
     url = BaseAPI(url).audio_speech
 
-    payload = {
-        "model": original_model,
-        "input": request.input,
-        "voice": request.voice,
-    }
+    if "api.minimaxi.com" in url:
+        payload = {
+            "model": original_model,
+            "text": request.input,
+            "voice_setting": {
+                "voice_id": request.voice
+            }
+        }
+    else:
+        payload = {
+            "model": original_model,
+            "input": request.input,
+            "voice": request.voice,
+        }
 
     if request.response_format:
         payload["response_format"] = request.response_format

{aient-1.1.51 → aient-1.1.52}/src/aient/core/response.py

@@ -666,6 +666,10 @@ async def fetch_response(client, url, headers, payload, engine, model):
 
         yield response_json
 
+    elif "dashscope.aliyuncs.com" in url and "multimodal-generation" in url:
+        response_json = response.json()
+        content = safe_get(response_json, "output", "choices", 0, "message", "content", 0, default=None)
+        yield content
     else:
         response_json = response.json()
         yield response_json

{aient-1.1.51 → aient-1.1.52}/src/aient/core/utils.py

@@ -49,10 +49,16 @@ class BaseAPI:
         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)
         self.image_url: str = urlunparse(parsed_url[:2] + (before_v1 + "images/generations",) + ("",) * 3)
-        self.audio_transcriptions: str = urlunparse(parsed_url[:2] + (before_v1 + "audio/transcriptions",) + ("",) * 3)
+        if parsed_url.hostname == "dashscope.aliyuncs.com":
+            self.audio_transcriptions: str = urlunparse(parsed_url[:2] + ("/api/v1/services/aigc/multimodal-generation/generation",) + ("",) * 3)
+        else:
+            self.audio_transcriptions: str = urlunparse(parsed_url[:2] + (before_v1 + "audio/transcriptions",) + ("",) * 3)
         self.moderations: str = urlunparse(parsed_url[:2] + (before_v1 + "moderations",) + ("",) * 3)
         self.embeddings: str = urlunparse(parsed_url[:2] + (before_v1 + "embeddings",) + ("",) * 3)
-        self.audio_speech: str = urlunparse(parsed_url[:2] + (before_v1 + "audio/speech",) + ("",) * 3)
+        if parsed_url.hostname == "api.minimaxi.com":
+            self.audio_speech: str = urlunparse(parsed_url[:2] + ("v1/t2a_v2",) + ("",) * 3)
+        else:
+            self.audio_speech: str = urlunparse(parsed_url[:2] + (before_v1 + "audio/speech",) + ("",) * 3)
 
         if parsed_url.hostname == "generativelanguage.googleapis.com":
             self.base_url = api_url
@@ -440,8 +446,23 @@ c4 = ThreadSafeCircularList(["us-east5", "us-central1", "europe-west4", "asia-so
 c3h = ThreadSafeCircularList(["us-east5", "us-central1", "europe-west1", "europe-west4"])
 gemini1 = ThreadSafeCircularList(["us-central1", "us-east4", "us-west1", "us-west4", "europe-west1", "europe-west2"])
 gemini2 = ThreadSafeCircularList(["us-central1"])
-gemini2_5_pro_exp = ThreadSafeCircularList(["global"])
-
+# gemini2_5_pro_exp = ThreadSafeCircularList(["global"])
+gemini2_5_pro_exp = ThreadSafeCircularList([
+  "us-central1",
+  "us-east1",
+  "us-east4",
+  "us-east5",
+  "us-south1",
+  "us-west1",
+  "us-west4",
+  "europe-central2",
+  "europe-north1",
+  "europe-southwest1",
+  "europe-west1",
+  "europe-west4",
+  "europe-west8",
+  "europe-west9"
+])
 
 
 # end_of_line = "\n\r\n"

{aient-1.1.51 → aient-1.1.52/src/aient.egg-info}/PKG-INFO

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