beswarm 0.1.65__tar.gz → 0.1.66__tar.gz

{beswarm-0.1.65 → beswarm-0.1.66}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.65
+Version: 0.1.66
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.65 → beswarm-0.1.66}/beswarm/aient/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.1.18",
+    version="1.1.19",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{beswarm-0.1.65 → beswarm-0.1.66}/beswarm/aient/src/aient/core/request.py

@@ -12,6 +12,7 @@ from .utils import (
     c35s,
     gemini1,
     gemini2,
+    gemini2_5_pro_exp,
     BaseAPI,
     safe_get,
     get_engine,
@@ -112,7 +113,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
     if system_prompt.strip():
         systemInstruction = {"parts": [{"text": system_prompt}]}
 
-    if any(off_model in original_model for off_model in gemini_max_token_65k_models) or original_model == "gemini-2.0-flash-preview-image-generation":
+    if any(off_model in original_model for off_model in gemini_max_token_65k_models) or original_model.endswith("-image-generation"):
         safety_settings = "OFF"
     else:
         safety_settings = "BLOCK_NONE"
@@ -167,6 +168,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
         'response_format',
         'stream_options',
         'prompt',
+        'size',
     ]
     generation_config = {}
 
@@ -221,7 +223,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
         else:
             payload["generationConfig"]["maxOutputTokens"] = 8192
 
-        if original_model == "gemini-2.0-flash-preview-image-generation":
+        if original_model.endswith("-image-generation"):
             payload["generationConfig"]["response_modalities"] = [
                 "Text",
                 "Image",
@@ -336,10 +338,11 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
     pro_models = ["gemini-2.5", "gemini-2.0"]
     if any(pro_model in original_model for pro_model in pro_models):
         location = gemini2
-        # search_tool = {"googleSearch": {}}
     else:
         location = gemini1
-        # search_tool = {"googleSearchRetrieval": {}}
+
+    if "gemini-2.5-pro-exp-03-25" == 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(
@@ -351,6 +354,13 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
     elif api_key is not None and api_key[2] == ".":
         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-pro-exp-03-25" == 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(),
@@ -464,6 +474,8 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
         'logprobs',
         'top_logprobs',
         'stream_options',
+        'prompt',
+        'size',
     ]
     generation_config = {}
 

{beswarm-0.1.65 → beswarm-0.1.66}/beswarm/aient/src/aient/core/utils.py

@@ -437,6 +437,7 @@ c3o = ThreadSafeCircularList(["us-east5"])
 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"])
 
 
 

{beswarm-0.1.65 → beswarm-0.1.66}/beswarm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.65
+Version: 0.1.66
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.65 → beswarm-0.1.66}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.1.65"
+version = "0.1.66"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"