aient 1.1.0__py3-none-any.whl → 1.1.2__py3-none-any.whl

aient/core/request.py

@@ -459,10 +459,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
             payload["generationConfig"]["max_output_tokens"] = 8192
 
     if request.model.endswith("-search"):
-        if "tools" not in payload:
-            payload["tools"] = [search_tool]
-        else:
-            payload["tools"].append(search_tool)
+        payload["tools"] = [search_tool]
 
     return url, headers, payload
 

aient/core/response.py

@@ -67,7 +67,7 @@ async def fetch_gemini_response_stream(client, url, headers, payload, model):
                 if (line and '"parts": [' in line or parts_json != "") and is_finish == False:
                     parts_json += line
                 if parts_json != "" and line and '],' == line.strip():
-                    parts_json =  "{" + parts_json.strip().rstrip(",} ]}") + "}]}"
+                    parts_json =  "{" + parts_json.strip().replace(',      "groundingMetadata": {}', "").rstrip(",} ]}").lstrip("{") + "}]}"
                     try:
                         json_data = json.loads(parts_json)
 

{aient-1.1.0.dist-info → aient-1.1.2.dist-info}/METADATA

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

{aient-1.1.0.dist-info → aient-1.1.2.dist-info}/RECORD

@@ -4,8 +4,8 @@ aient/core/.gitignore,sha256=5JRRlYYsqt_yt6iFvvzhbqh2FTUQMqwo6WwIuFzlGR8,13
 aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 aient/core/models.py,sha256=_1wYZg_n9kb2A3C8xCboyqleH2iHc9scwOvtx9DPeok,7582
-aient/core/request.py,sha256=eGoGL8muYGvPbamUjcnyI8qH28uZBSG4Sc6Eko5LeJw,61653
-aient/core/response.py,sha256=8bS1nAoP6QOMDeDvJvZDVAt34kZ1DpWBI3PUGyza0ZU,31447
+aient/core/request.py,sha256=8ghtI3gU7QMiw1voINWZ-PMsHu4IYjV6ncO6a9Qe9sI,61551
+aient/core/response.py,sha256=rJnt4tOyJXF2pT9_4VdU7aG3Ac-k2MiCAp9nANVyFqI,31505
 aient/core/utils.py,sha256=CAFqWzICaKVysH9GLHBcp-VeOShisLjWGhEsh6-beWo,26365
 aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
 aient/core/test/test_geminimask.py,sha256=HFX8jDbNg_FjjgPNxfYaR-0-roUrOO-ND-FVsuxSoiw,13254
@@ -37,8 +37,8 @@ aient/prompt/agent.py,sha256=6f5ZB66Rb8y0iQScHMRhvXZ1qMM3YsKpCBPCTAAw2rg,24917
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 aient/utils/scripts.py,sha256=NXmTxcZqHoRv3S13isLsv7kvqktXnA5ej7uMsxCJUe0,26656
-aient-1.1.0.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.0.dist-info/METADATA,sha256=T5DmF0T811yIKF3FF4_zrUqG_AdQbntaKagrQ-thfn4,4944
-aient-1.1.0.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-aient-1.1.0.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.0.dist-info/RECORD,,
+aient-1.1.2.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.1.2.dist-info/METADATA,sha256=LjjmPbYBhPxrTuZyewleUpS6M_2IyVOKIINcJIh2kyY,4944
+aient-1.1.2.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+aient-1.1.2.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.1.2.dist-info/RECORD,,