diff --git a/client/src/lib/chatLoop.ts b/client/src/lib/chatLoop.ts index 167dcba0e1..6b12da7ecf 100644 --- a/client/src/lib/chatLoop.ts +++ b/client/src/lib/chatLoop.ts @@ -126,12 +126,12 @@ export class QueryProcessor { tools: AnthropicTool[], onUpdate?: (content: string) => void, model: string = "claude-3-5-sonnet-latest", - provider?: string, + provider?: SupportedProvider, signal?: AbortSignal, ): Promise { // Get the specified provider or fall back to default const aiProvider = provider - ? providerManager.getProvider(provider as SupportedProvider) + ? providerManager.getProvider(provider) : providerManager.getDefaultProvider(); if (!aiProvider) { @@ -547,9 +547,6 @@ export class ChatLoop { }); try { - console.log("\nMCP Client Started!"); - console.log("Type your queries or 'quit' to exit."); - while (true) { const message = await rl.question("\nQuery: "); if (message.toLowerCase() === "quit") { diff --git a/client/src/mcpjamAgent.ts b/client/src/mcpjamAgent.ts index 906f4129e6..844704ffd3 100644 --- a/client/src/mcpjamAgent.ts +++ b/client/src/mcpjamAgent.ts @@ -19,6 +19,7 @@ import { ClientLogLevels } from "./hooks/helpers/types"; import { ElicitationResponse } from "./components/ElicitationModal"; import { ChatLoopProvider, ChatLoop, mappedTools, QueryProcessor, ToolCaller } from "./lib/chatLoop"; import { Tool as AnthropicTool } from "@anthropic-ai/sdk/resources/messages/messages.mjs"; +import { SupportedProvider } from "./lib/providers"; export interface MCPClientOptions { id?: string; @@ -473,7 +474,7 @@ export class MCPJamAgent implements ChatLoopProvider, ToolCaller { tools: AnthropicTool[], onUpdate?: (content: string) => void, model: string = "claude-3-5-sonnet-latest", - provider?: string, + provider?: SupportedProvider, signal?: AbortSignal, ): Promise { return this.queryProcessor.processQuery(query, tools, onUpdate, model, provider, signal); diff --git a/client/src/mcpjamClient.ts b/client/src/mcpjamClient.ts index 3a6b7e9524..5da57f309c 100644 --- a/client/src/mcpjamClient.ts +++ b/client/src/mcpjamClient.ts @@ -20,6 +20,7 @@ import { import { AIProvider, providerManager, + SupportedProvider, } from "@/lib/providers"; import { Tool, @@ -715,7 +716,7 @@ export class MCPJamClient extends Client implemen tools: Tool[], onUpdate?: (content: string) => void, model: string = "claude-3-5-sonnet-latest", - provider?: string, + provider?: SupportedProvider, signal?: AbortSignal, ): Promise { return this.queryProcessor.processQuery(query, tools, onUpdate, model, provider, signal);