LightAgent 0.2.8__tar.gz → 0.2.81__tar.gz

{lightagent-0.2.8 → lightagent-0.2.81}/LightAgent/la_core.py

@@ -11,6 +11,7 @@ import logging
 import os
 import httpx
 import importlib
+from openai.types.chat import ChatCompletionChunk
 
 # 全局工具注册表
 _FUNCTION_MAPPINGS = {}  # 工具名称 -> 工具函数
@@ -18,7 +19,7 @@ _FUNCTION_INFO = {}   # 工具名称 -> 工具info信息
 _OPENAI_FUNCTION_SCHEMAS = []  # OpenAI 格式的工具描述
 _PROMPT_FUNCTION_SCHEMAS = []  # prompt 格式的工具描述
 
-__version__ = "0.2.8"  # 你可以根据需要设置版本号
+__version__ = "0.2.81"  # 你可以根据需要设置版本号
 
 
 def register_tool_manually(tools: List[Union[str, Callable]]) -> bool:
@@ -653,14 +654,17 @@ class LightAgent:
                                         combined_response = ""
                                         for chunk in tool_response:
                                             # 将工具返回的数据继续流出
-                                            tool_output = {
-                                                "name": tool_call["name"],
-                                                "title": _FUNCTION_INFO.get(tool_call["name"], {}).get(
-                                                    'tool_title') or '',
-                                                "output": chunk,
-                                            }
-                                            self.log("INFO", "tool_call", {"tool_output": tool_output})
-                                            yield tool_output
+                                            if isinstance(chunk, ChatCompletionChunk):
+                                                yield chunk
+                                            else:
+                                                tool_output = {
+                                                    "name": tool_call["name"],
+                                                    "title": _FUNCTION_INFO.get(tool_call["name"], {}).get(
+                                                        'tool_title') or '',
+                                                    "output": chunk,
+                                                }
+                                                self.log("INFO", "tool_call", {"tool_output": tool_output})
+                                                yield tool_output
                                             # 将工具的调用信息推送给开发者
                                             if function_call_name == 'finish':
                                                 content = chunk.choices[0].delta.content or ""
@@ -1087,4 +1091,4 @@ class LightSwarm:
 if __name__ == "__main__":
     # Example of registering and using a tool
     print("This is LightAgent")
-    # print(dispatch_tool("example_tool", {"param1": "test"}))
+    # print(dispatch_tool("example_tool", {"param1": "test"}))

{lightagent-0.2.8 → lightagent-0.2.81}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LightAgent
-Version: 0.2.8
+Version: 0.2.81
 Summary: **LightAgent** is an extremely lightweight active Agentic Framework with memory (`mem0`), tools (`Tools`), and a Tree of Thought (`ToT`). It supports swarm-like multi-agent collaboration, automated tool generation, and agent assessment, with underlying model support for OpenAI, Zhipu ChatGLM, Baichuan Large Model, DeepSeek R1, Qwen series large models, and more. At the same time, LightAgent supports OpenAI streaming format API service output, seamlessly integrating with major mainstream chat frameworks. 🌟
 Home-page: https://github.com/wxai-space/LightAgent
 License: Apache 2.0
@@ -356,6 +356,7 @@ def get_weather(
 # Define tool information inside the function
 get_weather.tool_info = {
     "tool_name": "get_weather",
+    "tool_title": "get weather",
     "tool_description": "Get current weather information for the specified city.",
     "tool_params": [
         {"name": "city_name", "description": "The name of the city to query", "type": "string", "required": True},
@@ -378,6 +379,7 @@ def search_news(
 # Define tool information inside the function
 search_news.tool_info = {
     "tool_name": "search_news",
+    "tool_title": "search news",
     "tool_description": "Search news based on keywords.",
     "tool_params": [
         {"name": "keyword", "description": "Search keyword", "type": "string", "required": True},
@@ -1077,6 +1079,7 @@ def get_weather(
 # 在函数内部定义工具信息
 get_weather.tool_info = {
     "tool_name": "get_weather",
+    "tool_name": "获取天气",
     "tool_description": "获取指定城市的当前天气信息",
     "tool_params": [
         {"name": "city_name", "description": "要查询的城市名称", "type": "string", "required": True},
@@ -1099,6 +1102,7 @@ def search_news(
 # 在函数内部定义工具信息
 search_news.tool_info = {
     "tool_name": "search_news",
+    "tool_name": "联网搜索",
     "tool_description": "根据关键词搜索新闻",
     "tool_params": [
         {"name": "keyword", "description": "搜索关键词", "type": "string", "required": True},

{lightagent-0.2.8 → lightagent-0.2.81}/README.md

@@ -335,6 +335,7 @@ def get_weather(
 # Define tool information inside the function
 get_weather.tool_info = {
     "tool_name": "get_weather",
+    "tool_title": "get weather",
     "tool_description": "Get current weather information for the specified city.",
     "tool_params": [
         {"name": "city_name", "description": "The name of the city to query", "type": "string", "required": True},
@@ -357,6 +358,7 @@ def search_news(
 # Define tool information inside the function
 search_news.tool_info = {
     "tool_name": "search_news",
+    "tool_title": "search news",
     "tool_description": "Search news based on keywords.",
     "tool_params": [
         {"name": "keyword", "description": "Search keyword", "type": "string", "required": True},

{lightagent-0.2.8 → lightagent-0.2.81}/README.zh-CN.md

@@ -339,6 +339,7 @@ def get_weather(
 # 在函数内部定义工具信息
 get_weather.tool_info = {
     "tool_name": "get_weather",
+    "tool_name": "获取天气",
     "tool_description": "获取指定城市的当前天气信息",
     "tool_params": [
         {"name": "city_name", "description": "要查询的城市名称", "type": "string", "required": True},
@@ -361,6 +362,7 @@ def search_news(
 # 在函数内部定义工具信息
 search_news.tool_info = {
     "tool_name": "search_news",
+    "tool_name": "联网搜索",
     "tool_description": "根据关键词搜索新闻",
     "tool_params": [
         {"name": "keyword", "description": "搜索关键词", "type": "string", "required": True},

{lightagent-0.2.8 → lightagent-0.2.81}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "LightAgent"
-version = "0.2.8"
+version = "0.2.81"
 description = "**LightAgent** is an extremely lightweight active Agentic Framework with memory (`mem0`), tools (`Tools`), and a Tree of Thought (`ToT`). It supports swarm-like multi-agent collaboration, automated tool generation, and agent assessment, with underlying model support for OpenAI, Zhipu ChatGLM, Baichuan Large Model, DeepSeek R1, Qwen series large models, and more. At the same time, LightAgent supports OpenAI streaming format API service output, seamlessly integrating with major mainstream chat frameworks. 🌟"
 authors = ["caiweige <caiweige@qq.com>"]
 license = "Apache 2.0"