Skip to content

Commit 7bc9b59

Browse files
authored
fix(mcp): use logger (#2014)
1 parent 933dcda commit 7bc9b59

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

src/routes/api/mcp/health/+server.ts

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/
33
import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
44
import type { KeyValuePair } from "$lib/types/Tool";
55
import { config } from "$lib/server/config";
6+
import { logger } from "$lib/server/logger";
67
import type { RequestHandler } from "./$types";
78
import { isValidUrl } from "$lib/server/urlSafety";
89
import { isStrictHfMcpLogin, hasNonEmptyToken } from "$lib/server/mcp/hf";
@@ -88,16 +89,16 @@ export const POST: RequestHandler = async ({ request, locals }) => {
8889

8990
// Try Streamable HTTP transport first
9091
try {
91-
console.log(`[MCP Health] Trying HTTP transport for ${url}`);
92+
logger.info(`[MCP Health] Trying HTTP transport for ${url}`);
9293
client = new Client({
9394
name: "chat-ui-health-check",
9495
version: "1.0.0",
9596
});
9697

9798
const transport = new StreamableHTTPClientTransport(baseUrl, { requestInit });
98-
console.log(`[MCP Health] Connecting to ${url}...`);
99+
logger.info(`[MCP Health] Connecting to ${url}...`);
99100
await client.connect(transport);
100-
console.log(`[MCP Health] Connected successfully via HTTP`);
101+
logger.info(`[MCP Health] Connected successfully via HTTP`);
101102

102103
// Connection successful, get tools
103104
const toolsResponse = await client.listTools();
@@ -140,7 +141,7 @@ export const POST: RequestHandler = async ({ request, locals }) => {
140141
} catch (error) {
141142
httpError = error instanceof Error ? error : new Error(String(error));
142143
lastError = httpError;
143-
console.log("Streamable HTTP failed, trying SSE transport...", lastError.message);
144+
logger.info("Streamable HTTP failed, trying SSE transport...", lastError.message);
144145

145146
// Close failed client
146147
try {
@@ -151,16 +152,16 @@ export const POST: RequestHandler = async ({ request, locals }) => {
151152

152153
// Try SSE transport
153154
try {
154-
console.log(`[MCP Health] Trying SSE transport for ${url}`);
155+
logger.info(`[MCP Health] Trying SSE transport for ${url}`);
155156
client = new Client({
156157
name: "chat-ui-health-check",
157158
version: "1.0.0",
158159
});
159160

160161
const sseTransport = new SSEClientTransport(baseUrl, { requestInit });
161-
console.log(`[MCP Health] Connecting via SSE...`);
162+
logger.info(`[MCP Health] Connecting via SSE...`);
162163
await client.connect(sseTransport);
163-
console.log(`[MCP Health] Connected successfully via SSE`);
164+
logger.info(`[MCP Health] Connected successfully via SSE`);
164165

165166
// Connection successful, get tools
166167
const toolsResponse = await client.listTools();
@@ -210,7 +211,7 @@ export const POST: RequestHandler = async ({ request, locals }) => {
210211
{ cause: sseError instanceof Error ? sseError : undefined }
211212
);
212213
}
213-
console.error("Both transports failed. Last error:", lastError);
214+
logger.error("Both transports failed. Last error:", lastError);
214215
}
215216
}
216217

@@ -252,7 +253,7 @@ export const POST: RequestHandler = async ({ request, locals }) => {
252253
clearTimeout(timeoutId);
253254
return res;
254255
} catch (error) {
255-
console.error("MCP health check failed:", error);
256+
logger.error("MCP health check failed:", error);
256257

257258
// Clean up client if it exists
258259
try {

0 commit comments

Comments
 (0)