MCP機能追加

1 files changed, 2 insertions, 2 deletions

diff --git a/src/ollama_web/llm.py b/src/ollama_web/llm.py
index 461cef9..0ab00a3 100644
--- a/src/ollama_web/llm.py
+++ b/src/ollama_web/llm.py
@@ -191,7 +191,7 @@ def chat_with_tools(
     caps = capabilities or set()
     raw_client = get_client(host)
     reg = registry or default_registry()
-    tools: list[ToolCallable] = [cast(Any, c) for c in reg.callables]
+    tools: list[ToolCallable] = [cast(Any, t) for t in reg.ollama_tools]
     effective_tools = tools if "tools" in caps else []
     effective_think = think if "thinking" in caps else None
 
@@ -342,7 +342,7 @@ def stream_chat_with_tools(
     caps = capabilities or set()
     raw_client = get_client(host)
     reg = registry or default_registry()
-    tools: list[ToolCallable] = [cast(Any, c) for c in reg.callables]
+    tools: list[ToolCallable] = [cast(Any, t) for t in reg.ollama_tools]
     effective_tools = tools if "tools" in caps else []
     effective_think = think if "thinking" in caps else None