aient 1.1.13__py3-none-any.whl → 1.1.15__py3-none-any.whl

aient/core/request.py

@@ -241,7 +241,7 @@ async def get_gemini_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
                 payload[key] = value
 
     return url, headers, payload
@@ -511,7 +511,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
                 payload[key] = value
 
     return url, headers, payload
@@ -935,6 +935,9 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
         headers['Authorization'] = f"Bearer {api_key}"
 
     url = provider['base_url']
+    if "openrouter.ai" in url:
+        headers['HTTP-Referer'] = "https://github.com/yym68686/uni-api"
+        headers['X-Title'] = "Uni API"
 
     messages = []
     for msg in request.messages:
@@ -1055,7 +1058,7 @@ async def get_gpt_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
                 payload[key] = value
 
     return url, headers, payload
@@ -1153,7 +1156,7 @@ async def get_azure_payload(request, engine, provider, api_key=None):
             if key == request.model:
                 for k, v in value.items():
                     payload[k] = v
-            elif all(_model not in request.model for _model in ["gemini", "gpt", "claude"]):
+            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
                 payload[key] = value
 
     return url, headers, payload
@@ -1168,6 +1171,9 @@ async def get_openrouter_payload(request, engine, provider, api_key=None):
         headers['Authorization'] = f"Bearer {api_key}"
 
     url = provider['base_url']
+    if "openrouter.ai" in url:
+        headers['HTTP-Referer'] = "https://github.com/yym68686/uni-api"
+        headers['X-Title'] = "Uni API"
 
     messages = []
     for msg in request.messages:
@@ -1215,6 +1221,14 @@ async def get_openrouter_payload(request, engine, provider, api_key=None):
         if field not in miss_fields and value is not None:
             payload[field] = value
 
+    if safe_get(provider, "preferences", "post_body_parameter_overrides", default=None):
+        for key, value in safe_get(provider, "preferences", "post_body_parameter_overrides", default={}).items():
+            if key == request.model:
+                for k, v in value.items():
+                    payload[k] = v
+            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+                payload[key] = value
+
     return url, headers, payload
 
 async def get_cohere_payload(request, engine, provider, api_key=None):
@@ -1496,9 +1510,10 @@ async def get_claude_payload(request, engine, provider, api_key=None):
     payload = {
         "model": original_model,
         "messages": messages,
-        "system": system_prompt or "You are Claude, a large language model trained by Anthropic.",
         "max_tokens": max_tokens,
     }
+    if system_prompt:
+        payload["system"] = system_prompt
 
     if request.max_tokens:
         payload["max_tokens"] = int(request.max_tokens)
@@ -1546,7 +1561,7 @@ async def get_claude_payload(request, engine, provider, api_key=None):
         payload.pop("tools", None)
         payload.pop("tool_choice", None)
 
