MCP server機能をセキュア化

1 files changed, 5 insertions, 2 deletions

diff --git a/src/ollama_web/routes/chat.py b/src/ollama_web/routes/chat.py
index 75aabd7..2d855b1 100644
--- a/src/ollama_web/routes/chat.py
+++ b/src/ollama_web/routes/chat.py
@@ -90,7 +90,10 @@ async def _build_registry(session_id: str | None) -> ToolRegistry:
             name = fn.get("name")
             if not name:
                 continue
-            reg.register_mcp_tool(name, tool_def, executor)
+            try:
+                reg.register_mcp_tool(name, tool_def, executor)
+            except ValueError as exc:
+                logger.warning("skipping duplicate MCP tool %s: %s", name, exc)
 
     return reg
 
@@ -330,4 +333,4 @@ async def chat(request: Request) -> StreamingResponse:
 
 async def _single_event(data: str) -> AsyncIterator[bytes]:
     """Yield a single SSE event and finish."""
-    yield f"data: {data}\n\n".encode()
\ No newline at end of file
+    yield f"data: {data}\n\n".encode()