LightAgent 0.4.0__tar.gz → 0.4.2__tar.gz

{lightagent-0.4.0 → lightagent-0.4.2}/LightAgent/la_core.py

@@ -3,8 +3,8 @@
 
 """
 作者: [weego/WXAI-Team]
-版本: 0.4.0
-最后更新: 2025-06-12
+版本: 0.4.2
+最后更新: 2025-07-10
 """
 
 import asyncio
@@ -30,8 +30,7 @@ from mcp.client.sse import sse_client
 from mcp.client.stdio import stdio_client
 from openai.types.chat import ChatCompletionChunk
 
-
-__version__ = "0.4.0"  # 你可以根据需要设置版本号
+__version__ = "0.4.2"  # 你可以根据需要设置版本号
 
 
 # openai.langfuse_auth_check()
@@ -44,6 +43,7 @@ class MemoryProtocol(Protocol):
     def retrieve(self, query: str, user_id: str) -> List[Any]:
         ...
 
+
 class ToolRegistry:
     """集中管理工具注册表，避免全局变量"""
 
@@ -421,8 +421,9 @@ class MCPClientManager:
             if field not in arguments:
                 raise ValueError(f"缺少必要参数: {field}")
 
+
 class LightAgent:
-    __version__ = "0.4.0"  # 将版本号放在类中
+    __version__ = "0.4.2"  # 将版本号放在类中
 
     def __init__(
             self,
@@ -545,6 +546,7 @@ class LightAgent:
 
         # 初始化客户端
         self._initialize_clients(tracetools, tot_api_key, tot_base_url, tot_model)
+        self.chat_params = {}  # history 存储器
 
     def _initialize_clients(self, tracetools, tot_api_key, tot_base_url, tot_model):
         """初始化 OpenAI 客户端"""
@@ -574,6 +576,12 @@ class LightAgent:
                     api_key=tot_api_key or self.api_key
                 )
 