-    if "think" in request.model:
+    if "think" in request.model.lower():
         payload["thinking"] = {
             "budget_tokens": 4096,
             "type": "enabled"
@@ -1572,6 +1587,14 @@ async def get_claude_payload(request, engine, provider, api_key=None):
         payload.pop("top_k", None)
     # print("payload", json.dumps(payload, indent=2, ensure_ascii=False))
 
+    if safe_get(provider, "preferences", "post_body_parameter_overrides", default=None):
+        for key, value in safe_get(provider, "preferences", "post_body_parameter_overrides", default={}).items():
+            if key == request.model:
+                for k, v in value.items():
+                    payload[k] = v
+            elif all(_model not in request.model.lower() for _model in ["gemini", "gpt", "claude"]):
+                payload[key] = value
+
     return url, headers, payload
 
 async def get_dalle_payload(request, engine, provider, api_key=None):

aient/plugins/excute_command.py

@@ -172,7 +172,8 @@ def excute_command(command):
             process.stderr.close()
 
         new_output_lines = []
-        output_lines = "".join(output_lines).strip().replace("\\u001b[A", "").replace("\\r", "\r").replace("\\\\", "").replace("\\n", "\n").replace("\r", "+++").replace("\n", "+++")
+        output_lines = "".join(output_lines).strip().replace("\\r", "\r").replace("\\\\", "").replace("\\n", "\n").replace("\r", "+++").replace("\n", "+++")
+        output_lines = re.sub(r'\\u001b\[[0-9;]*[a-zA-Z]', '', output_lines)
         for line in output_lines.split("+++"):
             if line.strip() == "":
                 continue

aient/utils/scripts.py

@@ -667,7 +667,6 @@ def convert_functions_to_xml(functions_list):
 
 if __name__ == "__main__":
 
-    # 运行本文件：python -m beswarm.aient.src.aient.utils.scripts
     os.system("clear")
     test_xml = """
 ✅ 好的，我现在读取 `README.md` 文件。
@@ -742,4 +741,6 @@ if __name__ == "__main__":
 请提供前两个 `excute_command` 的执行结果。
 """
 
-    print(parse_function_xml(test_xml))
+    print(parse_function_xml(test_xml))
+
+# 运行本文件：python -m beswarm.aient.src.aient.utils.scripts

{aient-1.1.13.dist-info → aient-1.1.15.dist-info}/METADATA

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

{aient-1.1.13.dist-info → aient-1.1.15.dist-info}/RECORD

@@ -4,7 +4,7 @@ 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=kF-HLi1I2k_G5r153ZHuiGH8_NmpTlFMfK0_myB28YQ,7366
-aient/core/request.py,sha256=AmTnQ_Ri_ACRxDsWmPhhD6e79hNfwLxbsyBnpbAnmNA,64490
+aient/core/request.py,sha256=2u_vPJjoIqRsKOPOkPilGzM4iZXUpsx5xtouQy87Z4E,65693
 aient/core/response.py,sha256=Z0Bjl_QvpUguyky1LIcsVks4BKKqT0eYEpDmKa_cwpQ,31978
 aient/core/utils.py,sha256=-naFCv8V-qhnqvDUd8BNbW1HR9CVAPxISrXoAz464Qg,26580
 aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
@@ -23,7 +23,7 @@ aient/models/vertex.py,sha256=qVD5l1Q538xXUPulxG4nmDjXE1VoV4yuAkTCpIeJVw0,16795
 aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
 aient/plugins/config.py,sha256=Vp6CG9ocdC_FAlCMEGtKj45xamir76DFxdJVvURNtog,6539
-aient/plugins/excute_command.py,sha256=A3WmfZboEikU1EHvtMWhBv-xHxCyMxbDddQ982I_8wE,10482
+aient/plugins/excute_command.py,sha256=huQSbNbeImV8BUIsQKE13BIhCAMr7aYRyXO4saE1dTI,10534
 aient/plugins/get_time.py,sha256=Ih5XIW5SDAIhrZ9W4Qe5Hs1k4ieKPUc_LAd6ySNyqZk,654
 aient/plugins/image.py,sha256=ZElCIaZznE06TN9xW3DrSukS7U3A5_cjk1Jge4NzPxw,2072
 aient/plugins/list_directory.py,sha256=JZVuImecMSfEv6jLqii-0uQJ1UCsrpMNmYlwW3PEDg4,1374
@@ -37,9 +37,9 @@ aient/prompt/__init__.py,sha256=GBtn6-JDT8KHFCcuPpfSNE_aGddg5p4FEyMCy4BfwGs,20
 aient/prompt/agent.py,sha256=y2GETN6ScC5yQVs75VFfzm4YUWzblbqLYz0Sy6JnPRw,24950
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
-aient/utils/scripts.py,sha256=wutPtgbs-WXo5AACLpnCJaRQBOSKXWNnsf2grbYDzyQ,29098
-aient-1.1.13.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.13.dist-info/METADATA,sha256=uguXOGwPo-gVDnS50EHIjb9nm6KtITwdbNXH9a3IQzA,4968
-aient-1.1.13.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-aient-1.1.13.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.13.dist-info/RECORD,,
+aient/utils/scripts.py,sha256=LD8adnfuRrJoY2tWKseXOPJXaxbrUmz4czsnUvHswNY,29096
+aient-1.1.15.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.1.15.dist-info/METADATA,sha256=MTwSePn_DF0VVK3i0k9yW8E6Y2dOEn76NwsF7BGx_E0,4968
+aient-1.1.15.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+aient-1.1.15.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.1.15.dist-info/RECORD,,