@truefoundry/trueforge-core 0.0.0 → 0.1.0-rc.0

package/dist/core/runtime/DeferredTool.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/runtime/DeferredTool.ts"],"sourcesContent":["import type { CallToolRequest } from '@modelcontextprotocol/sdk/types.js';\nimport dedent from 'dedent';\nimport type { Logger } from 'winston';\nimport { z } from 'zod';\nimport type { ApprovalDecision } from '../events/schema';\nimport { InstructionBuilder } from '../InstructionBuilder';\nimport type { InternalToolCallInfo } from '../llm/LLMTypes';\nimport {\n  type AgentToolSchema,\n  type CallToolResponse,\n  isApprovalRequiredResponse,\n  isAuthRequired,\n  isCallToolResponseResult,\n  type IToolSet,\n  type ListToolsResolvedResponse,\n  toolResultResponse,\n} from '../mcp/IMCPServer';\nimport { defineTool, LocalToolMCP, type ToolDefinition } from '../mcp/LocalToolMCP';\nimport type { AgentTracing } from '../tracing/AgentTracing';\nimport { extractErrorLogFields } from '../util/errorLogFields';\n\nexport const DEFERRED_TOOLS_SERVER_ID = 'deferred-tools';\nconst DEFERRED_TOOLS_INSTRUCTION = 'deferred-tools-instructions';\nexport const LIST_TOOLS_NAME = 'list_tools';\nexport const GET_TOOL_INFO_NAME = 'get_tool_info';\nexport const GET_TOOL_OUTPUT_SCHEMA_NAME = 'get_tool_output_schema';\nconst CALL_TOOL_NAME = 'call_tool';\nconst MAX_DESCRIPTION_LENGTH = 200;\n\nconst listToolsSchema = z.object({\n  mcp_server: z.string().describe('Name of the MCP server to list tools for.'),\n});\n\nconst getToolInfoSchema = z.object({\n  mcp_server: z.string().describe('Name of the MCP server.'),\n  tool_name: z.string().describe('Name of the tool to inspect.'),\n});\n\nconst getToolOutputSchemaSchema = z.object({\n  mcp_server: z.string().describe('Name of the MCP server.'),\n  tool_name: z.string().describe('Name of the tool to get output schema for.'),\n});\n\nconst callToolSchema = z.object({\n  mcp_server: z.string().describe('Name of the MCP server.'),\n  tool_name: z.string().describe('Name of the tool to call.'),\n  input: z.record(z.string(), z.unknown()).default({}).describe('Arguments to pass to the tool.'),\n});\n\ninterface ResolvedServerTools {\n  server: IToolSet;\n  tools: AgentToolSchema[];\n  metadata: Omit<ListToolsResolvedResponse, 'result'>;\n}\n\nfunction isResolvedServerTools(result: CallToolResponse | ResolvedServerTools): result is ResolvedServerTools {\n  return 'tools' in result;\n}\n\nexport class DeferredTool extends LocalToolMCP {\n  readonly name = DEFERRED_TOOLS_SERVER_ID;\n  readonly displayName = 'DeferredTools';\n  override readonly description = 'Deferred tool loading for user MCP servers';\n\n  private readonly serverMap: Map<string, IToolSet>;\n  private readonly tools: ToolDefinition[];\n  private readonly logger: Logger;\n\n  constructor(\n    servers: readonly IToolSet[],\n    options: {\n      tracing: AgentTracing;\n      logger: Logger;\n    },\n  ) {\n    super({ tracing: options.tracing });\n    this.logger = options.logger.child({ module: 'DeferredTool' });\n    this.serverMap = new Map(servers.map(s => [s.name, s]));\n    this.tools = this.buildTools();\n  }\n\n  buildInstruction(builder: InstructionBuilder): void {\n    const servers = [...this.serverMap.values()];\n\n    // Membership follows `preload` only (origin/main). Selective `preload_tools` /\n    // `hasPreloadedTools` must not hide a server from deferred discovery instructions.\n    const deferredServers = servers.filter(s => !s.preload);\n\n    if (deferredServers.length === 0) return;\n\n    const deferredList = deferredServers\n      .map(s => (s.description ? `- ${s.name}: \"${s.description.slice(0, MAX_DESCRIPTION_LENGTH)}\"` : `- ${s.name}`))\n      .join('\\n');\n\n    builder.addSection(\n      DEFERRED_TOOLS_INSTRUCTION,\n      dedent`\n        The following MCP servers have deferred tool loading. Their tools are NOT pre-loaded.\n        ${deferredList}\n\n        The Agent MUST use ${LIST_TOOLS_NAME} and ${GET_TOOL_INFO_NAME} to discover tools before calling them.`,\n    );\n  }\n\n  protected getTools(): ToolDefinition[] {\n    return this.tools;\n  }\n\n  override async toolCallInfo(\n    params: CallToolRequest['params'],\n    resolveUnderlyingTool?: boolean,\n  ): Promise<InternalToolCallInfo> {\n    if (params.name !== CALL_TOOL_NAME) {\n      return super.toolCallInfo(params, resolveUnderlyingTool);\n    }\n    if (resolveUnderlyingTool === false) {\n      const base = await super.toolCallInfo(params, resolveUnderlyingTool);\n      return { ...base, is_deferred: true };\n    }\n    const parsed = callToolSchema.safeParse(params.arguments);\n    if (!parsed.success) {\n      this.logger.warn(`Failed to parse call_tool arguments`, {\n        tool: params.name,\n        errors: parsed.error.issues,\n      });\n      return { ...(await super.toolCallInfo(params)), is_deferred: true };\n    }\n    const server = this.resolveServer(parsed.data.mcp_server);\n    if (!server) {\n      this.logger.warn(`Server not found for toolCallInfo`, {\n        mcpServer: parsed.data.mcp_server,\n        toolName: parsed.data.tool_name,\n        availableServers: [...this.serverMap.keys()],\n      });\n      return { ...(await super.toolCallInfo(params)), is_deferred: true };\n    }\n    const underlying = await server.toolCallInfo({\n      name: parsed.data.tool_name,\n      arguments: parsed.data.input,\n    });\n    return { ...underlying, is_deferred: true };\n  }\n\n  private resolveServer(name: string): IToolSet | undefined {\n    return this.serverMap.get(name);\n  }\n\n  private async resolveServerTools(mcpServer: string): Promise<CallToolResponse | ResolvedServerTools> {\n    const server = this.resolveServer(mcpServer);\n    if (!server) {\n      return toolResultResponse({\n        text: JSON.stringify({ error: `MCP server '${mcpServer}' not found` }),\n        isError: true,\n      });\n    }\n\n    const listResponse = await server.listTools();\n    if (isAuthRequired(listResponse)) {\n      return listResponse;\n    }\n    const { result, ...metadata } = listResponse;\n    return { server, tools: result.tools, metadata };\n  }\n\n  private buildTools(): ToolDefinition[] {\n    return [this.buildListTools(), this.buildGetToolInfo(), this.buildGetToolOutputSchema(), this.buildCallTool()];\n  }\n\n  private buildListTools(): ToolDefinition {\n    return defineTool({\n      name: LIST_TOOLS_NAME,\n      description: dedent`\n        List tool names available on an MCP server.\n        Returns tool names only — use ${GET_TOOL_INFO_NAME} to get full details before calling a tool.`,\n      schema: listToolsSchema,\n      handler: async (input: { mcp_server: string }) => {\n        try {\n          const resolved = await this.resolveServerTools(input.mcp_server);\n          if (!isResolvedServerTools(resolved)) return resolved;\n          const { server, tools, metadata } = resolved;\n          return toolResultResponse({\n            text: `${server.name}:\\n  ${tools.map(t => t.name).join(', ')}`,\n            overrides: metadata,\n          });\n        } catch (error) {\n          this.logger.error('list_tools failed', {\n            ...extractErrorLogFields(error),\n            mcpServer: input.mcp_server,\n          });\n          return toolResultResponse({\n            text: JSON.stringify({ error: `Failed to list tools for '${input.mcp_server}'.` }),\n            isError: true,\n          });\n        }\n      },\n    });\n  }\n\n  private buildGetToolInfo(): ToolDefinition {\n    return defineTool({\n      name: GET_TOOL_INFO_NAME,\n      description: dedent`\n        Get description, inputSchema and outputSchema for a specific tool on an MCP server.\n        If outputSchema is null, it will NOT be present in ${GET_TOOL_OUTPUT_SCHEMA_NAME} as well.\n        The Agent MUST call ${LIST_TOOLS_NAME} first to know which tools are available.`,\n      schema: getToolInfoSchema,\n      handler: async (input: { mcp_server: string; tool_name: string }) => {\n        try {\n          const resolved = await this.resolveServerTools(input.mcp_server);\n          if (!isResolvedServerTools(resolved)) return resolved;\n          const { tools, metadata } = resolved;\n          const tool = tools.find(t => t.name === input.tool_name);\n          if (!tool) {\n            return toolResultResponse({\n              text: JSON.stringify({ error: `Tool '${input.tool_name}' not found on server '${input.mcp_server}'` }),\n              overrides: metadata,\n              isError: true,\n            });\n          }\n\n          return toolResultResponse({\n            text: JSON.stringify({\n              description: tool.description,\n              inputSchema: tool.inputSchema,\n              outputSchema: tool.outputSchema ?? null,\n            }),\n            overrides: metadata,\n          });\n        } catch (error) {\n          this.logger.error('get_tool_info failed', {\n            ...extractErrorLogFields(error),\n            mcpServer: input.mcp_server,\n            toolName: input.tool_name,\n          });\n          return toolResultResponse({\n            text: JSON.stringify({\n              error: `Failed to get tool info: ${error instanceof Error ? error.message : 'unknown'}`,\n            }),\n            isError: true,\n          });\n        }\n      },\n    });\n  }\n\n  private buildGetToolOutputSchema(): ToolDefinition {\n    return defineTool({\n      name: GET_TOOL_OUTPUT_SCHEMA_NAME,\n      description: dedent`\n        Get the outputSchema for a specific tool on an MCP server.\n        The Agent MUST not call ${GET_TOOL_OUTPUT_SCHEMA_NAME} and ${GET_TOOL_INFO_NAME} for the same tool.\n        The Agent MUST either call ${GET_TOOL_OUTPUT_SCHEMA_NAME} or ${GET_TOOL_INFO_NAME} before entering Code Mode.\n        The Agent must NOT assume the structure of a tool's response from its name or description alone.`,\n      schema: getToolOutputSchemaSchema,\n      handler: async (input: { mcp_server: string; tool_name: string }) => {\n        try {\n          const resolved = await this.resolveServerTools(input.mcp_server);\n          if (!isResolvedServerTools(resolved)) return resolved;\n          const { tools, metadata } = resolved;\n          const tool = tools.find(t => t.name === input.tool_name);\n          if (!tool) {\n            return toolResultResponse({\n              text: JSON.stringify({ error: `Tool '${input.tool_name}' not found on server '${input.mcp_server}'` }),\n              overrides: metadata,\n              isError: true,\n            });\n          }\n\n          return toolResultResponse({\n            text: JSON.stringify({\n              outputSchema: tool.outputSchema ?? null,\n            }),\n            overrides: metadata,\n          });\n        } catch (error) {\n          this.logger.error('get_tool_output_schema failed', {\n            ...extractErrorLogFields(error),\n            mcpServer: input.mcp_server,\n            toolName: input.tool_name,\n          });\n          return toolResultResponse({\n            text: JSON.stringify({\n              error: `Failed to get output schema: ${error instanceof Error ? error.message : 'unknown'}`,\n            }),\n            isError: true,\n          });\n        }\n      },\n    });\n  }\n\n  private buildCallTool(): ToolDefinition {\n    return defineTool({\n      name: CALL_TOOL_NAME,\n      description: dedent`\n        Call a tool on an MCP server.\n        The Agent MUST call ${LIST_TOOLS_NAME} and ${GET_TOOL_INFO_NAME} before calling a tool.`,\n      schema: callToolSchema,\n      handler: async (\n        input: { mcp_server: string; tool_name: string; input: Record<string, unknown> },\n        approvalDecision?: ApprovalDecision,\n      ) => {\n        const server = this.resolveServer(input.mcp_server);\n        if (!server) {\n          return toolResultResponse({\n            text: JSON.stringify({ error: `MCP server '${input.mcp_server}' not found` }),\n            isError: true,\n          });\n        }\n\n        try {\n          const response = await server.callTool({ name: input.tool_name, arguments: input.input }, approvalDecision);\n\n          if (isAuthRequired(response)) {\n            return response;\n          }\n\n          if (isApprovalRequiredResponse(response)) return response;\n\n          if (!isCallToolResponseResult(response)) {\n            throw new Error('Tool call returned an unexpected sub-agent response');\n          }\n\n          return response;\n        } catch (error) {\n          this.logger.error('call_tool failed', {\n            ...extractErrorLogFields(error),\n            mcpServer: input.mcp_server,\n            toolName: input.tool_name,\n          });\n          return toolResultResponse({\n            text: JSON.stringify({ error: `Tool call failed: ${error instanceof Error ? error.message : 'unknown'}` }),\n            isError: true,\n          });\n        }\n      },\n    });\n  }\n}\n"],"mappings":";AACA,OAAO,YAAY;AAEnB,SAAS,SAAS;AAElB,OAAmC;AAEnC;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,EAGA;AAAA,OACK;AACP,SAAS,YAAY,oBAAyC;AAE9D,SAAS,6BAA6B;AAE/B,IAAM,2BAA2B;AACxC,IAAM,6BAA6B;AAC5B,IAAM,kBAAkB;AACxB,IAAM,qBAAqB;AAC3B,IAAM,8BAA8B;AAC3C,IAAM,iBAAiB;AACvB,IAAM,yBAAyB;AAE/B,IAAM,kBAAkB,EAAE,OAAO;AAAA,EAC/B,YAAY,EAAE,OAAO,EAAE,SAAS,2CAA2C;AAC7E,CAAC;AAED,IAAM,oBAAoB,EAAE,OAAO;AAAA,EACjC,YAAY,EAAE,OAAO,EAAE,SAAS,yBAAyB;AAAA,EACzD,WAAW,EAAE,OAAO,EAAE,SAAS,8BAA8B;AAC/D,CAAC;AAED,IAAM,4BAA4B,EAAE,OAAO;AAAA,EACzC,YAAY,EAAE,OAAO,EAAE,SAAS,yBAAyB;AAAA,EACzD,WAAW,EAAE,OAAO,EAAE,SAAS,4CAA4C;AAC7E,CAAC;AAED,IAAM,iBAAiB,EAAE,OAAO;AAAA,EAC9B,YAAY,EAAE,OAAO,EAAE,SAAS,yBAAyB;AAAA,EACzD,WAAW,EAAE,OAAO,EAAE,SAAS,2BAA2B;AAAA,EAC1D,OAAO,EAAE,OAAO,EAAE,OAAO,GAAG,EAAE,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC,EAAE,SAAS,gCAAgC;AAChG,CAAC;AAQD,SAAS,sBAAsB,QAA+E;AAC5G,SAAO,WAAW;AACpB;AAEO,IAAM,eAAN,cAA2B,aAAa;AAAA,EACpC,OAAO;AAAA,EACP,cAAc;AAAA,EACL,cAAc;AAAA,EAEf;AAAA,EACA;AAAA,EACA;AAAA,EAEjB,YACE,SACA,SAIA;AACA,UAAM,EAAE,SAAS,QAAQ,QAAQ,CAAC;AAClC,SAAK,SAAS,QAAQ,OAAO,MAAM,EAAE,QAAQ,eAAe,CAAC;AAC7D,SAAK,YAAY,IAAI,IAAI,QAAQ,IAAI,OAAK,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;AACtD,SAAK,QAAQ,KAAK,WAAW;AAAA,EAC/B;AAAA,EAEA,iBAAiB,SAAmC;AAClD,UAAM,UAAU,CAAC,GAAG,KAAK,UAAU,OAAO,CAAC;AAI3C,UAAM,kBAAkB,QAAQ,OAAO,OAAK,CAAC,EAAE,OAAO;AAEtD,QAAI,gBAAgB,WAAW,EAAG;AAElC,UAAM,eAAe,gBAClB,IAAI,OAAM,EAAE,cAAc,KAAK,EAAE,IAAI,MAAM,EAAE,YAAY,MAAM,GAAG,sBAAsB,CAAC,MAAM,KAAK,EAAE,IAAI,EAAG,EAC7G,KAAK,IAAI;AAEZ,YAAQ;AAAA,MACN;AAAA,MACA;AAAA;AAAA,UAEI,YAAY;AAAA;AAAA,6BAEO,eAAe,QAAQ,kBAAkB;AAAA,IAClE;AAAA,EACF;AAAA,EAEU,WAA6B;AACrC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAe,aACb,QACA,uBAC+B;AAC/B,QAAI,OAAO,SAAS,gBAAgB;AAClC,aAAO,MAAM,aAAa,QAAQ,qBAAqB;AAAA,IACzD;AACA,QAAI,0BAA0B,OAAO;AACnC,YAAM,OAAO,MAAM,MAAM,aAAa,QAAQ,qBAAqB;AACnE,aAAO,EAAE,GAAG,MAAM,aAAa,KAAK;AAAA,IACtC;AACA,UAAM,SAAS,eAAe,UAAU,OAAO,SAAS;AACxD,QAAI,CAAC,OAAO,SAAS;AACnB,WAAK,OAAO,KAAK,uCAAuC;AAAA,QACtD,MAAM,OAAO;AAAA,QACb,QAAQ,OAAO,MAAM;AAAA,MACvB,CAAC;AACD,aAAO,EAAE,GAAI,MAAM,MAAM,aAAa,MAAM,GAAI,aAAa,KAAK;AAAA,IACpE;AACA,UAAM,SAAS,KAAK,cAAc,OAAO,KAAK,UAAU;AACxD,QAAI,CAAC,QAAQ;AACX,WAAK,OAAO,KAAK,qCAAqC;AAAA,QACpD,WAAW,OAAO,KAAK;AAAA,QACvB,UAAU,OAAO,KAAK;AAAA,QACtB,kBAAkB,CAAC,GAAG,KAAK,UAAU,KAAK,CAAC;AAAA,MAC7C,CAAC;AACD,aAAO,EAAE,GAAI,MAAM,MAAM,aAAa,MAAM,GAAI,aAAa,KAAK;AAAA,IACpE;AACA,UAAM,aAAa,MAAM,OAAO,aAAa;AAAA,MAC3C,MAAM,OAAO,KAAK;AAAA,MAClB,WAAW,OAAO,KAAK;AAAA,IACzB,CAAC;AACD,WAAO,EAAE,GAAG,YAAY,aAAa,KAAK;AAAA,EAC5C;AAAA,EAEQ,cAAc,MAAoC;AACxD,WAAO,KAAK,UAAU,IAAI,IAAI;AAAA,EAChC;AAAA,EAEA,MAAc,mBAAmB,WAAoE;AACnG,UAAM,SAAS,KAAK,cAAc,SAAS;AAC3C,QAAI,CAAC,QAAQ;AACX,aAAO,mBAAmB;AAAA,QACxB,MAAM,KAAK,UAAU,EAAE,OAAO,eAAe,SAAS,cAAc,CAAC;AAAA,QACrE,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AAEA,UAAM,eAAe,MAAM,OAAO,UAAU;AAC5C,QAAI,eAAe,YAAY,GAAG;AAChC,aAAO;AAAA,IACT;AACA,UAAM,EAAE,QAAQ,GAAG,SAAS,IAAI;AAChC,WAAO,EAAE,QAAQ,OAAO,OAAO,OAAO,SAAS;AAAA,EACjD;AAAA,EAEQ,aAA+B;AACrC,WAAO,CAAC,KAAK,eAAe,GAAG,KAAK,iBAAiB,GAAG,KAAK,yBAAyB,GAAG,KAAK,cAAc,CAAC;AAAA,EAC/G;AAAA,EAEQ,iBAAiC;AACvC,WAAO,WAAW;AAAA,MAChB,MAAM;AAAA,MACN,aAAa;AAAA;AAAA,wCAEqB,kBAAkB;AAAA,MACpD,QAAQ;AAAA,MACR,SAAS,OAAO,UAAkC;AAChD,YAAI;AACF,gBAAM,WAAW,MAAM,KAAK,mBAAmB,MAAM,UAAU;AAC/D,cAAI,CAAC,sBAAsB,QAAQ,EAAG,QAAO;AAC7C,gBAAM,EAAE,QAAQ,OAAO,SAAS,IAAI;AACpC,iBAAO,mBAAmB;AAAA,YACxB,MAAM,GAAG,OAAO,IAAI;AAAA,IAAQ,MAAM,IAAI,OAAK,EAAE,IAAI,EAAE,KAAK,IAAI,CAAC;AAAA,YAC7D,WAAW;AAAA,UACb,CAAC;AAAA,QACH,SAAS,OAAO;AACd,eAAK,OAAO,MAAM,qBAAqB;AAAA,YACrC,GAAG,sBAAsB,KAAK;AAAA,YAC9B,WAAW,MAAM;AAAA,UACnB,CAAC;AACD,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU,EAAE,OAAO,6BAA6B,MAAM,UAAU,KAAK,CAAC;AAAA,YACjF,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,mBAAmC;AACzC,WAAO,WAAW;AAAA,MAChB,MAAM;AAAA,MACN,aAAa;AAAA;AAAA,6DAE0C,2BAA2B;AAAA,8BAC1D,eAAe;AAAA,MACvC,QAAQ;AAAA,MACR,SAAS,OAAO,UAAqD;AACnE,YAAI;AACF,gBAAM,WAAW,MAAM,KAAK,mBAAmB,MAAM,UAAU;AAC/D,cAAI,CAAC,sBAAsB,QAAQ,EAAG,QAAO;AAC7C,gBAAM,EAAE,OAAO,SAAS,IAAI;AAC5B,gBAAM,OAAO,MAAM,KAAK,OAAK,EAAE,SAAS,MAAM,SAAS;AACvD,cAAI,CAAC,MAAM;AACT,mBAAO,mBAAmB;AAAA,cACxB,MAAM,KAAK,UAAU,EAAE,OAAO,SAAS,MAAM,SAAS,0BAA0B,MAAM,UAAU,IAAI,CAAC;AAAA,cACrG,WAAW;AAAA,cACX,SAAS;AAAA,YACX,CAAC;AAAA,UACH;AAEA,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU;AAAA,cACnB,aAAa,KAAK;AAAA,cAClB,aAAa,KAAK;AAAA,cAClB,cAAc,KAAK,gBAAgB;AAAA,YACrC,CAAC;AAAA,YACD,WAAW;AAAA,UACb,CAAC;AAAA,QACH,SAAS,OAAO;AACd,eAAK,OAAO,MAAM,wBAAwB;AAAA,YACxC,GAAG,sBAAsB,KAAK;AAAA,YAC9B,WAAW,MAAM;AAAA,YACjB,UAAU,MAAM;AAAA,UAClB,CAAC;AACD,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU;AAAA,cACnB,OAAO,4BAA4B,iBAAiB,QAAQ,MAAM,UAAU,SAAS;AAAA,YACvF,CAAC;AAAA,YACD,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,2BAA2C;AACjD,WAAO,WAAW;AAAA,MAChB,MAAM;AAAA,MACN,aAAa;AAAA;AAAA,kCAEe,2BAA2B,QAAQ,kBAAkB;AAAA,qCAClD,2BAA2B,OAAO,kBAAkB;AAAA;AAAA,MAEnF,QAAQ;AAAA,MACR,SAAS,OAAO,UAAqD;AACnE,YAAI;AACF,gBAAM,WAAW,MAAM,KAAK,mBAAmB,MAAM,UAAU;AAC/D,cAAI,CAAC,sBAAsB,QAAQ,EAAG,QAAO;AAC7C,gBAAM,EAAE,OAAO,SAAS,IAAI;AAC5B,gBAAM,OAAO,MAAM,KAAK,OAAK,EAAE,SAAS,MAAM,SAAS;AACvD,cAAI,CAAC,MAAM;AACT,mBAAO,mBAAmB;AAAA,cACxB,MAAM,KAAK,UAAU,EAAE,OAAO,SAAS,MAAM,SAAS,0BAA0B,MAAM,UAAU,IAAI,CAAC;AAAA,cACrG,WAAW;AAAA,cACX,SAAS;AAAA,YACX,CAAC;AAAA,UACH;AAEA,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU;AAAA,cACnB,cAAc,KAAK,gBAAgB;AAAA,YACrC,CAAC;AAAA,YACD,WAAW;AAAA,UACb,CAAC;AAAA,QACH,SAAS,OAAO;AACd,eAAK,OAAO,MAAM,iCAAiC;AAAA,YACjD,GAAG,sBAAsB,KAAK;AAAA,YAC9B,WAAW,MAAM;AAAA,YACjB,UAAU,MAAM;AAAA,UAClB,CAAC;AACD,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU;AAAA,cACnB,OAAO,gCAAgC,iBAAiB,QAAQ,MAAM,UAAU,SAAS;AAAA,YAC3F,CAAC;AAAA,YACD,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,gBAAgC;AACtC,WAAO,WAAW;AAAA,MAChB,MAAM;AAAA,MACN,aAAa;AAAA;AAAA,8BAEW,eAAe,QAAQ,kBAAkB;AAAA,MACjE,QAAQ;AAAA,MACR,SAAS,OACP,OACA,qBACG;AACH,cAAM,SAAS,KAAK,cAAc,MAAM,UAAU;AAClD,YAAI,CAAC,QAAQ;AACX,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU,EAAE,OAAO,eAAe,MAAM,UAAU,cAAc,CAAC;AAAA,YAC5E,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAEA,YAAI;AACF,gBAAM,WAAW,MAAM,OAAO,SAAS,EAAE,MAAM,MAAM,WAAW,WAAW,MAAM,MAAM,GAAG,gBAAgB;AAE1G,cAAI,eAAe,QAAQ,GAAG;AAC5B,mBAAO;AAAA,UACT;AAEA,cAAI,2BAA2B,QAAQ,EAAG,QAAO;AAEjD,cAAI,CAAC,yBAAyB,QAAQ,GAAG;AACvC,kBAAM,IAAI,MAAM,qDAAqD;AAAA,UACvE;AAEA,iBAAO;AAAA,QACT,SAAS,OAAO;AACd,eAAK,OAAO,MAAM,oBAAoB;AAAA,YACpC,GAAG,sBAAsB,KAAK;AAAA,YAC9B,WAAW,MAAM;AAAA,YACjB,UAAU,MAAM;AAAA,UAClB,CAAC;AACD,iBAAO,mBAAmB;AAAA,YACxB,MAAM,KAAK,UAAU,EAAE,OAAO,qBAAqB,iBAAiB,QAAQ,MAAM,UAAU,SAAS,GAAG,CAAC;AAAA,YACzG,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;","names":[]}

package/dist/core/runtime/OpenToolCallCloser.d.ts

@@ -0,0 +1,21 @@
+/**
+ * OpenToolCallCloser - Context processor that closes unresolved tool calls.
+ *
+ * When an assistant message contains tool_calls but corresponding tool response
+ * messages are missing (e.g., due to a failed or interrupted tool execution),
+ * LLMs will reject the request. This processor detects such cases in the
+ * **last** assistant message and appends dummy tool responses so the
+ * conversation can continue.
+ *
+ * Uses is_thread_creation from InternalToolCallInfo to identify sub-agent
+ * tool calls (which should remain open); does NOT use SUB_AGENTS_SERVER_ID.
+ * Older persisted contexts without is_thread_creation follow the ordinary
+ * non-thread-creation path (no compatibility fallback).
+ */
+import type { AgentContextProcessorAppendContext, AgentThreadExecutionContext, PreSendContextProcessor } from '../capabilities/AgentContextProcessor';
+import type { ContextMessage } from './AgentThread.types';
+export declare function getClosableOpenToolCallIds(context: ContextMessage[]): Set<string>;
+export declare class OpenToolCallCloser implements PreSendContextProcessor {
+    processPreSend(execution: Readonly<AgentThreadExecutionContext>): AsyncGenerator<AgentContextProcessorAppendContext, void, unknown>;
+}
+//# sourceMappingURL=OpenToolCallCloser.d.ts.map

package/dist/core/runtime/OpenToolCallCloser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenToolCallCloser.d.ts","sourceRoot":"","sources":["../../../src/core/runtime/OpenToolCallCloser.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AACH,OAAO,KAAK,EACV,kCAAkC,EAClC,2BAA2B,EAC3B,uBAAuB,EACxB,MAAM,uCAAuC,CAAC;AAE/C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAS1D,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,cAAc,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,CAqCjF;AAED,qBAAa,kBAAmB,YAAW,uBAAuB;IAEzD,cAAc,CACnB,SAAS,EAAE,QAAQ,CAAC,2BAA2B,CAAC,GAC/C,cAAc,CAAC,kCAAkC,EAAE,IAAI,EAAE,OAAO,CAAC,CAuBnE;CACF"}

package/dist/core/runtime/OpenToolCallCloser.js

@@ -0,0 +1,92 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/core/runtime/OpenToolCallCloser.ts
+var OpenToolCallCloser_exports = {};
+__export(OpenToolCallCloser_exports, {
+  OpenToolCallCloser: () => OpenToolCallCloser,
+  getClosableOpenToolCallIds: () => getClosableOpenToolCallIds
+});
+module.exports = __toCommonJS(OpenToolCallCloser_exports);
+var import_AgentThread = require("./AgentThread.types.js");
+var import_contextUsage = require("./contextUsage.js");
+var import_contextUtils = require("./contextUtils.js");
+var DUMMY_TOOL_MESSAGE_CONTENT = JSON.stringify({
+  error: "Tool call was not executed. Please retry this tool call."
+});
+function getClosableOpenToolCallIds(context) {
+  const lastIdx = context.findLastIndex(
+    (msg) => (0, import_contextUtils.isLLMContextMessage)(msg) && msg.role === "assistant" && !!msg.tool_calls?.length
+  );
+  if (lastIdx === -1) {
+    return /* @__PURE__ */ new Set();
+  }
+  const lastAssistant = context[lastIdx];
+  if (lastAssistant === void 0 || !(0, import_contextUtils.isLLMContextMessage)(lastAssistant) || lastAssistant.role !== "assistant") {
+    throw new Error("Unreachable");
+  }
+  if (!lastAssistant.tool_calls) {
+    return /* @__PURE__ */ new Set();
+  }
+  if (lastAssistant.tool_calls.some(
+    (tc) => tc.tool_info.is_approval_required === true || tc.tool_info.is_client_side === true
+  )) {
+    return /* @__PURE__ */ new Set();
+  }
+  const requestedIds = new Set(
+    lastAssistant.tool_calls.filter((tc) => tc.tool_info.is_thread_creation !== true).map((tc) => tc.id)
+  );
+  const messagesAfterAssistant = context.slice(lastIdx + 1);
+  for (const msg of messagesAfterAssistant) {
+    if ((0, import_contextUtils.isLLMContextMessage)(msg) && msg.role === "tool") {
+      requestedIds.delete(msg.tool_call_id);
+    }
+  }
+  return requestedIds;
+}
+var OpenToolCallCloser = class {
+  // eslint-disable-next-line @typescript-eslint/require-await -- async *: AsyncIterable contract; body is sync
+  async *processPreSend(execution) {
+    const requestedIds = getClosableOpenToolCallIds(execution.context);
+    if (requestedIds.size === 0) {
+      return;
+    }
+    const dummyToolMessages = [...requestedIds].map((toolCallId) => ({
+      role: "tool",
+      tool_call_id: toolCallId,
+      content: DUMMY_TOOL_MESSAGE_CONTENT
+    }));
+    const currentContextUsage = (0, import_contextUsage.mergeCurrentContextUsage)(
+      execution.currentContextUsage,
+      (0, import_contextUtils.estimateTokensForContextMessages)(dummyToolMessages)
+    );
+    yield {
+      type: import_AgentThread.InternalEventType.AGENT_CONTEXT_APPEND,
+      context: dummyToolMessages,
+      output: [],
+      current_context_usage: currentContextUsage
+    };
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  OpenToolCallCloser,
+  getClosableOpenToolCallIds
+});
+//# sourceMappingURL=OpenToolCallCloser.js.map

package/dist/core/runtime/OpenToolCallCloser.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/runtime/OpenToolCallCloser.ts"],"sourcesContent":["/**\n * OpenToolCallCloser - Context processor that closes unresolved tool calls.\n *\n * When an assistant message contains tool_calls but corresponding tool response\n * messages are missing (e.g., due to a failed or interrupted tool execution),\n * LLMs will reject the request. This processor detects such cases in the\n * **last** assistant message and appends dummy tool responses so the\n * conversation can continue.\n *\n * Uses is_thread_creation from InternalToolCallInfo to identify sub-agent\n * tool calls (which should remain open); does NOT use SUB_AGENTS_SERVER_ID.\n * Older persisted contexts without is_thread_creation follow the ordinary\n * non-thread-creation path (no compatibility fallback).\n */\nimport type {\n  AgentContextProcessorAppendContext,\n  AgentThreadExecutionContext,\n  PreSendContextProcessor,\n} from '../capabilities/AgentContextProcessor';\nimport type { InternalEnrichedAssistantMessage, LLMToolMessage } from '../llm/LLMTypes';\nimport type { ContextMessage } from './AgentThread.types';\nimport { InternalEventType } from './AgentThread.types';\nimport { mergeCurrentContextUsage } from './contextUsage';\nimport { estimateTokensForContextMessages, isLLMContextMessage } from './contextUtils';\n\nconst DUMMY_TOOL_MESSAGE_CONTENT = JSON.stringify({\n  error: 'Tool call was not executed. Please retry this tool call.',\n});\n\nexport function getClosableOpenToolCallIds(context: ContextMessage[]): Set<string> {\n  const lastIdx = context.findLastIndex(\n    (msg): msg is InternalEnrichedAssistantMessage =>\n      isLLMContextMessage(msg) && msg.role === 'assistant' && !!msg.tool_calls?.length,\n  );\n  if (lastIdx === -1) {\n    return new Set();\n  }\n\n  const lastAssistant = context[lastIdx];\n  if (lastAssistant === undefined || !isLLMContextMessage(lastAssistant) || lastAssistant.role !== 'assistant') {\n    throw new Error('Unreachable');\n  }\n  if (!lastAssistant.tool_calls) {\n    return new Set();\n  }\n\n  if (\n    lastAssistant.tool_calls.some(\n      tc => tc.tool_info.is_approval_required === true || tc.tool_info.is_client_side === true,\n    )\n  ) {\n    return new Set();\n  }\n\n  const requestedIds = new Set(\n    lastAssistant.tool_calls.filter(tc => tc.tool_info.is_thread_creation !== true).map(tc => tc.id),\n  );\n\n  const messagesAfterAssistant = context.slice(lastIdx + 1);\n  for (const msg of messagesAfterAssistant) {\n    if (isLLMContextMessage(msg) && msg.role === 'tool') {\n      requestedIds.delete(msg.tool_call_id);\n    }\n  }\n\n  return requestedIds;\n}\n\nexport class OpenToolCallCloser implements PreSendContextProcessor {\n  // eslint-disable-next-line @typescript-eslint/require-await -- async *: AsyncIterable contract; body is sync\n  async *processPreSend(\n    execution: Readonly<AgentThreadExecutionContext>,\n  ): AsyncGenerator<AgentContextProcessorAppendContext, void, unknown> {\n    const requestedIds = getClosableOpenToolCallIds(execution.context);\n    if (requestedIds.size === 0) {\n      return;\n    }\n\n    const dummyToolMessages: LLMToolMessage[] = [...requestedIds].map(toolCallId => ({\n      role: 'tool',\n      tool_call_id: toolCallId,\n      content: DUMMY_TOOL_MESSAGE_CONTENT,\n    }));\n\n    const currentContextUsage = mergeCurrentContextUsage(\n      execution.currentContextUsage,\n      estimateTokensForContextMessages(dummyToolMessages),\n    );\n\n    yield {\n      type: InternalEventType.AGENT_CONTEXT_APPEND,\n      context: dummyToolMessages,\n      output: [],\n      current_context_usage: currentContextUsage,\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAqBA,yBAAkC;AAClC,0BAAyC;AACzC,0BAAsE;AAEtE,IAAM,6BAA6B,KAAK,UAAU;AAAA,EAChD,OAAO;AACT,CAAC;AAEM,SAAS,2BAA2B,SAAwC;AACjF,QAAM,UAAU,QAAQ;AAAA,IACtB,CAAC,YACC,yCAAoB,GAAG,KAAK,IAAI,SAAS,eAAe,CAAC,CAAC,IAAI,YAAY;AAAA,EAC9E;AACA,MAAI,YAAY,IAAI;AAClB,WAAO,oBAAI,IAAI;AAAA,EACjB;AAEA,QAAM,gBAAgB,QAAQ,OAAO;AACrC,MAAI,kBAAkB,UAAa,KAAC,yCAAoB,aAAa,KAAK,cAAc,SAAS,aAAa;AAC5G,UAAM,IAAI,MAAM,aAAa;AAAA,EAC/B;AACA,MAAI,CAAC,cAAc,YAAY;AAC7B,WAAO,oBAAI,IAAI;AAAA,EACjB;AAEA,MACE,cAAc,WAAW;AAAA,IACvB,QAAM,GAAG,UAAU,yBAAyB,QAAQ,GAAG,UAAU,mBAAmB;AAAA,EACtF,GACA;AACA,WAAO,oBAAI,IAAI;AAAA,EACjB;AAEA,QAAM,eAAe,IAAI;AAAA,IACvB,cAAc,WAAW,OAAO,QAAM,GAAG,UAAU,uBAAuB,IAAI,EAAE,IAAI,QAAM,GAAG,EAAE;AAAA,EACjG;AAEA,QAAM,yBAAyB,QAAQ,MAAM,UAAU,CAAC;AACxD,aAAW,OAAO,wBAAwB;AACxC,YAAI,yCAAoB,GAAG,KAAK,IAAI,SAAS,QAAQ;AACnD,mBAAa,OAAO,IAAI,YAAY;AAAA,IACtC;AAAA,EACF;AAEA,SAAO;AACT;AAEO,IAAM,qBAAN,MAA4D;AAAA;AAAA,EAEjE,OAAO,eACL,WACmE;AACnE,UAAM,eAAe,2BAA2B,UAAU,OAAO;AACjE,QAAI,aAAa,SAAS,GAAG;AAC3B;AAAA,IACF;AAEA,UAAM,oBAAsC,CAAC,GAAG,YAAY,EAAE,IAAI,iBAAe;AAAA,MAC/E,MAAM;AAAA,MACN,cAAc;AAAA,MACd,SAAS;AAAA,IACX,EAAE;AAEF,UAAM,0BAAsB;AAAA,MAC1B,UAAU;AAAA,UACV,sDAAiC,iBAAiB;AAAA,IACpD;AAEA,UAAM;AAAA,MACJ,MAAM,qCAAkB;AAAA,MACxB,SAAS;AAAA,MACT,QAAQ,CAAC;AAAA,MACT,uBAAuB;AAAA,IACzB;AAAA,EACF;AACF;","names":[]}

package/dist/core/runtime/OpenToolCallCloser.mjs

@@ -0,0 +1,66 @@
+// src/core/runtime/OpenToolCallCloser.ts
+import { InternalEventType } from "./AgentThread.types.mjs";
+import { mergeCurrentContextUsage } from "./contextUsage.mjs";
+import { estimateTokensForContextMessages, isLLMContextMessage } from "./contextUtils.mjs";
+var DUMMY_TOOL_MESSAGE_CONTENT = JSON.stringify({
+  error: "Tool call was not executed. Please retry this tool call."
+});
+function getClosableOpenToolCallIds(context) {
+  const lastIdx = context.findLastIndex(
+    (msg) => isLLMContextMessage(msg) && msg.role === "assistant" && !!msg.tool_calls?.length
+  );
+  if (lastIdx === -1) {
+    return /* @__PURE__ */ new Set();
+  }
+  const lastAssistant = context[lastIdx];
+  if (lastAssistant === void 0 || !isLLMContextMessage(lastAssistant) || lastAssistant.role !== "assistant") {
+    throw new Error("Unreachable");
+  }
+  if (!lastAssistant.tool_calls) {
+    return /* @__PURE__ */ new Set();
+  }
+  if (lastAssistant.tool_calls.some(
+    (tc) => tc.tool_info.is_approval_required === true || tc.tool_info.is_client_side === true
+  )) {
+    return /* @__PURE__ */ new Set();
+  }
+  const requestedIds = new Set(
+    lastAssistant.tool_calls.filter((tc) => tc.tool_info.is_thread_creation !== true).map((tc) => tc.id)
+  );
+  const messagesAfterAssistant = context.slice(lastIdx + 1);
+  for (const msg of messagesAfterAssistant) {
+    if (isLLMContextMessage(msg) && msg.role === "tool") {
+      requestedIds.delete(msg.tool_call_id);
+    }
+  }
+  return requestedIds;
+}
+var OpenToolCallCloser = class {
+  // eslint-disable-next-line @typescript-eslint/require-await -- async *: AsyncIterable contract; body is sync
+  async *processPreSend(execution) {
+    const requestedIds = getClosableOpenToolCallIds(execution.context);
+    if (requestedIds.size === 0) {
+      return;
+    }
+    const dummyToolMessages = [...requestedIds].map((toolCallId) => ({
+      role: "tool",
+      tool_call_id: toolCallId,
+      content: DUMMY_TOOL_MESSAGE_CONTENT
+    }));
+    const currentContextUsage = mergeCurrentContextUsage(
+      execution.currentContextUsage,
+      estimateTokensForContextMessages(dummyToolMessages)
+    );
+    yield {
+      type: InternalEventType.AGENT_CONTEXT_APPEND,
+      context: dummyToolMessages,
+      output: [],
+      current_context_usage: currentContextUsage
+    };
+  }
+};
+export {
+  OpenToolCallCloser,
+  getClosableOpenToolCallIds
+};
+//# sourceMappingURL=OpenToolCallCloser.mjs.map

package/dist/core/runtime/OpenToolCallCloser.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/runtime/OpenToolCallCloser.ts"],"sourcesContent":["/**\n * OpenToolCallCloser - Context processor that closes unresolved tool calls.\n *\n * When an assistant message contains tool_calls but corresponding tool response\n * messages are missing (e.g., due to a failed or interrupted tool execution),\n * LLMs will reject the request. This processor detects such cases in the\n * **last** assistant message and appends dummy tool responses so the\n * conversation can continue.\n *\n * Uses is_thread_creation from InternalToolCallInfo to identify sub-agent\n * tool calls (which should remain open); does NOT use SUB_AGENTS_SERVER_ID.\n * Older persisted contexts without is_thread_creation follow the ordinary\n * non-thread-creation path (no compatibility fallback).\n */\nimport type {\n  AgentContextProcessorAppendContext,\n  AgentThreadExecutionContext,\n  PreSendContextProcessor,\n} from '../capabilities/AgentContextProcessor';\nimport type { InternalEnrichedAssistantMessage, LLMToolMessage } from '../llm/LLMTypes';\nimport type { ContextMessage } from './AgentThread.types';\nimport { InternalEventType } from './AgentThread.types';\nimport { mergeCurrentContextUsage } from './contextUsage';\nimport { estimateTokensForContextMessages, isLLMContextMessage } from './contextUtils';\n\nconst DUMMY_TOOL_MESSAGE_CONTENT = JSON.stringify({\n  error: 'Tool call was not executed. Please retry this tool call.',\n});\n\nexport function getClosableOpenToolCallIds(context: ContextMessage[]): Set<string> {\n  const lastIdx = context.findLastIndex(\n    (msg): msg is InternalEnrichedAssistantMessage =>\n      isLLMContextMessage(msg) && msg.role === 'assistant' && !!msg.tool_calls?.length,\n  );\n  if (lastIdx === -1) {\n    return new Set();\n  }\n\n  const lastAssistant = context[lastIdx];\n  if (lastAssistant === undefined || !isLLMContextMessage(lastAssistant) || lastAssistant.role !== 'assistant') {\n    throw new Error('Unreachable');\n  }\n  if (!lastAssistant.tool_calls) {\n    return new Set();\n  }\n\n  if (\n    lastAssistant.tool_calls.some(\n      tc => tc.tool_info.is_approval_required === true || tc.tool_info.is_client_side === true,\n    )\n  ) {\n    return new Set();\n  }\n\n  const requestedIds = new Set(\n    lastAssistant.tool_calls.filter(tc => tc.tool_info.is_thread_creation !== true).map(tc => tc.id),\n  );\n\n  const messagesAfterAssistant = context.slice(lastIdx + 1);\n  for (const msg of messagesAfterAssistant) {\n    if (isLLMContextMessage(msg) && msg.role === 'tool') {\n      requestedIds.delete(msg.tool_call_id);\n    }\n  }\n\n  return requestedIds;\n}\n\nexport class OpenToolCallCloser implements PreSendContextProcessor {\n  // eslint-disable-next-line @typescript-eslint/require-await -- async *: AsyncIterable contract; body is sync\n  async *processPreSend(\n    execution: Readonly<AgentThreadExecutionContext>,\n  ): AsyncGenerator<AgentContextProcessorAppendContext, void, unknown> {\n    const requestedIds = getClosableOpenToolCallIds(execution.context);\n    if (requestedIds.size === 0) {\n      return;\n    }\n\n    const dummyToolMessages: LLMToolMessage[] = [...requestedIds].map(toolCallId => ({\n      role: 'tool',\n      tool_call_id: toolCallId,\n      content: DUMMY_TOOL_MESSAGE_CONTENT,\n    }));\n\n    const currentContextUsage = mergeCurrentContextUsage(\n      execution.currentContextUsage,\n      estimateTokensForContextMessages(dummyToolMessages),\n    );\n\n    yield {\n      type: InternalEventType.AGENT_CONTEXT_APPEND,\n      context: dummyToolMessages,\n      output: [],\n      current_context_usage: currentContextUsage,\n    };\n  }\n}\n"],"mappings":";AAqBA,SAAS,yBAAyB;AAClC,SAAS,gCAAgC;AACzC,SAAS,kCAAkC,2BAA2B;AAEtE,IAAM,6BAA6B,KAAK,UAAU;AAAA,EAChD,OAAO;AACT,CAAC;AAEM,SAAS,2BAA2B,SAAwC;AACjF,QAAM,UAAU,QAAQ;AAAA,IACtB,CAAC,QACC,oBAAoB,GAAG,KAAK,IAAI,SAAS,eAAe,CAAC,CAAC,IAAI,YAAY;AAAA,EAC9E;AACA,MAAI,YAAY,IAAI;AAClB,WAAO,oBAAI,IAAI;AAAA,EACjB;AAEA,QAAM,gBAAgB,QAAQ,OAAO;AACrC,MAAI,kBAAkB,UAAa,CAAC,oBAAoB,aAAa,KAAK,cAAc,SAAS,aAAa;AAC5G,UAAM,IAAI,MAAM,aAAa;AAAA,EAC/B;AACA,MAAI,CAAC,cAAc,YAAY;AAC7B,WAAO,oBAAI,IAAI;AAAA,EACjB;AAEA,MACE,cAAc,WAAW;AAAA,IACvB,QAAM,GAAG,UAAU,yBAAyB,QAAQ,GAAG,UAAU,mBAAmB;AAAA,EACtF,GACA;AACA,WAAO,oBAAI,IAAI;AAAA,EACjB;AAEA,QAAM,eAAe,IAAI;AAAA,IACvB,cAAc,WAAW,OAAO,QAAM,GAAG,UAAU,uBAAuB,IAAI,EAAE,IAAI,QAAM,GAAG,EAAE;AAAA,EACjG;AAEA,QAAM,yBAAyB,QAAQ,MAAM,UAAU,CAAC;AACxD,aAAW,OAAO,wBAAwB;AACxC,QAAI,oBAAoB,GAAG,KAAK,IAAI,SAAS,QAAQ;AACnD,mBAAa,OAAO,IAAI,YAAY;AAAA,IACtC;AAAA,EACF;AAEA,SAAO;AACT;AAEO,IAAM,qBAAN,MAA4D;AAAA;AAAA,EAEjE,OAAO,eACL,WACmE;AACnE,UAAM,eAAe,2BAA2B,UAAU,OAAO;AACjE,QAAI,aAAa,SAAS,GAAG;AAC3B;AAAA,IACF;AAEA,UAAM,oBAAsC,CAAC,GAAG,YAAY,EAAE,IAAI,iBAAe;AAAA,MAC/E,MAAM;AAAA,MACN,cAAc;AAAA,MACd,SAAS;AAAA,IACX,EAAE;AAEF,UAAM,sBAAsB;AAAA,MAC1B,UAAU;AAAA,MACV,iCAAiC,iBAAiB;AAAA,IACpD;AAEA,UAAM;AAAA,MACJ,MAAM,kBAAkB;AAAA,MACxB,SAAS;AAAA,MACT,QAAQ,CAAC;AAAA,MACT,uBAAuB;AAAA,IACzB;AAAA,EACF;AACF;","names":[]}

package/dist/core/runtime/UserInputMessage.d.ts

@@ -0,0 +1,16 @@
+import { type AgentInputUserMessage, type FileContentPart, type TextContentPart, type UserContentPart } from '../events/schema';
+import type { LLMUserMessage } from '../llm/LLMTypes';
+import type { Sandbox, SandboxInfo } from '../sandbox/Sandbox';
+import type { AgentSendInput } from './AgentThread.types';
+export type { AgentInputUserMessage, FileContentPart, TextContentPart, UserContentPart };
+export type AgentUserMessageContent = AgentInputUserMessage['content'];
+export type AgentUserContentPart = UserContentPart;
+export declare function isFileContentPart(part: UserContentPart): part is FileContentPart;
+export declare function isAgentInputUserMessage(msg: AgentSendInput): msg is AgentInputUserMessage;
+export declare function isEmptyMessageContent(content: AgentUserMessageContent): boolean;
+export interface ProcessAgentUserInputResult {
+    message: LLMUserMessage;
+    sandboxCreated?: SandboxInfo | undefined;
+}
+export declare function processAgentUserInput(msg: AgentInputUserMessage, sandbox: Sandbox | undefined): Promise<ProcessAgentUserInputResult>;
+//# sourceMappingURL=UserInputMessage.d.ts.map

package/dist/core/runtime/UserInputMessage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"UserInputMessage.d.ts","sourceRoot":"","sources":["../../../src/core/runtime/UserInputMessage.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,qBAAqB,EAC1B,KAAK,eAAe,EACpB,KAAK,eAAe,EACpB,KAAK,eAAe,EACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAG1D,YAAY,EAAE,qBAAqB,EAAE,eAAe,EAAE,eAAe,EAAE,eAAe,EAAE,CAAC;AAEzF,MAAM,MAAM,uBAAuB,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;AACvE,MAAM,MAAM,oBAAoB,GAAG,eAAe,CAAC;AAEnD,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,eAAe,GAAG,IAAI,IAAI,eAAe,CAEhF;AAED,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,cAAc,GAAG,GAAG,IAAI,qBAAqB,CAEzF;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAE/E;AA+ED,MAAM,WAAW,2BAA2B;IAC1C,OAAO,EAAE,cAAc,CAAC;IACxB,cAAc,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;CAC1C;AAED,wBAAsB,qBAAqB,CACzC,GAAG,EAAE,qBAAqB,EAC1B,OAAO,EAAE,OAAO,GAAG,SAAS,GAC3B,OAAO,CAAC,2BAA2B,CAAC,CAoFtC"}

package/dist/core/runtime/UserInputMessage.js

@@ -0,0 +1,183 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/core/runtime/UserInputMessage.ts
+var UserInputMessage_exports = {};
+__export(UserInputMessage_exports, {
+  isAgentInputUserMessage: () => isAgentInputUserMessage,
+  isEmptyMessageContent: () => isEmptyMessageContent,
+  isFileContentPart: () => isFileContentPart,
+  processAgentUserInput: () => processAgentUserInput
+});
+module.exports = __toCommonJS(UserInputMessage_exports);
+var import_errors = require("../errors.js");
+var import_schema = require("../events/schema.js");
+var import_contextUtils = require("./contextUtils.js");
+function isFileContentPart(part) {
+  return part.type === "file";
+}
+function isAgentInputUserMessage(msg) {
+  return "type" in msg && msg.type === import_schema.EventType.USER_MESSAGE;
+}
+function isEmptyMessageContent(content) {
+  return typeof content === "string" ? content.trim().length === 0 : content.length === 0;
+}
+var INLINE_MIME_PREFIXES = ["image/", "application/pdf"];
+function isInlineMime(mime) {
+  return INLINE_MIME_PREFIXES.some((prefix) => mime.startsWith(prefix));
+}
+function isImageMime(mime) {
+  return mime.startsWith("image/");
+}
+function parseMimeFromFileData(fileData) {
+  const match = /^data:([^;,]+)/.exec(fileData);
+  if (!match?.[1]) {
+    throw new import_errors.InvalidFileInputError("File data URI is missing or has unparseable MIME type");
+  }
+  return match[1];
+}
+function extractBase64(dataUri) {
+  const commaIndex = dataUri.indexOf(",");
+  if (commaIndex === -1) {
+    throw new import_errors.InvalidFileInputError("File data URI has no base64 payload");
+  }
+  const raw = dataUri.substring(commaIndex + 1);
+  if (!raw) {
+    throw new import_errors.InvalidFileInputError("File data URI has empty base64 payload");
+  }
+  const buffer = Buffer.from(raw, "base64");
+  if (buffer.length === 0) {
+    throw new import_errors.InvalidFileInputError("File data URI decoded to zero bytes");
+  }
+  return buffer;
+}
+function validatePublicFilename(filename) {
+  if (!filename.length) {
+    throw new import_errors.InvalidFileInputError("File name must not be empty");
+  }
+  const normalized = filename.replace(/\\/g, "/");
+  const parts = normalized.split("/");
+  for (const part of parts) {
+    if (part === "..") {
+      throw new import_errors.InvalidFileInputError(`File name contains path traversal: ${filename}`);
+    }
+  }
+}
+function buildUploadedFilesAnnouncement(files) {
+  const entries = files.map(
+    (f, i) => `[file_${String(i + 1)}]
+  filename: ${f.filename}
+  path: ${f.path}
+  mime: ${f.mime}
+  size: ${String(f.size)} bytes`
+  ).join("\n\n");
+  return (0, import_contextUtils.internalSystemTag)(`Uploaded files (${String(files.length)}):
+${entries}`);
+}
+function partitionContent(parts) {
+  const fileParts = [];
+  const textParts = [];
+  for (const part of parts) {
+    if (isFileContentPart(part)) {
+      fileParts.push(part);
+    } else {
+      textParts.push(part);
+    }
+  }
+  return { fileParts, textParts };
+}
+async function processAgentUserInput(msg, sandbox) {
+  if (typeof msg.content === "string") {
+    return { message: { role: "user", content: msg.content }, sandboxCreated: void 0 };
+  }
+  const { fileParts, textParts } = partitionContent(msg.content);
+  if (fileParts.length === 0) {
+    return { message: { role: "user", content: textParts }, sandboxCreated: void 0 };
+  }
+  const modelParts = [];
+  const filesToUploadToSandbox = [];
+  for (const filePart of fileParts) {
+    const filename = filePart.name.replace(/^\/+/, "");
+    if (!filename.length) {
+      throw new import_errors.InvalidFileInputError("File name must not be empty");
+    }
+    validatePublicFilename(filename);
+    const file_data = filePart.data;
+    const mime = parseMimeFromFileData(file_data);
+    if (isInlineMime(mime)) {
+      modelParts.push(
+        isImageMime(mime) ? { type: "image_url", image_url: { url: file_data } } : { type: "file", file: { filename, file_data } }
+      );
+    } else {
+      const buffer = extractBase64(file_data);
+      filesToUploadToSandbox.push({ filename, buffer, mime });
+    }
+  }
+  let sandboxCreated;
+  if (filesToUploadToSandbox.length > 0) {
+    if (!sandbox) {
+      throw new import_errors.AgentSandboxRequiredError(
+        "File uploads requiring sandbox are not supported: no sandbox configured for this agent"
+      );
+    }
+    const uploadResults = await Promise.allSettled(
+      filesToUploadToSandbox.map(
+        (file) => sandbox.uploadUserFile({
+          fileName: file.filename,
+          content: file.buffer,
+          mime: file.mime
+        })
+      )
+    );
+    const uploadedFiles = [];
+    for (let i = 0; i < uploadResults.length; i++) {
+      const result = uploadResults[i];
+      const file = filesToUploadToSandbox[i];
+      if (result === void 0 || file === void 0) {
+        throw new Error("Unreachable: uploadResults/files length mismatch");
+      }
+      if (result.status === "fulfilled") {
+        if (result.value.sandboxCreated) {
+          sandboxCreated = result.value.sandboxCreated;
+        }
+        uploadedFiles.push({
+          filename: file.filename,
+          path: result.value.filePath,
+          mime: file.mime,
+          size: file.buffer.length
+        });
+      } else {
+        throw result.reason;
+      }
+    }
+    if (uploadedFiles.length > 0) {
+      modelParts.push({ type: "text", text: buildUploadedFilesAnnouncement(uploadedFiles) });
+    }
+  }
+  modelParts.push(...textParts);
+  return { message: { role: "user", content: modelParts }, sandboxCreated };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  isAgentInputUserMessage,
+  isEmptyMessageContent,
+  isFileContentPart,
+  processAgentUserInput
+});
+//# sourceMappingURL=UserInputMessage.js.map

package/dist/core/runtime/UserInputMessage.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/runtime/UserInputMessage.ts"],"sourcesContent":["import type { ChatCompletionContentPart } from 'openai/resources/chat';\nimport { AgentSandboxRequiredError, InvalidFileInputError } from '../errors';\nimport {\n  EventType,\n  type AgentInputUserMessage,\n  type FileContentPart,\n  type TextContentPart,\n  type UserContentPart,\n} from '../events/schema';\nimport type { LLMUserMessage } from '../llm/LLMTypes';\nimport type { Sandbox, SandboxInfo } from '../sandbox/Sandbox';\nimport type { AgentSendInput } from './AgentThread.types';\nimport { internalSystemTag } from './contextUtils';\n\nexport type { AgentInputUserMessage, FileContentPart, TextContentPart, UserContentPart };\n\nexport type AgentUserMessageContent = AgentInputUserMessage['content'];\nexport type AgentUserContentPart = UserContentPart;\n\nexport function isFileContentPart(part: UserContentPart): part is FileContentPart {\n  return part.type === 'file';\n}\n\nexport function isAgentInputUserMessage(msg: AgentSendInput): msg is AgentInputUserMessage {\n  return 'type' in msg && msg.type === EventType.USER_MESSAGE;\n}\n\nexport function isEmptyMessageContent(content: AgentUserMessageContent): boolean {\n  return typeof content === 'string' ? content.trim().length === 0 : content.length === 0;\n}\n\nconst INLINE_MIME_PREFIXES = ['image/', 'application/pdf'];\n\nfunction isInlineMime(mime: string): boolean {\n  return INLINE_MIME_PREFIXES.some(prefix => mime.startsWith(prefix));\n}\n\nfunction isImageMime(mime: string): boolean {\n  return mime.startsWith('image/');\n}\n\nfunction parseMimeFromFileData(fileData: string): string {\n  const match = /^data:([^;,]+)/.exec(fileData);\n  if (!match?.[1]) {\n    throw new InvalidFileInputError('File data URI is missing or has unparseable MIME type');\n  }\n  return match[1];\n}\n\nfunction extractBase64(dataUri: string): Buffer {\n  const commaIndex = dataUri.indexOf(',');\n  if (commaIndex === -1) {\n    throw new InvalidFileInputError('File data URI has no base64 payload');\n  }\n  const raw = dataUri.substring(commaIndex + 1);\n  if (!raw) {\n    throw new InvalidFileInputError('File data URI has empty base64 payload');\n  }\n  const buffer = Buffer.from(raw, 'base64');\n  if (buffer.length === 0) {\n    throw new InvalidFileInputError('File data URI decoded to zero bytes');\n  }\n  return buffer;\n}\n\nfunction validatePublicFilename(filename: string): void {\n  if (!filename.length) {\n    throw new InvalidFileInputError('File name must not be empty');\n  }\n  const normalized = filename.replace(/\\\\/g, '/');\n  const parts = normalized.split('/');\n  for (const part of parts) {\n    if (part === '..') {\n      throw new InvalidFileInputError(`File name contains path traversal: ${filename}`);\n    }\n  }\n}\n\nfunction buildUploadedFilesAnnouncement(\n  files: { filename: string; path: string; mime: string; size: number }[],\n): string {\n  const entries = files\n    .map(\n      (f, i) =>\n        `[file_${String(i + 1)}]\\n  filename: ${f.filename}\\n  path: ${f.path}\\n  mime: ${f.mime}\\n  size: ${String(f.size)} bytes`,\n    )\n    .join('\\n\\n');\n  return internalSystemTag(`Uploaded files (${String(files.length)}):\\n${entries}`);\n}\n\nfunction partitionContent(parts: readonly AgentUserContentPart[]): {\n  fileParts: FileContentPart[];\n  textParts: TextContentPart[];\n} {\n  const fileParts: FileContentPart[] = [];\n  const textParts: TextContentPart[] = [];\n\n  for (const part of parts) {\n    if (isFileContentPart(part)) {\n      fileParts.push(part);\n    } else {\n      textParts.push(part);\n    }\n  }\n\n  return { fileParts, textParts };\n}\n\nexport interface ProcessAgentUserInputResult {\n  message: LLMUserMessage;\n  sandboxCreated?: SandboxInfo | undefined;\n}\n\nexport async function processAgentUserInput(\n  msg: AgentInputUserMessage,\n  sandbox: Sandbox | undefined,\n): Promise<ProcessAgentUserInputResult> {\n  if (typeof msg.content === 'string') {\n    return { message: { role: 'user', content: msg.content }, sandboxCreated: undefined };\n  }\n\n  const { fileParts, textParts } = partitionContent(msg.content);\n\n  if (fileParts.length === 0) {\n    return { message: { role: 'user', content: textParts }, sandboxCreated: undefined };\n  }\n\n  const modelParts: ChatCompletionContentPart[] = [];\n  const filesToUploadToSandbox: { filename: string; buffer: Buffer; mime: string }[] = [];\n\n  for (const filePart of fileParts) {\n    const filename = filePart.name.replace(/^\\/+/, '');\n    if (!filename.length) {\n      throw new InvalidFileInputError('File name must not be empty');\n    }\n    validatePublicFilename(filename);\n\n    const file_data = filePart.data;\n    const mime = parseMimeFromFileData(file_data);\n\n    if (isInlineMime(mime)) {\n      modelParts.push(\n        isImageMime(mime)\n          ? { type: 'image_url', image_url: { url: file_data } }\n          : { type: 'file', file: { filename, file_data } },\n      );\n    } else {\n      const buffer = extractBase64(file_data);\n      filesToUploadToSandbox.push({ filename, buffer, mime });\n    }\n  }\n\n  let sandboxCreated: SandboxInfo | undefined;\n\n  if (filesToUploadToSandbox.length > 0) {\n    if (!sandbox) {\n      throw new AgentSandboxRequiredError(\n        'File uploads requiring sandbox are not supported: no sandbox configured for this agent',\n      );\n    }\n\n    const uploadResults = await Promise.allSettled(\n      filesToUploadToSandbox.map(file =>\n        sandbox.uploadUserFile({\n          fileName: file.filename,\n          content: file.buffer,\n          mime: file.mime,\n        }),\n      ),\n    );\n\n    const uploadedFiles: { filename: string; path: string; mime: string; size: number }[] = [];\n    for (let i = 0; i < uploadResults.length; i++) {\n      const result = uploadResults[i];\n      const file = filesToUploadToSandbox[i];\n      if (result === undefined || file === undefined) {\n        throw new Error('Unreachable: uploadResults/files length mismatch');\n      }\n      if (result.status === 'fulfilled') {\n        if (result.value.sandboxCreated) {\n          sandboxCreated = result.value.sandboxCreated;\n        }\n        uploadedFiles.push({\n          filename: file.filename,\n          path: result.value.filePath,\n          mime: file.mime,\n          size: file.buffer.length,\n        });\n      } else {\n        throw result.reason;\n      }\n    }\n    if (uploadedFiles.length > 0) {\n      modelParts.push({ type: 'text', text: buildUploadedFilesAnnouncement(uploadedFiles) });\n    }\n  }\n\n  modelParts.push(...textParts);\n\n  return { message: { role: 'user', content: modelParts }, sandboxCreated };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,oBAAiE;AACjE,oBAMO;AAIP,0BAAkC;AAO3B,SAAS,kBAAkB,MAAgD;AAChF,SAAO,KAAK,SAAS;AACvB;AAEO,SAAS,wBAAwB,KAAmD;AACzF,SAAO,UAAU,OAAO,IAAI,SAAS,wBAAU;AACjD;AAEO,SAAS,sBAAsB,SAA2C;AAC/E,SAAO,OAAO,YAAY,WAAW,QAAQ,KAAK,EAAE,WAAW,IAAI,QAAQ,WAAW;AACxF;AAEA,IAAM,uBAAuB,CAAC,UAAU,iBAAiB;AAEzD,SAAS,aAAa,MAAuB;AAC3C,SAAO,qBAAqB,KAAK,YAAU,KAAK,WAAW,MAAM,CAAC;AACpE;AAEA,SAAS,YAAY,MAAuB;AAC1C,SAAO,KAAK,WAAW,QAAQ;AACjC;AAEA,SAAS,sBAAsB,UAA0B;AACvD,QAAM,QAAQ,iBAAiB,KAAK,QAAQ;AAC5C,MAAI,CAAC,QAAQ,CAAC,GAAG;AACf,UAAM,IAAI,oCAAsB,uDAAuD;AAAA,EACzF;AACA,SAAO,MAAM,CAAC;AAChB;AAEA,SAAS,cAAc,SAAyB;AAC9C,QAAM,aAAa,QAAQ,QAAQ,GAAG;AACtC,MAAI,eAAe,IAAI;AACrB,UAAM,IAAI,oCAAsB,qCAAqC;AAAA,EACvE;AACA,QAAM,MAAM,QAAQ,UAAU,aAAa,CAAC;AAC5C,MAAI,CAAC,KAAK;AACR,UAAM,IAAI,oCAAsB,wCAAwC;AAAA,EAC1E;AACA,QAAM,SAAS,OAAO,KAAK,KAAK,QAAQ;AACxC,MAAI,OAAO,WAAW,GAAG;AACvB,UAAM,IAAI,oCAAsB,qCAAqC;AAAA,EACvE;AACA,SAAO;AACT;AAEA,SAAS,uBAAuB,UAAwB;AACtD,MAAI,CAAC,SAAS,QAAQ;AACpB,UAAM,IAAI,oCAAsB,6BAA6B;AAAA,EAC/D;AACA,QAAM,aAAa,SAAS,QAAQ,OAAO,GAAG;AAC9C,QAAM,QAAQ,WAAW,MAAM,GAAG;AAClC,aAAW,QAAQ,OAAO;AACxB,QAAI,SAAS,MAAM;AACjB,YAAM,IAAI,oCAAsB,sCAAsC,QAAQ,EAAE;AAAA,IAClF;AAAA,EACF;AACF;AAEA,SAAS,+BACP,OACQ;AACR,QAAM,UAAU,MACb;AAAA,IACC,CAAC,GAAG,MACF,SAAS,OAAO,IAAI,CAAC,CAAC;AAAA,cAAkB,EAAE,QAAQ;AAAA,UAAa,EAAE,IAAI;AAAA,UAAa,EAAE,IAAI;AAAA,UAAa,OAAO,EAAE,IAAI,CAAC;AAAA,EACvH,EACC,KAAK,MAAM;AACd,aAAO,uCAAkB,mBAAmB,OAAO,MAAM,MAAM,CAAC;AAAA,EAAO,OAAO,EAAE;AAClF;AAEA,SAAS,iBAAiB,OAGxB;AACA,QAAM,YAA+B,CAAC;AACtC,QAAM,YAA+B,CAAC;AAEtC,aAAW,QAAQ,OAAO;AACxB,QAAI,kBAAkB,IAAI,GAAG;AAC3B,gBAAU,KAAK,IAAI;AAAA,IACrB,OAAO;AACL,gBAAU,KAAK,IAAI;AAAA,IACrB;AAAA,EACF;AAEA,SAAO,EAAE,WAAW,UAAU;AAChC;AAOA,eAAsB,sBACpB,KACA,SACsC;AACtC,MAAI,OAAO,IAAI,YAAY,UAAU;AACnC,WAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,SAAS,IAAI,QAAQ,GAAG,gBAAgB,OAAU;AAAA,EACtF;AAEA,QAAM,EAAE,WAAW,UAAU,IAAI,iBAAiB,IAAI,OAAO;AAE7D,MAAI,UAAU,WAAW,GAAG;AAC1B,WAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,SAAS,UAAU,GAAG,gBAAgB,OAAU;AAAA,EACpF;AAEA,QAAM,aAA0C,CAAC;AACjD,QAAM,yBAA+E,CAAC;AAEtF,aAAW,YAAY,WAAW;AAChC,UAAM,WAAW,SAAS,KAAK,QAAQ,QAAQ,EAAE;AACjD,QAAI,CAAC,SAAS,QAAQ;AACpB,YAAM,IAAI,oCAAsB,6BAA6B;AAAA,IAC/D;AACA,2BAAuB,QAAQ;AAE/B,UAAM,YAAY,SAAS;AAC3B,UAAM,OAAO,sBAAsB,SAAS;AAE5C,QAAI,aAAa,IAAI,GAAG;AACtB,iBAAW;AAAA,QACT,YAAY,IAAI,IACZ,EAAE,MAAM,aAAa,WAAW,EAAE,KAAK,UAAU,EAAE,IACnD,EAAE,MAAM,QAAQ,MAAM,EAAE,UAAU,UAAU,EAAE;AAAA,MACpD;AAAA,IACF,OAAO;AACL,YAAM,SAAS,cAAc,SAAS;AACtC,6BAAuB,KAAK,EAAE,UAAU,QAAQ,KAAK,CAAC;AAAA,IACxD;AAAA,EACF;AAEA,MAAI;AAEJ,MAAI,uBAAuB,SAAS,GAAG;AACrC,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,gBAAgB,MAAM,QAAQ;AAAA,MAClC,uBAAuB;AAAA,QAAI,UACzB,QAAQ,eAAe;AAAA,UACrB,UAAU,KAAK;AAAA,UACf,SAAS,KAAK;AAAA,UACd,MAAM,KAAK;AAAA,QACb,CAAC;AAAA,MACH;AAAA,IACF;AAEA,UAAM,gBAAkF,CAAC;AACzF,aAAS,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;AAC7C,YAAM,SAAS,cAAc,CAAC;AAC9B,YAAM,OAAO,uBAAuB,CAAC;AACrC,UAAI,WAAW,UAAa,SAAS,QAAW;AAC9C,cAAM,IAAI,MAAM,kDAAkD;AAAA,MACpE;AACA,UAAI,OAAO,WAAW,aAAa;AACjC,YAAI,OAAO,MAAM,gBAAgB;AAC/B,2BAAiB,OAAO,MAAM;AAAA,QAChC;AACA,sBAAc,KAAK;AAAA,UACjB,UAAU,KAAK;AAAA,UACf,MAAM,OAAO,MAAM;AAAA,UACnB,MAAM,KAAK;AAAA,UACX,MAAM,KAAK,OAAO;AAAA,QACpB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,OAAO;AAAA,MACf;AAAA,IACF;AACA,QAAI,cAAc,SAAS,GAAG;AAC5B,iBAAW,KAAK,EAAE,MAAM,QAAQ,MAAM,+BAA+B,aAAa,EAAE,CAAC;AAAA,IACvF;AAAA,EACF;AAEA,aAAW,KAAK,GAAG,SAAS;AAE5B,SAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,SAAS,WAAW,GAAG,eAAe;AAC1E;","names":[]}

package/dist/core/runtime/UserInputMessage.mjs

@@ -0,0 +1,157 @@
+// src/core/runtime/UserInputMessage.ts
+import { AgentSandboxRequiredError, InvalidFileInputError } from "../errors.mjs";
+import {
+  EventType
+} from "../events/schema.mjs";
+import { internalSystemTag } from "./contextUtils.mjs";
+function isFileContentPart(part) {
+  return part.type === "file";
+}
+function isAgentInputUserMessage(msg) {
+  return "type" in msg && msg.type === EventType.USER_MESSAGE;
+}
+function isEmptyMessageContent(content) {
+  return typeof content === "string" ? content.trim().length === 0 : content.length === 0;
+}
+var INLINE_MIME_PREFIXES = ["image/", "application/pdf"];
+function isInlineMime(mime) {
+  return INLINE_MIME_PREFIXES.some((prefix) => mime.startsWith(prefix));
+}
+function isImageMime(mime) {
+  return mime.startsWith("image/");
+}
+function parseMimeFromFileData(fileData) {
+  const match = /^data:([^;,]+)/.exec(fileData);
+  if (!match?.[1]) {
+    throw new InvalidFileInputError("File data URI is missing or has unparseable MIME type");
+  }
+  return match[1];
+}
+function extractBase64(dataUri) {
+  const commaIndex = dataUri.indexOf(",");
+  if (commaIndex === -1) {
+    throw new InvalidFileInputError("File data URI has no base64 payload");
+  }
+  const raw = dataUri.substring(commaIndex + 1);
+  if (!raw) {
+    throw new InvalidFileInputError("File data URI has empty base64 payload");
+  }
+  const buffer = Buffer.from(raw, "base64");
+  if (buffer.length === 0) {
+    throw new InvalidFileInputError("File data URI decoded to zero bytes");
+  }
+  return buffer;
+}
+function validatePublicFilename(filename) {
+  if (!filename.length) {
+    throw new InvalidFileInputError("File name must not be empty");
+  }
+  const normalized = filename.replace(/\\/g, "/");
+  const parts = normalized.split("/");
+  for (const part of parts) {
+    if (part === "..") {
+      throw new InvalidFileInputError(`File name contains path traversal: ${filename}`);
+    }
+  }
+}
+function buildUploadedFilesAnnouncement(files) {
+  const entries = files.map(
+    (f, i) => `[file_${String(i + 1)}]
+  filename: ${f.filename}
+  path: ${f.path}
+  mime: ${f.mime}
+  size: ${String(f.size)} bytes`
+  ).join("\n\n");
+  return internalSystemTag(`Uploaded files (${String(files.length)}):
+${entries}`);
+}
+function partitionContent(parts) {
+  const fileParts = [];
+  const textParts = [];
+  for (const part of parts) {
+    if (isFileContentPart(part)) {
+      fileParts.push(part);
+    } else {
+      textParts.push(part);
+    }
+  }
+  return { fileParts, textParts };
+}
+async function processAgentUserInput(msg, sandbox) {
+  if (typeof msg.content === "string") {
+    return { message: { role: "user", content: msg.content }, sandboxCreated: void 0 };
+  }
+  const { fileParts, textParts } = partitionContent(msg.content);
+  if (fileParts.length === 0) {
+    return { message: { role: "user", content: textParts }, sandboxCreated: void 0 };
+  }
+  const modelParts = [];
+  const filesToUploadToSandbox = [];
+  for (const filePart of fileParts) {
+    const filename = filePart.name.replace(/^\/+/, "");
+    if (!filename.length) {
+      throw new InvalidFileInputError("File name must not be empty");
+    }
+    validatePublicFilename(filename);
+    const file_data = filePart.data;
+    const mime = parseMimeFromFileData(file_data);
+    if (isInlineMime(mime)) {
+      modelParts.push(
+        isImageMime(mime) ? { type: "image_url", image_url: { url: file_data } } : { type: "file", file: { filename, file_data } }
+      );
+    } else {
+      const buffer = extractBase64(file_data);
+      filesToUploadToSandbox.push({ filename, buffer, mime });
+    }
+  }
+  let sandboxCreated;
+  if (filesToUploadToSandbox.length > 0) {
+    if (!sandbox) {
+      throw new AgentSandboxRequiredError(
+        "File uploads requiring sandbox are not supported: no sandbox configured for this agent"
+      );
+    }
+    const uploadResults = await Promise.allSettled(
+      filesToUploadToSandbox.map(
+        (file) => sandbox.uploadUserFile({
+          fileName: file.filename,
+          content: file.buffer,
+          mime: file.mime
+        })
+      )
+    );
+    const uploadedFiles = [];
+    for (let i = 0; i < uploadResults.length; i++) {
+      const result = uploadResults[i];
+      const file = filesToUploadToSandbox[i];
+      if (result === void 0 || file === void 0) {
+        throw new Error("Unreachable: uploadResults/files length mismatch");
+      }
+      if (result.status === "fulfilled") {
+        if (result.value.sandboxCreated) {
+          sandboxCreated = result.value.sandboxCreated;
+        }
+        uploadedFiles.push({
+          filename: file.filename,
+          path: result.value.filePath,
+          mime: file.mime,
+          size: file.buffer.length
+        });
+      } else {
+        throw result.reason;
+      }
+    }
+    if (uploadedFiles.length > 0) {
+      modelParts.push({ type: "text", text: buildUploadedFilesAnnouncement(uploadedFiles) });
+    }
+  }
+  modelParts.push(...textParts);
+  return { message: { role: "user", content: modelParts }, sandboxCreated };
+}
+export {
+  isAgentInputUserMessage,
+  isEmptyMessageContent,
+  isFileContentPart,
+  processAgentUserInput
+};
+//# sourceMappingURL=UserInputMessage.mjs.map