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 +3 -17
- aient/core/utils.py +1 -0
- {aient-1.1.40.dist-info → aient-1.1.41.dist-info}/METADATA +1 -1
- {aient-1.1.40.dist-info → aient-1.1.41.dist-info}/RECORD +7 -7
- {aient-1.1.40.dist-info → aient-1.1.41.dist-info}/WHEEL +0 -0
- {aient-1.1.40.dist-info → aient-1.1.41.dist-info}/licenses/LICENSE +0 -0
- {aient-1.1.40.dist-info → aient-1.1.41.dist-info}/top_level.txt +0 -0
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"])
|
@@ -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=
|
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=
|
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.
|
41
|
-
aient-1.1.
|
42
|
-
aient-1.1.
|
43
|
-
aient-1.1.
|
44
|
-
aient-1.1.
|
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
|
File without changes
|
File without changes
|