sycommon-python-lib 0.2.1a30__tar.gz → 0.2.1a31__tar.gz

{sycommon_python_lib-0.2.1a30 → sycommon_python_lib-0.2.1a31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sycommon-python-lib
-Version: 0.2.1a30
+Version: 0.2.1a31
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{sycommon_python_lib-0.2.1a30 → sycommon_python_lib-0.2.1a31}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sycommon-python-lib"
-version = "0.2.1a30"
+version = "0.2.1a31"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"

{sycommon_python_lib-0.2.1a30 → sycommon_python_lib-0.2.1a31}/src/sycommon/middleware/background_execution.py

@@ -168,6 +168,8 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             backend: 后端实例或工厂函数。如果为 None，从 ToolRuntime 获取。
         """
         self._backend = backend
+        # 缓存已解析的 backend（在 wrap_model_call 中首次获取后会缓存）
+        self._resolved_backend: BackendProtocol | None = None
 
         # 预创建工具（会在 wrap_model_call 中根据 backend 能力过滤）
         self.tools = [
@@ -176,14 +178,58 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             self._create_kill_process_tool(),
         ]
 
-    def _get_backend(self, runtime: ToolRuntime) -> BackendProtocol:
-        """获取 backend 实例"""
-        if self._backend is None:
-            # 从 runtime 获取 backend
-            return runtime.config.get("configurable", {}).get("backend")
-        if callable(self._backend):
-            return self._backend(runtime)
-        return self._backend
+    def _get_backend_from_tool_runtime(self, runtime: ToolRuntime) -> BackendProtocol | None:
+        """从 ToolRuntime 获取 backend（工具函数中使用）"""
+        if self._resolved_backend is not None:
+            return self._resolved_backend
+
+        if self._backend is not None:
+            if callable(self._backend):
+                return self._backend(runtime)
+            return self._backend
+
+        # 尝试从 runtime.config 获取
+        config = getattr(runtime, 'config', None)
+        if config:
+            backend = config.get("configurable", {}).get("backend")
+            if backend:
+                self._resolved_backend = backend
+                return backend
+
+        # 尝试从 state 获取
+        state = getattr(runtime, 'state', None)
+        if state is not None:
+            self._resolved_backend = state
+            return state
+
+        return None
+
+    def _get_backend_from_runtime(self, runtime) -> BackendProtocol | None:
+        """从 Runtime 获取 backend（wrap_model_call 中使用）"""
+        if self._resolved_backend is not None:
+            return self._resolved_backend
+
+        if self._backend is not None:
+            if callable(self._backend):
+                # 构造 ToolRuntime 来调用工厂函数
+                from langchain.tools import ToolRuntime
+                config = getattr(runtime, 'config', {})
+                tool_runtime = ToolRuntime(
+                    state=None,
+                    context=getattr(runtime, 'context', None),
+                    stream_writer=getattr(runtime, 'stream_writer', None),
+                    store=getattr(runtime, 'store', None),
+                    config=config,
+                    tool_call_id=None,
+                )
+                backend = self._backend(tool_runtime)
+                if backend:
+                    self._resolved_backend = backend
+                return backend
+            self._resolved_backend = self._backend
+            return self._backend
+
+        return None
 
     def _create_execute_background_tool(self) -> BaseTool:
         """创建 execute_background 工具"""
@@ -193,7 +239,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             runtime: ToolRuntime[None, BackgroundExecutionState],
             timeout: Annotated[int, "命令最大运行时间（秒），默认 600 秒"] = 600,
         ) -> str:
-            backend = self._get_backend(runtime)
+            backend = self._get_backend_from_tool_runtime(runtime)
 
             if not _supports_background_execution(backend):
                 return "Error: 后台执行不可用。请确保 backend 实现了 execute_background 方法。"
@@ -215,7 +261,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             runtime: ToolRuntime[None, BackgroundExecutionState],
             timeout: Annotated[int, "命令最大运行时间（秒），默认 600 秒"] = 600,
         ) -> str:
-            backend = self._get_backend(runtime)
+            backend = self._get_backend_from_tool_runtime(runtime)
 
             if not _supports_background_execution(backend):
                 return "Error: 后台执行不可用。请确保 backend 实现了 execute_background 方法。"
@@ -246,7 +292,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             process_id: Annotated[str, "要检查的进程 ID"],
             runtime: ToolRuntime[None, BackgroundExecutionState],
         ) -> str:
-            backend = self._get_backend(runtime)
+            backend = self._get_backend_from_tool_runtime(runtime)
 
             if not _supports_background_execution(backend):
                 return "Error: 后台执行不可用。"
@@ -280,7 +326,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             process_id: Annotated[str, "要检查的进程 ID"],
             runtime: ToolRuntime[None, BackgroundExecutionState],
         ) -> str:
-            backend = self._get_backend(runtime)
+            backend = self._get_backend_from_tool_runtime(runtime)
 
             if not _supports_background_execution(backend):
                 return "Error: 后台执行不可用。"
@@ -324,7 +370,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             process_id: Annotated[str, "要终止的进程 ID"],
             runtime: ToolRuntime[None, BackgroundExecutionState],
         ) -> str:
-            backend = self._get_backend(runtime)
+            backend = self._get_backend_from_tool_runtime(runtime)
 
             if not _supports_background_execution(backend):
                 return "Error: 后台执行不可用。"
@@ -349,7 +395,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
             process_id: Annotated[str, "要终止的进程 ID"],
             runtime: ToolRuntime[None, BackgroundExecutionState],
         ) -> str:
-            backend = self._get_backend(runtime)
+            backend = self._get_backend_from_tool_runtime(runtime)
 
             if not _supports_background_execution(backend):
                 return "Error: 后台执行不可用。"
@@ -384,7 +430,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
     ) -> ModelResponse[ResponseT]:
         """在 model call 前添加工具和系统提示"""
         # 检查 backend 是否支持后台执行
-        backend = self._get_backend(request.runtime)  # type: ignore
+        backend = self._get_backend_from_runtime(request.runtime)  # type: ignore
         if not _supports_background_execution(backend):
             # 不支持则直接透传
             return handler(request)
@@ -420,7 +466,7 @@ class BackgroundExecutionMiddleware(AgentMiddleware[BackgroundExecutionState, Co
     ) -> ModelResponse[ResponseT]:
         """(async) 在 model call 前添加工具和系统提示"""
         # 检查 backend 是否支持后台执行
-        backend = self._get_backend(request.runtime)  # type: ignore
+        backend = self._get_backend_from_runtime(request.runtime)  # type: ignore
         if not _supports_background_execution(backend):
             return await handler(request)
 

{sycommon_python_lib-0.2.1a30 → sycommon_python_lib-0.2.1a31}/src/sycommon_python_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sycommon-python-lib
-Version: 0.2.1a30
+Version: 0.2.1a31
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown