From 98bfa82b246c604b510447ccc26f6f7e8ff7efcb Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Fri, 28 Mar 2025 11:40:46 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=20Agent=20=E9=85=8D=E7=BD=AE?=
 =?UTF-8?q?=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/agents/chatbot/configuration.py |  6 ------
 src/agents/chatbot/graph.py         | 24 ++++++++++++------------
 src/agents/registry.py              | 27 +++++++++++++++++++++------
 3 files changed, 33 insertions(+), 24 deletions(-)

diff --git a/src/agents/chatbot/configuration.py b/src/agents/chatbot/configuration.py
index d459b05b..2c4fafd1 100644
--- a/src/agents/chatbot/configuration.py
+++ b/src/agents/chatbot/configuration.py
@@ -22,9 +22,3 @@ class ChatbotConfiguration(Configuration):
     model_provider: str = "zhipu"
     model_name: str = "glm-4-plus"
 
-    def __post_init__(self):
-        # TODO 需要确保这里的模型是支持 tools 的
-        if self.llm is None:
-            self.llm = select_model(config=config,
-                                    model_provider=self.model_provider,
-                                    model_name=self.model_name).chat_open_ai
diff --git a/src/agents/chatbot/graph.py b/src/agents/chatbot/graph.py
index ed421fcc..1c08bbc9 100644
--- a/src/agents/chatbot/graph.py
+++ b/src/agents/chatbot/graph.py
@@ -15,19 +15,19 @@ class ChatbotAgent(BaseAgent):
     name = "chatbot"
     description = "A chatbot that can answer questions and help with tasks."
     _graph_cache = None
+    config_schema = ChatbotConfiguration
 
-    def __init__(self, configuration: ChatbotConfiguration = None):
-        super().__init__(configuration)
-        self.configuration = configuration or ChatbotConfiguration()
-        self.llm = self.configuration.llm
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
 
-    def _get_tools(self, config: RunnableConfig):
+    def _get_tools(self, config_schema: RunnableConfig):
         """根据配置获取工具"""
+        conf = ChatbotConfiguration.from_runnable_config(config_schema)
         tools = [multiply]
-        if not config:
+        if not conf:
             return tools
 
-        if config.get("configurable", {}).get("use_web", None):
+        if conf.get("use_web", None):
             from langchain_community.tools.tavily_search import TavilySearchResults
             tools.append(TavilySearchResults(max_results=10))
 
@@ -39,11 +39,11 @@ class ChatbotAgent(BaseAgent):
         res = model.invoke(state["messages"])
         return {"messages": [res]}
 
-    def get_graph(self, config: RunnableConfig = None):
+    def get_graph(self, config_schema: RunnableConfig = None):
         """构建图"""
         workflow = StateGraph(State)
         workflow.add_node("chatbot", self.llm_call)
-        workflow.add_node("tools", ToolNode(tools=self._get_tools(config)))
+        workflow.add_node("tools", ToolNode(tools=self._get_tools(config_schema)))
         workflow.add_edge(START, "chatbot")
         workflow.add_conditional_edges(
             "chatbot",
@@ -55,9 +55,9 @@ class ChatbotAgent(BaseAgent):
         graph = workflow.compile(checkpointer=MemorySaver())
         return graph
 
-    def stream_values(self, messages: list[str], config: RunnableConfig = None):
-        graph = self.get_graph(config)
-        for event in graph.stream({"messages": messages}, stream_mode="values", config=config):
+    def stream_values(self, messages: list[str], config_schema: RunnableConfig = None):
+        graph = self.get_graph(config_schema)
+        for event in graph.stream({"messages": messages}, stream_mode="values", config=config_schema):
             yield event["messages"]
 
     def stream_messages(self, messages: list[str], config: RunnableConfig = None):
diff --git a/src/agents/registry.py b/src/agents/registry.py
index 39f0838f..714725df 100644
--- a/src/agents/registry.py
+++ b/src/agents/registry.py
@@ -1,14 +1,15 @@
+from __future__ import annotations
+
 from typing import Type, Annotated, Optional, TypedDict
 from abc import abstractmethod
+from dataclasses import dataclass, fields, field
 
-from langchain_openai import ChatOpenAI
-
+from langchain_core.runnables import RunnableConfig
 from langchain_core.messages import BaseMessage
 from langgraph.graph.state import CompiledStateGraph
 from langgraph.graph.message import add_messages
 
 
-from dataclasses import dataclass
 class State(TypedDict):
     """
     定义一个基础 State 供 各类 graph 继承, 其中:
@@ -25,14 +26,28 @@ class Configuration:
     """
     定义一个基础 Configuration 供 各类 graph 继承
     """
-    pass
+
+    user_id: str = field(metadata={"description": "Unique identifier for the user."})
+
+    @classmethod
+    def from_runnable_config(
+        cls, config: Optional[RunnableConfig] = None
+    ) -> Configuration:
+        """Create a Configuration instance from a RunnableConfig object."""
+        configurable = (config.get("configurable") or {}) if config else {}
+        _fields = {f.name for f in fields(cls) if f.init}
+        return cls(**{k: v for k, v in configurable.items() if k in _fields})
+
+    @classmethod
+    def to_dict(cls):
+        pass
 
 
 
 class BaseAgent():
 
-    def __init__(self, configuration: Configuration):
-        self.configuration = configuration
+    def __init__(self, **kwargs):
+        pass
 
     @abstractmethod
     def get_graph(self) -> CompiledStateGraph: