chore: cleanup

Author: damassi

README.md

@@ -115,6 +115,8 @@ const config = {
   systemPrompt: getPrompt("system.md"),
   mcpServers: {
     someMcpServer: {
+      description:
+        "A detailed description of the MCP server and its capabilities used to provide hints to inference agent",
       command: "bunx",
       args: ["..."],
       prompt: getPrompt("someMcpServer.md"),
@@ -123,6 +125,8 @@ const config = {
 }
 ```
 
+The `description` field is **critical**; it's used by the inference routing agent to determine when to invoke the server or subagent.
+
 #### Remote Prompts
 
 Prompts can be loaded from remote sources (e.g., APIs) using `getRemotePrompt`. This enables dynamic prompt management where prompts are stored in a database or CMS rather than in files.
@@ -236,8 +240,6 @@ When a user asks something like "Analyze partner churn", the routing agent will:
 2. Automatically connect to the required `salesforce` MCP server
 3. Invoke the subagent with its specialized prompt and tools
 
-The `description` field is **critical**; it's used by the routing agent to determine when to invoke the subagent.
-
 **Note:** Subagents also support remote prompts via `getRemotePrompt`, allowing you to manage agent prompts dynamically from an API or database.
 
 ### Note on Lazy MCP Server Initialization

src/mcp/getMcpServer.ts

@@ -5,7 +5,7 @@ import { z } from "zod"
 
 export const getMcpServer = () => {
   // Store Claude Agent SDK sessionId per-instance (not shared across threads)
-  let claudeSessionId: string | undefined
+  let sessionId: string | undefined
 
   // Map thread IDs to Claude Agent SDK session IDs for per-thread isolation
   const threadSessions = new Map<string, string>()
@@ -35,23 +35,23 @@ export const getMcpServer = () => {
       },
     },
     async ({ query }) => {
-      const existingConnectedServers = claudeSessionId
-        ? sessionConnectedServers.get(claudeSessionId)
+      const existingConnectedServers = sessionId
+        ? sessionConnectedServers.get(sessionId)
         : undefined
 
       const { response, connectedServers } = await runStandaloneAgentLoop({
         prompt: query,
         mcpServer,
-        sessionId: claudeSessionId,
+        sessionId,
         existingConnectedServers,
         onSessionIdReceived: (newSessionId) => {
-          claudeSessionId = newSessionId
+          sessionId = newSessionId
         },
       })
 
       // Update the session's connected servers
-      if (claudeSessionId) {
-        sessionConnectedServers.set(claudeSessionId, connectedServers)
+      if (sessionId) {
+        sessionConnectedServers.set(sessionId, connectedServers)
       }
 
       return {

src/store.ts

@@ -41,16 +41,17 @@ export interface ToolDenied {
 export type ChatHistoryEntry = Message | ToolUse | ToolDenied
 
 type McpServerConfigWithPrompt = McpServerConfig & {
+  /** A detailed description of the MCP server that the inference agent evaluates */
   description: string
-  prompt?: () => Promise<string>
   disallowedTools?: string[]
   enabled?: boolean
+  prompt?: () => Promise<string>
 }
 
 export interface AgentChatConfig {
   agents?: Record<string, AgentConfig>
-  disallowedTools?: string[]
   connectionTimeout?: number
+  disallowedTools?: string[]
   maxRetries?: number
   mcpServers: Record<string, McpServerConfigWithPrompt>
   model?: "sonnet" | "haiku"
@@ -61,8 +62,8 @@ export interface AgentChatConfig {
 }
 
 export interface PendingToolPermission {
-  toolName: string
   input: any
+  toolName: string
 }
 
 export interface StoreModel {
@@ -80,9 +81,9 @@ export interface StoreModel {
   stats?: string | null
 
   // Computed
-  isBooted: Computed<StoreModel, boolean>
-  availableMcpServers: Computed<StoreModel, string[]>
   availableAgents: Computed<StoreModel, string[]>
+  availableMcpServers: Computed<StoreModel, string[]>
+  isBooted: Computed<StoreModel, boolean>
 
   // Actions
   abortRequest: Action<StoreModel>
@@ -91,6 +92,7 @@ export interface StoreModel {
   appendCurrentAssistantMessage: Action<StoreModel, string>
   clearCurrentAssistantMessage: Action<StoreModel>
   clearToolUses: Action<StoreModel>
+  handleMcpServerStatus: Thunk<StoreModel, McpServerStatus[]>
   reset: Action<StoreModel>
   sendMessage: Action<StoreModel, string>
   setPendingToolPermission: Action<
@@ -106,22 +108,21 @@ export interface StoreModel {
   setMcpServers: Action<StoreModel, McpServerStatus[]>
   setSessionId: Action<StoreModel, string>
   setStats: Action<StoreModel, string | null>
-  handleMcpServerStatus: Thunk<StoreModel, McpServerStatus[]>
 }
 
 export const AgentStore = createContextStore<StoreModel>({
   abortController: new AbortController(),
   chatHistory: [],
-  messageQueue: new MessageQueue(),
-  sessionId: undefined,
-  mcpServers: [],
-  input: "",
-  isProcessing: false,
+  config: null as unknown as AgentChatConfig,
   currentAssistantMessage: "",
   currentToolUses: [],
+  input: "",
+  isProcessing: false,
+  mcpServers: [],
+  messageQueue: new MessageQueue(),
   pendingToolPermission: undefined,
+  sessionId: undefined,
   stats: undefined,
-  config: null as unknown as AgentChatConfig,
 
   // Computed
   isBooted: computed((state) => {