diff --git a/main/xiaozhi-server/core/mcp/MCPClient.py b/main/xiaozhi-server/core/mcp/MCPClient.py index 38a8e62a..b6b6335f 100644 --- a/main/xiaozhi-server/core/mcp/MCPClient.py +++ b/main/xiaozhi-server/core/mcp/MCPClient.py @@ -92,15 +92,21 @@ class MCPClient: args=self.config.get("args", []), env=env, ) - stdio_r, stdio_w = await stack.enter_async_context(stdio_client(params)) + stdio_r, stdio_w = await stack.enter_async_context( + stdio_client(params) + ) read_stream, write_stream = stdio_r, stdio_w # 建立SSEClient elif "url" in self.config: if "API_ACCESS_TOKEN" in self.config: - headers = {"Authorization": f"Bearer {self.config['API_ACCESS_TOKEN']}"} + headers = { + "Authorization": f"Bearer {self.config['API_ACCESS_TOKEN']}" + } else: headers = {} - sse_r, sse_w = await stack.enter_async_context(sse_client(self.config["url"], , headers=headers)) + sse_r, sse_w = await stack.enter_async_context( + sse_client(self.config["url"], headers=headers) + ) read_stream, write_stream = sse_r, sse_w else: