@@ -3,6 +3,7 @@ import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/
33import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js" ;
44import type { KeyValuePair } from "$lib/types/Tool" ;
55import { config } from "$lib/server/config" ;
6+ import { logger } from "$lib/server/logger" ;
67import type { RequestHandler } from "./$types" ;
78import { isValidUrl } from "$lib/server/urlSafety" ;
89import { 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