aient 1.1.40__py3-none-any.whl → 1.1.41__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
aient/core/request.py CHANGED
@@ -10,6 +10,7 @@ from .utils import (
10
10
  c3o,
11
11
  c3h,
12
12
  c35s,
13
+ c4,
13
14
  gemini1,
14
15
  gemini2,
15
16
  gemini2_5_pro_exp,
@@ -576,6 +577,8 @@ async def get_vertex_claude_payload(request, engine, provider, api_key=None):
576
577
  location = c35s
577
578
  elif "claude-3-opus" in original_model:
578
579
  location = c3o
580
+ elif "claude-sonnet-4" in original_model or "claude-opus-4" in original_model:
581
+ location = c4
579
582
  elif "claude-3-sonnet" in original_model:
580
583
  location = c3s
581
584
  elif "claude-3-haiku" in original_model:
@@ -798,23 +801,6 @@ async def get_aws_payload(request, engine, provider, api_key=None):
798
801
  # url = f"{base_url}/model/{original_model}/invoke"
799
802
  url = f"{base_url}/model/{original_model}/invoke-with-response-stream"
800
803
 
801
- # if "claude-3-5-sonnet" in original_model or "claude-3-7-sonnet" in original_model:
802
- # location = c35s
803
- # elif "claude-3-opus" in original_model:
804
- # location = c3o
805
- # elif "claude-3-sonnet" in original_model:
806
- # location = c3s
807
- # elif "claude-3-haiku" in original_model:
808
- # location = c3h
809
-
810
- # claude_stream = "streamRawPredict"
811
- # url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/anthropic/models/{MODEL}:{stream}".format(
812
- # LOCATION=await location.next(),
813
- # PROJECT_ID=project_id,
814
- # MODEL=original_model,
815
- # stream=claude_stream
816
- # )
817
-
818
804
  messages = []
819
805
  system_prompt = None
820
806
  tool_id = None
aient/core/utils.py CHANGED
@@ -436,6 +436,7 @@ provider_api_circular_list = defaultdict(ThreadSafeCircularList)
436
436
  c35s = ThreadSafeCircularList(["us-east5", "europe-west1"])
437
437
  c3s = ThreadSafeCircularList(["us-east5", "us-central1", "asia-southeast1"])
438
438
  c3o = ThreadSafeCircularList(["us-east5"])
439
+ c4 = ThreadSafeCircularList(["us-east5", "us-central1", "europe-west4", "asia-southeast1"])
439
440
  c3h = ThreadSafeCircularList(["us-east5", "us-central1", "europe-west1", "europe-west4"])
440
441
  gemini1 = ThreadSafeCircularList(["us-central1", "us-east4", "us-west1", "us-west4", "europe-west1", "europe-west2"])
441
442
  gemini2 = ThreadSafeCircularList(["us-central1"])
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aient
3
- Version: 1.1.40
3
+ Version: 1.1.41
4
4
  Summary: Aient: The Awakening of Agent.
5
5
  Description-Content-Type: text/markdown
6
6
  License-File: LICENSE
@@ -4,9 +4,9 @@ aient/core/.gitignore,sha256=5JRRlYYsqt_yt6iFvvzhbqh2FTUQMqwo6WwIuFzlGR8,13
4
4
  aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
5
5
  aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
6
6
  aient/core/models.py,sha256=d4MISNezTSe0ls0-fjuToI2SoT-sk5fWqAJuKVinIlo,7502
7
- aient/core/request.py,sha256=gXHdx61emExUTZbUyLng_AYmlfyKl3uOcCmlAY5eN98,72330
7
+ aient/core/request.py,sha256=1iY-GUgbGCt0gXpO91QDEERIfkptI_anCHg9ElDvVVc,71765
8
8
  aient/core/response.py,sha256=mAVsCnNhWY09DXNe0lyPUJq-1ljtGjC67Az-Uh7ozIw,35166
9
- aient/core/utils.py,sha256=fhI5wBxr01lVEp8nMfjG9dQ859AE-VdrWyb9suLzzqM,27400
9
+ aient/core/utils.py,sha256=NcXdb8zBN0GE01OGaUzg8U34RaraoFf2MaLDDGFvvC4,27492
10
10
  aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
11
11
  aient/core/test/test_geminimask.py,sha256=HFX8jDbNg_FjjgPNxfYaR-0-roUrOO-ND-FVsuxSoiw,13254
12
12
  aient/core/test/test_image.py,sha256=_T4peNGdXKBHHxyQNx12u-NTyFE8TlYI6NvvagsG2LE,319
@@ -37,8 +37,8 @@ aient/plugins/write_file.py,sha256=hExFLuoNPtjYxJI3pVbofZRpokvUabpXdEkd3mZJPPc,3
37
37
  aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
38
38
  aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
39
39
  aient/utils/scripts.py,sha256=h7EA2xBydUF_wdZLsPgjCq4Egdycx1gf2qrdrm0I7y0,40909
40
- aient-1.1.40.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
41
- aient-1.1.40.dist-info/METADATA,sha256=hD9dZLHXJ2D7cZ0jM88HNw0lNcgMnaeXHyIHFCYN_6I,4968
42
- aient-1.1.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
43
- aient-1.1.40.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
44
- aient-1.1.40.dist-info/RECORD,,
40
+ aient-1.1.41.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
41
+ aient-1.1.41.dist-info/METADATA,sha256=hMAJ7yUTFHx9-QqfRUy1Vu9BBRZGHKUco1m3nGdQC28,4968
42
+ aient-1.1.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
43
+ aient-1.1.41.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
44
+ aient-1.1.41.dist-info/RECORD,,
File without changes