aient 1.0.63__tar.gz → 1.0.65__tar.gz

{aient-1.0.63/src/aient.egg-info → aient-1.0.65}/PKG-INFO

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

{aient-1.0.63 → aient-1.0.65}/setup.py

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

{aient-1.0.63 → aient-1.0.65}/src/aient/core/test/test_base_api.py

@@ -1,7 +1,5 @@
 from ..utils import BaseAPI
 
-import os
-
 # GOOGLE_AI_API_KEY = os.environ.get('GOOGLE_AI_API_KEY', None)
 
 # base_api = BaseAPI("https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:streamGenerateContent?key=" + GOOGLE_AI_API_KEY)
@@ -9,9 +7,12 @@ import os
 # print(base_api.chat_url)
 
 base_api = BaseAPI("http://127.0.0.1:8000/v1")
-
 print(base_api.chat_url)
 
+base_api = BaseAPI("http://127.0.0.1:8000/v1/images/generations")
+print(base_api.image_url)
+
 """
 python -m core.test.test_base_api
+python -m aient.src.aient.core.test.test_base_api
 """

{aient-1.0.63 → aient-1.0.65}/src/aient/core/utils.py

@@ -124,6 +124,7 @@ def get_engine(provider, endpoint=None, original_model=""):
 
     return engine, stream
 
+from httpx_socks import AsyncProxyTransport
 def get_proxy(proxy, client_config = {}):
     if proxy:
         # 解析代理URL
@@ -132,7 +133,6 @@ def get_proxy(proxy, client_config = {}):
 
         if scheme == 'socks5':
             try:
-                from httpx_socks import AsyncProxyTransport
                 proxy = proxy.replace('socks5h://', 'socks5://')
                 transport = AsyncProxyTransport.from_url(proxy)
                 client_config["transport"] = transport

{aient-1.0.63 → aient-1.0.65}/src/aient/models/base.py

@@ -37,11 +37,6 @@ class BaseLLM:
         self.presence_penalty: float = presence_penalty
         self.frequency_penalty: float = frequency_penalty
         self.reply_count: int = reply_count
-        self.max_tokens: int = max_tokens or (
-            4096
-            if "gpt-4" or "claude" in engine
-            else 4000
-        )
         self.truncate_limit: int = truncate_limit or (
             198000
             if "claude" in engine

{aient-1.0.63 → aient-1.0.65}/src/aient/models/chatgpt.py

@@ -83,8 +83,6 @@ class chatgpt(BaseLLM):
         # 注册和处理传入的工具
         self._register_tools(tools)
 
-        if self.tokens_usage["default"] > self.max_tokens:
-            raise Exception("System prompt is too long")
 
     def _register_tools(self, tools):
         """动态注册工具函数并更新配置"""
@@ -255,7 +253,6 @@ class chatgpt(BaseLLM):
                 {"role": "system","content": self.system_prompt},
                 {"role": role, "content": prompt}
             ],
-            "max_tokens": kwargs.get("max_tokens", self.max_tokens),
             "stream": True,
             "stream_options": {
                 "include_usage": True
@@ -263,6 +260,9 @@ class chatgpt(BaseLLM):
             "temperature": kwargs.get("temperature", self.temperature)
         }
 
+        if kwargs.get("max_tokens", self.max_tokens):
+            request_data["max_tokens"] = kwargs.get("max_tokens", self.max_tokens)
+
         # 添加工具相关信息
         if kwargs.get("plugins", None):
             self.plugins = kwargs.get("plugins")

{aient-1.0.63 → aient-1.0.65}/src/aient/plugins/__init__.py

@@ -29,4 +29,5 @@ __all__ = [
     'register_tool',
     'register_agent',
     'update_tools_config',
+    'get_function_call_list',
 ] + list(registry.tools.keys())

{aient-1.0.63 → aient-1.0.65}/src/aient/plugins/config.py

@@ -143,10 +143,15 @@ def get_plugins():
     }
 
 # 修改function_call_list定义，使用registry中的工具
-def get_function_call_list():
+def get_function_call_list(tools_list=None):
     function_list = {}
+    if tools_list is None:
+        filtered_tools = registry.tools.keys()
+    else:
+        filtered_tools = [tool.__name__ if callable(tool) else str(tool) for tool in tools_list]
     for tool_name, tool_func in registry.tools.items():
-        function_list[tool_name] = function_to_json(tool_func)
+        if tool_name in filtered_tools:
+            function_list[tool_name] = function_to_json(tool_func)
     return function_list
 
 def get_claude_tools_list():

{aient-1.0.63 → aient-1.0.65/src/aient.egg-info}/PKG-INFO

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