+    def get_history(self) -> List[Dict[str, Any]]:
+        """
+        获取对话的history的描述（OpenAI 格式）
+        """
+        return deepcopy(self.chat_params['messages'])
+
     def get_tools(self) -> List[Dict[str, Any]]:
         """
         获取所有工具的描述（OpenAI 格式）
@@ -674,25 +682,31 @@ class LightAgent:
             self.logger.log("DEBUG", "tree_of_thought", {"response": tot_response, "active_tools": active_tools})
 
         # 准备API参数
-        params = {
+        self.chat_params =  {
             "model": self.model,
             "messages": [{"role": "system", "content": system_prompt}] + history + [{"role": "user", "content": query}],
             "stream": stream
         }
 
+        # 添加参数
+        if metadata:
+            for key, value in metadata.items():
+                self.chat_params[key] = value
+
         # 添加工具
         tools = active_tools or self.tool_registry.get_tools()
         if tools:
-            params["tools"] = tools
-            params["tool_choice"] = "auto"
+            self.chat_params["tools"] = tools
+            self.chat_params["tool_choice"] = "auto"
 
         # 添加跟踪会话
         if hasattr(self, 'tracetools') and self.tracetools:
-            params["session_id"] = traceid
+            self.chat_params["session_id"] = traceid
 
         # 调用模型
-        response = self.client.chat.completions.create(**params)
-        return self._core_run_logic(response, params, stream, max_retry)
+        self.logger.log("DEBUG", "first_request_params", {"params": self.chat_params})
+        response = self.client.chat.completions.create(**self.chat_params)
+        return self._core_run_logic(response, stream, max_retry)
 
     def _add_memory_context(self, query: str, user_id: str) -> str:
         """添加记忆上下文"""
@@ -717,14 +731,14 @@ class LightAgent:
 
         return f"{context}\n##用户提问：\n{query}" if context else query
 
-    def _core_run_logic(self, response, params, stream, max_retry) -> Union[Generator[str, None, None], str]:
+    def _core_run_logic(self, response, stream, max_retry) -> Union[Generator[str, None, None], str]:
         """核心运行逻辑"""
         if stream:
-            return self._run_stream_logic(response, params, max_retry)
+            return self._run_stream_logic(response, max_retry)
         else:
-            return self._run_non_stream_logic(response, params, max_retry)
+            return self._run_non_stream_logic(response, max_retry)
 
-    def _run_non_stream_logic(self, response, params, max_retry) -> Union[str, None]:
+    def _run_non_stream_logic(self, response, max_retry) -> Union[str, None]:
         """
         非流式处理逻辑。
         """
@@ -793,7 +807,8 @@ class LightAgent:
                                 pass  # 如果不是 JSON 字符串，保持原样
                         single_tool_response = combined_response  # 处理单个工具的方法
 
-                    self.logger.log("INFO", "non_stream single_tool_response", {"single_tool_response": single_tool_response})
+                    self.logger.log("INFO", "non_stream single_tool_response",
+                                    {"single_tool_response": single_tool_response})
 
                     # 将单个工具的响应结果添加到列表中
                     tool_responses.append(single_tool_response)
@@ -804,13 +819,13 @@ class LightAgent:
                 combined_tool_response = "\n".join(tool_responses)
 
                 # 将工具调用和响应添加到消息列表中
-                params["messages"].append(
+                self.chat_params["messages"].append(
                     {
                         "role": "assistant",
                         "content": f"使用工具： \n {json.dumps([tool_call.function.model_dump() for tool_call in tool_calls], ensure_ascii=False)}\n",
                     }
                 )
-                params["messages"].append(
+                self.chat_params["messages"].append(
                     {
                         "role": "user",
                         "content": f"工具响应内容：\n {combined_tool_response} \n 请给出下一步输出",
@@ -825,11 +840,11 @@ class LightAgent:
             # 更新响应
             if function_call_name == 'finish':
                 return  # 如果最后调用了finish工具，则结束生成器
-            # print("params:",params)
-            self.logger.log("DEBUG", "non_stream chat-completions params", {"params": params})
+            # print("params:",self.chat_params)
+            self.logger.log("DEBUG", "non_stream chat-completions params", {"params": self.chat_params})
 
             try:
-                response = self.client.chat.completions.create(**params)
+                response = self.client.chat.completions.create(**self.chat_params)
             except Exception as e:
                 print(f"An error occurred: {e}")
 
@@ -837,7 +852,7 @@ class LightAgent:
         self.logger.log("ERROR", "max_retry_reached", {"message": "Failed to generate a valid response."})
         return "Failed to generate a valid response."
 
-    def _run_stream_logic(self, response, params, max_retry) -> Generator[str, None, None]:
+    def _run_stream_logic(self, response, max_retry) -> Generator[str, None, None]:
         """流式处理逻辑"""
         for _ in range(max_retry):
             # 初始化变量
@@ -950,7 +965,8 @@ class LightAgent:
                                                     "title": tool_title,
                                                     "output": chunk,
                                                 }
-                                                self.logger.log("DEBUG", "stream tool_output", {"tool_output": tool_output})
+                                                self.logger.log("DEBUG", "stream tool_output",
+                                                                {"tool_output": tool_output})
                                                 yield tool_output
                                             # 将工具的调用信息推送给开发者
                                             if tool_name == 'finish':
@@ -972,7 +988,7 @@ class LightAgent:
 
                                     # 记录工具响应
                                     self.logger.log("INFO", "stream single_tool_response",
-                                             {"single_tool_response": single_tool_response})
+                                                    {"single_tool_response": single_tool_response})
 
                                     # 将单个工具的响应结果保存到列表中
                                     tool_responses.append(single_tool_response)
@@ -984,7 +1000,8 @@ class LightAgent:
 
                             except json.JSONDecodeError as e:
                                 error_msg = f"JSON解析错误: {str(e)}\n参数: {arguments}"
-                                self.logger.log("ERROR", "json_decode_error", {"tool": tool_name, "title": tool_title, "error": error_msg})
+                                self.logger.log("ERROR", "json_decode_error",
+                                                {"tool": tool_name, "title": tool_title, "error": error_msg})
                                 tool_responses.append(error_msg)
                                 yield {"name": tool_name, "title": tool_title, "error": error_msg}
 
@@ -1009,13 +1026,13 @@ class LightAgent:
                         ensure_ascii=False)
 
                     # 添加工具调用和响应到消息历史
-                    params["messages"].append(
+                    self.chat_params["messages"].append(
                         {
                             "role": "assistant",
                             "content": f"使用工具： \n {tool_str}\n"
                         }
                     )
-                    params["messages"].append(
+                    self.chat_params["messages"].append(
                         {
                             "role": "user",
                             "content": combined_tool_response,
@@ -1023,15 +1040,14 @@ class LightAgent:
                     )
 
                     # 创建新的响应流
-                    self.logger.log("DEBUG", "stream next_request_params", {"params": params})
-                    response = self.client.chat.completions.create(**params)
+                    self.logger.log("DEBUG", "stream next_request_params", {"params": self.chat_params})
+                    response = self.client.chat.completions.create(**self.chat_params)
                     break
 
         # 重试次数用尽
         self.logger.log("ERROR", "max_retry_reached", {"message": "Failed to generate a valid response."})
         yield "Failed to generate a valid response."
 
-
     def _handle_task_transfer(
             self,
             query: str,

{lightagent-0.4.0 → lightagent-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LightAgent
-Version: 0.4.0
+Version: 0.4.2
 Summary: LightAgent: Lightweight AI agent framework with memory, tools & tree-of-thought. Supports multi-agent collaboration, self-learning, and major LLMs (OpenAI/DeepSeek/Qwen). Open-source with MCP/SSE protocol integration.
 Home-page: https://github.com/wxai-space/LightAgent
 License: Apache 2.0

{lightagent-0.4.0 → lightagent-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "LightAgent"
-version = "0.4.0"
+version = "0.4.2"
 description = "LightAgent: Lightweight AI agent framework with memory, tools & tree-of-thought. Supports multi-agent collaboration, self-learning, and major LLMs (OpenAI/DeepSeek/Qwen). Open-source with MCP/SSE protocol integration."
 authors = ["caiweige <caiweige@qq.com>"]
 license = "Apache 2.0"