aient 1.1.40__tar.gz → 1.1.42__tar.gz

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

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

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

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

@@ -10,6 +10,7 @@ from .utils import (
     c3o,
     c3h,
     c35s,
+    c4,
     gemini1,
     gemini2,
     gemini2_5_pro_exp,
@@ -238,15 +239,17 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
         if m:
             try:
                 val = int(m.group(1))
-                if val < 0:
-                    val = 0
-                elif val > 32768 and "gemini-2.5-pro" in original_model:
+                if 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 <= 0:
+                    val = 0
                 elif val > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
+                if val == 0:
+                    payload["generationConfig"].pop("thinkingConfig", None)
             except ValueError:
                 # 如果转换为整数失败，忽略思考预算设置
                 pass
@@ -534,15 +537,17 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
         if m:
             try:
                 val = int(m.group(1))
-                if val < 0:
-                    val = 0
-                elif val > 32768 and "gemini-2.5-pro" in original_model:
+                if 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 <= 0:
+                    val = 0
                 elif val > 24576:
                     val = 24576
                 payload["generationConfig"]["thinkingConfig"]["thinkingBudget"] = val
+                if val == 0:
+                    payload["generationConfig"].pop("thinkingConfig", None)
             except ValueError:
                 # 如果转换为整数失败，忽略思考预算设置
                 pass
@@ -576,6 +581,8 @@ async def get_vertex_claude_payload(request, engine, provider, api_key=None):
         location = c35s
     elif "claude-3-opus" in original_model:
         location = c3o
+    elif "claude-sonnet-4" in original_model or "claude-opus-4" in original_model:
+        location = c4
     elif "claude-3-sonnet" in original_model:
         location = c3s
     elif "claude-3-haiku" in original_model:
@@ -798,23 +805,6 @@ async def get_aws_payload(request, engine, provider, api_key=None):
     # url = f"{base_url}/model/{original_model}/invoke"
     url = f"{base_url}/model/{original_model}/invoke-with-response-stream"
 
-    # if "claude-3-5-sonnet" in original_model or "claude-3-7-sonnet" in original_model:
-    #     location = c35s
-    # elif "claude-3-opus" in original_model:
-    #     location = c3o
-    # elif "claude-3-sonnet" in original_model:
-    #     location = c3s
-    # elif "claude-3-haiku" in original_model:
-    #     location = c3h
-
-    # claude_stream = "streamRawPredict"
-    # url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/anthropic/models/{MODEL}:{stream}".format(
-    #     LOCATION=await location.next(),
-    #     PROJECT_ID=project_id,
-    #     MODEL=original_model,
-    #     stream=claude_stream
-    # )
-
     messages = []
     system_prompt = None
     tool_id = None

{aient-1.1.40 → aient-1.1.42}/src/aient/core/utils.py

@@ -436,6 +436,7 @@ provider_api_circular_list = defaultdict(ThreadSafeCircularList)
 c35s = ThreadSafeCircularList(["us-east5", "europe-west1"])
 c3s = ThreadSafeCircularList(["us-east5", "us-central1", "asia-southeast1"])
 c3o = ThreadSafeCircularList(["us-east5"])
+c4 = ThreadSafeCircularList(["us-east5", "us-central1", "europe-west4", "asia-southeast1"])
 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"])

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

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