From fce9dc1b2940ee6699d2acf0c9e56456d9899581 Mon Sep 17 00:00:00 2001 From: tanaka takashi Date: Sat, 27 Jun 2026 01:09:00 +0900 Subject: MCP server機能をセキュア化 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/ollama_web/routes/chat.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'src/ollama_web/routes/chat.py') 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() -- cgit v1.2.3