@osdk/aip-core 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +20 -0
- package/README.md +211 -0
- package/ai-sdk.d.ts +17 -0
- package/build/browser/ai-sdk/convert-prompt.js +136 -0
- package/build/browser/ai-sdk/convert-prompt.js.map +1 -0
- package/build/browser/ai-sdk/foundry-chat-language-model.js +386 -0
- package/build/browser/ai-sdk/foundry-chat-language-model.js.map +1 -0
- package/build/browser/ai-sdk/map-finish-reason.js +42 -0
- package/build/browser/ai-sdk/map-finish-reason.js.map +1 -0
- package/build/browser/ai-sdk/provider.js +62 -0
- package/build/browser/ai-sdk/provider.js.map +1 -0
- package/build/browser/ai-sdk/types.js +2 -0
- package/build/browser/ai-sdk/types.js.map +1 -0
- package/build/browser/generateText.js +76 -0
- package/build/browser/generateText.js.map +1 -0
- package/build/browser/index.js +23 -0
- package/build/browser/index.js.map +1 -0
- package/build/browser/internal/__test__/mockPlatformClient.js +36 -0
- package/build/browser/internal/__test__/mockPlatformClient.js.map +1 -0
- package/build/browser/internal/ids.js +23 -0
- package/build/browser/internal/ids.js.map +1 -0
- package/build/browser/internal/openAi.js +293 -0
- package/build/browser/internal/openAi.js.map +1 -0
- package/build/browser/internal/options.js +58 -0
- package/build/browser/internal/options.js.map +1 -0
- package/build/browser/internal/runStep.js +107 -0
- package/build/browser/internal/runStep.js.map +1 -0
- package/build/browser/internal/streamStep.js +223 -0
- package/build/browser/internal/streamStep.js.map +1 -0
- package/build/browser/internal/uiMessageBridge.js +55 -0
- package/build/browser/internal/uiMessageBridge.js.map +1 -0
- package/build/browser/lmsChatTransport.js +185 -0
- package/build/browser/lmsChatTransport.js.map +1 -0
- package/build/browser/model.js +74 -0
- package/build/browser/model.js.map +1 -0
- package/build/browser/public/ai-sdk.js +18 -0
- package/build/browser/public/ai-sdk.js.map +1 -0
- package/build/browser/streamText.js +210 -0
- package/build/browser/streamText.js.map +1 -0
- package/build/browser/types.js +2 -0
- package/build/browser/types.js.map +1 -0
- package/build/browser/uiMessage.js +34 -0
- package/build/browser/uiMessage.js.map +1 -0
- package/build/cjs/index.cjs +1012 -0
- package/build/cjs/index.cjs.map +1 -0
- package/build/cjs/index.d.cts +467 -0
- package/build/cjs/model-B5uS56VB.d.cts +45 -0
- package/build/cjs/public/ai-sdk.cjs +534 -0
- package/build/cjs/public/ai-sdk.cjs.map +1 -0
- package/build/cjs/public/ai-sdk.d.cts +49 -0
- package/build/esm/ai-sdk/convert-prompt.js +136 -0
- package/build/esm/ai-sdk/convert-prompt.js.map +1 -0
- package/build/esm/ai-sdk/foundry-chat-language-model.js +386 -0
- package/build/esm/ai-sdk/foundry-chat-language-model.js.map +1 -0
- package/build/esm/ai-sdk/map-finish-reason.js +42 -0
- package/build/esm/ai-sdk/map-finish-reason.js.map +1 -0
- package/build/esm/ai-sdk/provider.js +62 -0
- package/build/esm/ai-sdk/provider.js.map +1 -0
- package/build/esm/ai-sdk/types.js +2 -0
- package/build/esm/ai-sdk/types.js.map +1 -0
- package/build/esm/generateText.js +76 -0
- package/build/esm/generateText.js.map +1 -0
- package/build/esm/index.js +23 -0
- package/build/esm/index.js.map +1 -0
- package/build/esm/internal/__test__/mockPlatformClient.js +36 -0
- package/build/esm/internal/__test__/mockPlatformClient.js.map +1 -0
- package/build/esm/internal/ids.js +23 -0
- package/build/esm/internal/ids.js.map +1 -0
- package/build/esm/internal/openAi.js +293 -0
- package/build/esm/internal/openAi.js.map +1 -0
- package/build/esm/internal/options.js +58 -0
- package/build/esm/internal/options.js.map +1 -0
- package/build/esm/internal/runStep.js +107 -0
- package/build/esm/internal/runStep.js.map +1 -0
- package/build/esm/internal/streamStep.js +223 -0
- package/build/esm/internal/streamStep.js.map +1 -0
- package/build/esm/internal/uiMessageBridge.js +55 -0
- package/build/esm/internal/uiMessageBridge.js.map +1 -0
- package/build/esm/lmsChatTransport.js +185 -0
- package/build/esm/lmsChatTransport.js.map +1 -0
- package/build/esm/model.js +74 -0
- package/build/esm/model.js.map +1 -0
- package/build/esm/public/ai-sdk.js +18 -0
- package/build/esm/public/ai-sdk.js.map +1 -0
- package/build/esm/streamText.js +210 -0
- package/build/esm/streamText.js.map +1 -0
- package/build/esm/types.js +2 -0
- package/build/esm/types.js.map +1 -0
- package/build/esm/uiMessage.js +34 -0
- package/build/esm/uiMessage.js.map +1 -0
- package/build/types/ai-sdk/convert-prompt.d.ts +19 -0
- package/build/types/ai-sdk/convert-prompt.d.ts.map +1 -0
- package/build/types/ai-sdk/foundry-chat-language-model.d.ts +20 -0
- package/build/types/ai-sdk/foundry-chat-language-model.d.ts.map +1 -0
- package/build/types/ai-sdk/map-finish-reason.d.ts +2 -0
- package/build/types/ai-sdk/map-finish-reason.d.ts.map +1 -0
- package/build/types/ai-sdk/provider.d.ts +46 -0
- package/build/types/ai-sdk/provider.d.ts.map +1 -0
- package/build/types/ai-sdk/types.d.ts +1 -0
- package/build/types/ai-sdk/types.d.ts.map +1 -0
- package/build/types/generateText.d.ts +56 -0
- package/build/types/generateText.d.ts.map +1 -0
- package/build/types/index.d.ts +12 -0
- package/build/types/index.d.ts.map +1 -0
- package/build/types/internal/__test__/mockPlatformClient.d.ts +13 -0
- package/build/types/internal/__test__/mockPlatformClient.d.ts.map +1 -0
- package/build/types/internal/ids.d.ts +1 -0
- package/build/types/internal/ids.d.ts.map +1 -0
- package/build/types/internal/openAi.d.ts +107 -0
- package/build/types/internal/openAi.d.ts.map +1 -0
- package/build/types/internal/options.d.ts +4 -0
- package/build/types/internal/options.d.ts.map +1 -0
- package/build/types/internal/runStep.d.ts +29 -0
- package/build/types/internal/runStep.d.ts.map +1 -0
- package/build/types/internal/streamStep.d.ts +52 -0
- package/build/types/internal/streamStep.d.ts.map +1 -0
- package/build/types/internal/uiMessageBridge.d.ts +15 -0
- package/build/types/internal/uiMessageBridge.d.ts.map +1 -0
- package/build/types/lmsChatTransport.d.ts +35 -0
- package/build/types/lmsChatTransport.d.ts.map +1 -0
- package/build/types/model.d.ts +42 -0
- package/build/types/model.d.ts.map +1 -0
- package/build/types/public/ai-sdk.d.ts +2 -0
- package/build/types/public/ai-sdk.d.ts.map +1 -0
- package/build/types/streamText.d.ts +82 -0
- package/build/types/streamText.d.ts.map +1 -0
- package/build/types/types.d.ts +189 -0
- package/build/types/types.d.ts.map +1 -0
- package/build/types/uiMessage.d.ts +85 -0
- package/build/types/uiMessage.d.ts.map +1 -0
- package/package.json +87 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openAi.js","names":["getOpenAiBaseUrl","_getFoundryInternal","getModelApiName","identifier","type","apiName","registeredModelRid","mapUsage","raw","inputTokens","prompt_tokens","outputTokens","completion_tokens","totalTokens","total_tokens","reasoningTokens","completion_tokens_details","reasoning_tokens","cachedInputTokens","prompt_tokens_details","cached_tokens","EMPTY_USAGE","undefined","mapFinishReason","reason","convertMessage","m","warnings","role","content","parts","p","push","text","message","toolCalls","id","toolCallId","function","name","toolName","arguments","JSON","stringify","input","length","tool_calls","map","part","stringifyToolResult","tool_call_id","output","value","c","filter","s","join","convertTools","tools","Object","entries","tool","parameters","isJsonSchemaLike","inputSchema","details","properties","description","convertToolChoice","choice","parseToolArguments","args","parse","buildOpenAiRequestBody","streaming","common","model","messages","flat","max_tokens","maxOutputTokens","temperature","top_p","topP","stop","stopSequences","seed","presence_penalty","presencePenalty","frequency_penalty","frequencyPenalty","tool_choice","toolChoice","stream","stream_options","include_usage","buildAssistantContent","tc","filterHeaders","out","k","v","safeReadText","res","postChatCompletions","handle","baseUrl","client","url","URL","toString","fetch","method","headers","Accept","accept","body","signal","abortSignal","ok","errBody","Error","status","statusText"],"sources":["openAi.ts"],"sourcesContent":["/*\n * Copyright 2026 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { getOpenAiBaseUrl } from \"@osdk/language-models\";\nimport {\n _getFoundryInternal,\n type LanguageModel,\n type ModelIdentifier,\n} from \"../model.js\";\nimport type {\n AssistantModelMessage,\n FinishReason,\n LanguageModelUsage,\n ModelMessage,\n ToolCall,\n ToolChoice,\n ToolResultPart,\n ToolSet,\n Warning,\n} from \"../types.js\";\n\nexport interface OpenAiMessage {\n role: \"system\" | \"user\" | \"assistant\" | \"tool\";\n content: string | Array<OpenAiContentPart> | null;\n tool_calls?: Array<OpenAiAssistantToolCall>;\n tool_call_id?: string;\n}\n\ntype OpenAiContentPart = { type: \"text\"; text: string };\n\nexport interface OpenAiTool {\n type: \"function\";\n function: {\n name: string;\n description?: string;\n parameters: unknown;\n };\n}\n\nexport type OpenAiToolChoice =\n | \"auto\"\n | \"none\"\n | \"required\"\n | { type: \"function\"; function: { name: string } };\n\nexport interface OpenAiAssistantToolCall {\n id: string;\n type: \"function\";\n function: { name: string; arguments: string };\n}\n\nexport function getModelApiName(identifier: ModelIdentifier): string {\n return identifier.type === \"lmsModel\"\n ? identifier.apiName\n : identifier.registeredModelRid;\n}\n\nexport interface OpenAiUsageRaw {\n prompt_tokens: number;\n completion_tokens: number;\n total_tokens: number;\n completion_tokens_details?: { reasoning_tokens?: number };\n prompt_tokens_details?: { cached_tokens?: number };\n}\n\nexport function mapUsage(raw: OpenAiUsageRaw): LanguageModelUsage {\n return {\n inputTokens: raw.prompt_tokens,\n outputTokens: raw.completion_tokens,\n totalTokens: raw.total_tokens,\n reasoningTokens: raw.completion_tokens_details?.reasoning_tokens,\n cachedInputTokens: raw.prompt_tokens_details?.cached_tokens,\n };\n}\n\nexport const EMPTY_USAGE: LanguageModelUsage = {\n inputTokens: undefined,\n outputTokens: undefined,\n totalTokens: undefined,\n};\n\nexport function mapFinishReason(reason: string): FinishReason {\n switch (reason) {\n case \"stop\":\n return \"stop\";\n case \"length\":\n return \"length\";\n case \"tool_calls\":\n case \"function_call\":\n return \"tool-calls\";\n case \"content_filter\":\n return \"content-filter\";\n default:\n return \"other\";\n }\n}\n\nexport function convertMessage(\n m: ModelMessage,\n warnings: Array<Warning>,\n): Array<OpenAiMessage> {\n switch (m.role) {\n case \"system\":\n return [{ role: \"system\", content: m.content }];\n\n case \"user\": {\n if (typeof m.content === \"string\") {\n return [{ role: \"user\", content: m.content }];\n }\n const parts: Array<OpenAiContentPart> = [];\n for (const p of m.content) {\n if (p.type === \"text\") {\n parts.push({ type: \"text\", text: p.text });\n } else {\n warnings.push({\n type: \"other\",\n message: `Unsupported user content part \"${p.type}\": ignored in v0`,\n });\n }\n }\n return [{ role: \"user\", content: parts }];\n }\n\n case \"assistant\": {\n if (typeof m.content === \"string\") {\n return [{ role: \"assistant\", content: m.content }];\n }\n let text = \"\";\n const toolCalls: Array<OpenAiAssistantToolCall> = [];\n for (const p of m.content) {\n switch (p.type) {\n case \"text\":\n text += p.text;\n break;\n case \"tool-call\":\n toolCalls.push({\n id: p.toolCallId,\n type: \"function\",\n function: {\n name: p.toolName,\n arguments: JSON.stringify(p.input ?? {}),\n },\n });\n break;\n case \"reasoning\":\n case \"file\":\n warnings.push({\n type: \"other\",\n message:\n `Unsupported assistant content part \"${p.type}\": ignored in v0`,\n });\n break;\n }\n }\n return [{\n role: \"assistant\",\n content: text.length > 0 ? text : null,\n tool_calls: toolCalls.length > 0 ? toolCalls : undefined,\n }];\n }\n\n case \"tool\":\n return m.content.map((part: ToolResultPart) => ({\n role: \"tool\" as const,\n content: stringifyToolResult(part, warnings),\n tool_call_id: part.toolCallId,\n }));\n }\n}\n\nfunction stringifyToolResult(\n part: ToolResultPart,\n warnings: Array<Warning>,\n): string {\n switch (part.output.type) {\n case \"text\":\n case \"error-text\":\n return part.output.value;\n case \"json\":\n case \"error-json\":\n return JSON.stringify(part.output.value);\n case \"content\":\n return part.output.value\n .map((c) => {\n if (c.type === \"text\") {\n return c.text;\n }\n warnings.push({\n type: \"other\",\n message: `Unsupported tool result media: ignored in v0`,\n });\n return \"\";\n })\n .filter((s) => s.length > 0)\n .join(\"\\n\");\n }\n}\n\nexport function convertTools<TOOLS extends ToolSet>(\n tools: TOOLS,\n warnings: Array<Warning>,\n): Array<OpenAiTool> {\n return Object.entries(tools).map(([name, tool]) => {\n const parameters = isJsonSchemaLike(tool.inputSchema)\n ? tool.inputSchema\n : (warnings.push({\n type: \"unsupported-tool\",\n details:\n `Tool \"${name}\" inputSchema is not a JSON Schema; v0 only supports plain JSON Schema objects. Defaulting to {} parameters.`,\n }),\n { type: \"object\", properties: {} });\n return {\n type: \"function\" as const,\n function: {\n name,\n description: tool.description,\n parameters,\n },\n };\n });\n}\n\nexport function convertToolChoice<TOOLS extends ToolSet>(\n choice: ToolChoice<TOOLS> | undefined,\n): OpenAiToolChoice | undefined {\n if (choice == null) {\n return undefined;\n }\n if (choice === \"auto\" || choice === \"none\" || choice === \"required\") {\n return choice;\n }\n return { type: \"function\", function: { name: choice.toolName } };\n}\n\nfunction isJsonSchemaLike(value: unknown): boolean {\n return (\n typeof value === \"object\"\n && value != null\n && (\n \"type\" in value\n || \"properties\" in value\n || \"$ref\" in value\n || \"oneOf\" in value\n || \"anyOf\" in value\n || \"allOf\" in value\n )\n );\n}\n\nexport function parseToolArguments(\n args: string,\n warnings: Array<Warning>,\n): unknown {\n if (args === \"\") {\n return {};\n }\n try {\n return JSON.parse(args);\n } catch {\n warnings.push({\n type: \"other\",\n message:\n `Tool call arguments were not valid JSON; passing through as a raw string`,\n });\n return args;\n }\n}\n\nexport interface BuildRequestBodyArgs<TOOLS extends ToolSet> {\n apiName: string;\n messages: ReadonlyArray<ModelMessage>;\n tools: TOOLS | undefined;\n toolChoice: ToolChoice<TOOLS> | undefined;\n maxOutputTokens: number | undefined;\n temperature: number | undefined;\n topP: number | undefined;\n stopSequences: ReadonlyArray<string> | undefined;\n seed: number | undefined;\n presencePenalty: number | undefined;\n frequencyPenalty: number | undefined;\n warnings: Array<Warning>;\n}\n\ninterface OpenAiChatRequestCommon {\n model: string;\n messages: Array<OpenAiMessage>;\n max_tokens?: number;\n temperature?: number;\n top_p?: number;\n stop?: ReadonlyArray<string>;\n seed?: number;\n presence_penalty?: number;\n frequency_penalty?: number;\n tools?: Array<OpenAiTool>;\n tool_choice?: OpenAiToolChoice;\n}\n\nexport type OpenAiChatRequestNonStreaming = OpenAiChatRequestCommon & {\n stream: false;\n};\n\nexport type OpenAiChatRequestStreaming = OpenAiChatRequestCommon & {\n stream: true;\n stream_options?: { include_usage?: boolean };\n};\n\nexport function buildOpenAiRequestBody<TOOLS extends ToolSet>(\n args: BuildRequestBodyArgs<TOOLS>,\n streaming: false,\n): OpenAiChatRequestNonStreaming;\nexport function buildOpenAiRequestBody<TOOLS extends ToolSet>(\n args: BuildRequestBodyArgs<TOOLS>,\n streaming: true,\n): OpenAiChatRequestStreaming;\nexport function buildOpenAiRequestBody<TOOLS extends ToolSet>(\n args: BuildRequestBodyArgs<TOOLS>,\n streaming: boolean,\n): OpenAiChatRequestNonStreaming | OpenAiChatRequestStreaming {\n const common: OpenAiChatRequestCommon = {\n model: args.apiName,\n messages: args.messages.map((m) => convertMessage(m, args.warnings)).flat(),\n max_tokens: args.maxOutputTokens,\n temperature: args.temperature,\n top_p: args.topP,\n stop: args.stopSequences,\n seed: args.seed,\n presence_penalty: args.presencePenalty,\n frequency_penalty: args.frequencyPenalty,\n tools: args.tools != null\n ? convertTools(args.tools, args.warnings)\n : undefined,\n tool_choice: convertToolChoice(args.toolChoice),\n };\n if (streaming) {\n return { ...common, stream: true, stream_options: { include_usage: true } };\n }\n return { ...common, stream: false };\n}\n\nexport function buildAssistantContent(\n text: string,\n toolCalls: ReadonlyArray<ToolCall>,\n): AssistantModelMessage[\"content\"] {\n if (toolCalls.length === 0) {\n return text;\n }\n const parts: Array<\n { type: \"text\"; text: string } | {\n type: \"tool-call\";\n toolCallId: string;\n toolName: string;\n input: unknown;\n }\n > = [];\n if (text.length > 0) {\n parts.push({ type: \"text\", text });\n }\n for (const tc of toolCalls) {\n parts.push({\n type: \"tool-call\",\n toolCallId: tc.toolCallId,\n toolName: tc.toolName,\n input: tc.input,\n });\n }\n return parts;\n}\n\nexport function filterHeaders(\n input: Record<string, string | undefined>,\n): Record<string, string> {\n const out: Record<string, string> = {};\n for (const [k, v] of Object.entries(input)) {\n if (v != null) {\n out[k] = v;\n }\n }\n return out;\n}\n\nexport async function safeReadText(\n res: Response,\n): Promise<string | undefined> {\n try {\n return await res.text();\n } catch {\n return undefined;\n }\n}\n\nexport interface PostChatCompletionsArgs {\n model: LanguageModel;\n body: OpenAiChatRequestNonStreaming | OpenAiChatRequestStreaming;\n accept: \"application/json\" | \"text/event-stream\";\n headers: Record<string, string | undefined> | undefined;\n abortSignal: AbortSignal | undefined;\n}\n\n/**\n * POST against the LMS OpenAI proxy's chat/completions endpoint and return\n * the raw response. Throws on a non-2xx response.\n */\nexport async function postChatCompletions(\n args: PostChatCompletionsArgs,\n): Promise<Response> {\n const handle = _getFoundryInternal(args.model);\n const baseUrl = getOpenAiBaseUrl(handle.client);\n const url = new URL(\"chat/completions\", `${baseUrl}/`).toString();\n\n const res = await handle.client.fetch(url, {\n method: \"POST\",\n headers: filterHeaders({\n \"Content-Type\": \"application/json\",\n Accept: args.accept,\n ...(args.headers ?? {}),\n }),\n body: JSON.stringify(args.body),\n signal: args.abortSignal,\n });\n\n if (!res.ok) {\n const errBody = await safeReadText(res);\n throw new Error(\n `LMS chat/completions request failed: ${res.status} ${res.statusText}`\n + (errBody ? `: ${errBody}` : \"\"),\n );\n }\n\n return res;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,SAASA,gBAAgB,QAAQ,uBAAuB;AACxD,SACEC,mBAAmB,QAGd,aAAa;AA2CpB,OAAO,SAASC,eAAeA,CAACC,UAA2B,EAAU;EACnE,OAAOA,UAAU,CAACC,IAAI,KAAK,UAAU,GACjCD,UAAU,CAACE,OAAO,GAClBF,UAAU,CAACG,kBAAkB;AACnC;AAUA,OAAO,SAASC,QAAQA,CAACC,GAAmB,EAAsB;EAChE,OAAO;IACLC,WAAW,EAAED,GAAG,CAACE,aAAa;IAC9BC,YAAY,EAAEH,GAAG,CAACI,iBAAiB;IACnCC,WAAW,EAAEL,GAAG,CAACM,YAAY;IAC7BC,eAAe,EAAEP,GAAG,CAACQ,yBAAyB,EAAEC,gBAAgB;IAChEC,iBAAiB,EAAEV,GAAG,CAACW,qBAAqB,EAAEC;EAChD,CAAC;AACH;AAEA,OAAO,MAAMC,WAA+B,GAAG;EAC7CZ,WAAW,EAAEa,SAAS;EACtBX,YAAY,EAAEW,SAAS;EACvBT,WAAW,EAAES;AACf,CAAC;AAED,OAAO,SAASC,eAAeA,CAACC,MAAc,EAAgB;EAC5D,QAAQA,MAAM;IACZ,KAAK,MAAM;MACT,OAAO,MAAM;IACf,KAAK,QAAQ;MACX,OAAO,QAAQ;IACjB,KAAK,YAAY;IACjB,KAAK,eAAe;MAClB,OAAO,YAAY;IACrB,KAAK,gBAAgB;MACnB,OAAO,gBAAgB;IACzB;MACE,OAAO,OAAO;EAClB;AACF;AAEA,OAAO,SAASC,cAAcA,CAC5BC,CAAe,EACfC,QAAwB,EACF;EACtB,QAAQD,CAAC,CAACE,IAAI;IACZ,KAAK,QAAQ;MACX,OAAO,CAAC;QAAEA,IAAI,EAAE,QAAQ;QAAEC,OAAO,EAAEH,CAAC,CAACG;MAAQ,CAAC,CAAC;IAEjD,KAAK,MAAM;MAAE;QACX,IAAI,OAAOH,CAAC,CAACG,OAAO,KAAK,QAAQ,EAAE;UACjC,OAAO,CAAC;YAAED,IAAI,EAAE,MAAM;YAAEC,OAAO,EAAEH,CAAC,CAACG;UAAQ,CAAC,CAAC;QAC/C;QACA,MAAMC,KAA+B,GAAG,EAAE;QAC1C,KAAK,MAAMC,CAAC,IAAIL,CAAC,CAACG,OAAO,EAAE;UACzB,IAAIE,CAAC,CAAC3B,IAAI,KAAK,MAAM,EAAE;YACrB0B,KAAK,CAACE,IAAI,CAAC;cAAE5B,IAAI,EAAE,MAAM;cAAE6B,IAAI,EAAEF,CAAC,CAACE;YAAK,CAAC,CAAC;UAC5C,CAAC,MAAM;YACLN,QAAQ,CAACK,IAAI,CAAC;cACZ5B,IAAI,EAAE,OAAO;cACb8B,OAAO,EAAE,kCAAkCH,CAAC,CAAC3B,IAAI;YACnD,CAAC,CAAC;UACJ;QACF;QACA,OAAO,CAAC;UAAEwB,IAAI,EAAE,MAAM;UAAEC,OAAO,EAAEC;QAAM,CAAC,CAAC;MAC3C;IAEA,KAAK,WAAW;MAAE;QAChB,IAAI,OAAOJ,CAAC,CAACG,OAAO,KAAK,QAAQ,EAAE;UACjC,OAAO,CAAC;YAAED,IAAI,EAAE,WAAW;YAAEC,OAAO,EAAEH,CAAC,CAACG;UAAQ,CAAC,CAAC;QACpD;QACA,IAAII,IAAI,GAAG,EAAE;QACb,MAAME,SAAyC,GAAG,EAAE;QACpD,KAAK,MAAMJ,CAAC,IAAIL,CAAC,CAACG,OAAO,EAAE;UACzB,QAAQE,CAAC,CAAC3B,IAAI;YACZ,KAAK,MAAM;cACT6B,IAAI,IAAIF,CAAC,CAACE,IAAI;cACd;YACF,KAAK,WAAW;cACdE,SAAS,CAACH,IAAI,CAAC;gBACbI,EAAE,EAAEL,CAAC,CAACM,UAAU;gBAChBjC,IAAI,EAAE,UAAU;gBAChBkC,QAAQ,EAAE;kBACRC,IAAI,EAAER,CAAC,CAACS,QAAQ;kBAChBC,SAAS,EAAEC,IAAI,CAACC,SAAS,CAACZ,CAAC,CAACa,KAAK,IAAI,CAAC,CAAC;gBACzC;cACF,CAAC,CAAC;cACF;YACF,KAAK,WAAW;YAChB,KAAK,MAAM;cACTjB,QAAQ,CAACK,IAAI,CAAC;gBACZ5B,IAAI,EAAE,OAAO;gBACb8B,OAAO,EACL,uCAAuCH,CAAC,CAAC3B,IAAI;cACjD,CAAC,CAAC;cACF;UACJ;QACF;QACA,OAAO,CAAC;UACNwB,IAAI,EAAE,WAAW;UACjBC,OAAO,EAAEI,IAAI,CAACY,MAAM,GAAG,CAAC,GAAGZ,IAAI,GAAG,IAAI;UACtCa,UAAU,EAAEX,SAAS,CAACU,MAAM,GAAG,CAAC,GAAGV,SAAS,GAAGb;QACjD,CAAC,CAAC;MACJ;IAEA,KAAK,MAAM;MACT,OAAOI,CAAC,CAACG,OAAO,CAACkB,GAAG,CAAEC,IAAoB,KAAM;QAC9CpB,IAAI,EAAE,MAAe;QACrBC,OAAO,EAAEoB,mBAAmB,CAACD,IAAI,EAAErB,QAAQ,CAAC;QAC5CuB,YAAY,EAAEF,IAAI,CAACX;MACrB,CAAC,CAAC,CAAC;EACP;AACF;AAEA,SAASY,mBAAmBA,CAC1BD,IAAoB,EACpBrB,QAAwB,EAChB;EACR,QAAQqB,IAAI,CAACG,MAAM,CAAC/C,IAAI;IACtB,KAAK,MAAM;IACX,KAAK,YAAY;MACf,OAAO4C,IAAI,CAACG,MAAM,CAACC,KAAK;IAC1B,KAAK,MAAM;IACX,KAAK,YAAY;MACf,OAAOV,IAAI,CAACC,SAAS,CAACK,IAAI,CAACG,MAAM,CAACC,KAAK,CAAC;IAC1C,KAAK,SAAS;MACZ,OAAOJ,IAAI,CAACG,MAAM,CAACC,KAAK,CACrBL,GAAG,CAAEM,CAAC,IAAK;QACV,IAAIA,CAAC,CAACjD,IAAI,KAAK,MAAM,EAAE;UACrB,OAAOiD,CAAC,CAACpB,IAAI;QACf;QACAN,QAAQ,CAACK,IAAI,CAAC;UACZ5B,IAAI,EAAE,OAAO;UACb8B,OAAO,EAAE;QACX,CAAC,CAAC;QACF,OAAO,EAAE;MACX,CAAC,CAAC,CACDoB,MAAM,CAAEC,CAAC,IAAKA,CAAC,CAACV,MAAM,GAAG,CAAC,CAAC,CAC3BW,IAAI,CAAC,IAAI,CAAC;EACjB;AACF;AAEA,OAAO,SAASC,YAAYA,CAC1BC,KAAY,EACZ/B,QAAwB,EACL;EACnB,OAAOgC,MAAM,CAACC,OAAO,CAACF,KAAK,CAAC,CAACX,GAAG,CAAC,CAAC,CAACR,IAAI,EAAEsB,IAAI,CAAC,KAAK;IACjD,MAAMC,UAAU,GAAGC,gBAAgB,CAACF,IAAI,CAACG,WAAW,CAAC,GACjDH,IAAI,CAACG,WAAW,IACfrC,QAAQ,CAACK,IAAI,CAAC;MACf5B,IAAI,EAAE,kBAAkB;MACxB6D,OAAO,EACL,SAAS1B,IAAI;IACjB,CAAC,CAAC,EACA;MAAEnC,IAAI,EAAE,QAAQ;MAAE8D,UAAU,EAAE,CAAC;IAAE,CAAC,CAAC;IACvC,OAAO;MACL9D,IAAI,EAAE,UAAmB;MACzBkC,QAAQ,EAAE;QACRC,IAAI;QACJ4B,WAAW,EAAEN,IAAI,CAACM,WAAW;QAC7BL;MACF;IACF,CAAC;EACH,CAAC,CAAC;AACJ;AAEA,OAAO,SAASM,iBAAiBA,CAC/BC,MAAqC,EACP;EAC9B,IAAIA,MAAM,IAAI,IAAI,EAAE;IAClB,OAAO/C,SAAS;EAClB;EACA,IAAI+C,MAAM,KAAK,MAAM,IAAIA,MAAM,KAAK,MAAM,IAAIA,MAAM,KAAK,UAAU,EAAE;IACnE,OAAOA,MAAM;EACf;EACA,OAAO;IAAEjE,IAAI,EAAE,UAAU;IAAEkC,QAAQ,EAAE;MAAEC,IAAI,EAAE8B,MAAM,CAAC7B;IAAS;EAAE,CAAC;AAClE;AAEA,SAASuB,gBAAgBA,CAACX,KAAc,EAAW;EACjD,OACE,OAAOA,KAAK,KAAK,QAAQ,IACtBA,KAAK,IAAI,IAAI,KAEd,MAAM,IAAIA,KAAK,IACZ,YAAY,IAAIA,KAAK,IACrB,MAAM,IAAIA,KAAK,IACf,OAAO,IAAIA,KAAK,IAChB,OAAO,IAAIA,KAAK,IAChB,OAAO,IAAIA,KAAK,CACpB;AAEL;AAEA,OAAO,SAASkB,kBAAkBA,CAChCC,IAAY,EACZ5C,QAAwB,EACf;EACT,IAAI4C,IAAI,KAAK,EAAE,EAAE;IACf,OAAO,CAAC,CAAC;EACX;EACA,IAAI;IACF,OAAO7B,IAAI,CAAC8B,KAAK,CAACD,IAAI,CAAC;EACzB,CAAC,CAAC,MAAM;IACN5C,QAAQ,CAACK,IAAI,CAAC;MACZ5B,IAAI,EAAE,OAAO;MACb8B,OAAO,EACL;IACJ,CAAC,CAAC;IACF,OAAOqC,IAAI;EACb;AACF;AAgDA,OAAO,SAASE,sBAAsBA,CACpCF,IAAiC,EACjCG,SAAkB,EAC0C;EAC5D,MAAMC,MAA+B,GAAG;IACtCC,KAAK,EAAEL,IAAI,CAAClE,OAAO;IACnBwE,QAAQ,EAAEN,IAAI,CAACM,QAAQ,CAAC9B,GAAG,CAAErB,CAAC,IAAKD,cAAc,CAACC,CAAC,EAAE6C,IAAI,CAAC5C,QAAQ,CAAC,CAAC,CAACmD,IAAI,CAAC,CAAC;IAC3EC,UAAU,EAAER,IAAI,CAACS,eAAe;IAChCC,WAAW,EAAEV,IAAI,CAACU,WAAW;IAC7BC,KAAK,EAAEX,IAAI,CAACY,IAAI;IAChBC,IAAI,EAAEb,IAAI,CAACc,aAAa;IACxBC,IAAI,EAAEf,IAAI,CAACe,IAAI;IACfC,gBAAgB,EAAEhB,IAAI,CAACiB,eAAe;IACtCC,iBAAiB,EAAElB,IAAI,CAACmB,gBAAgB;IACxChC,KAAK,EAAEa,IAAI,CAACb,KAAK,IAAI,IAAI,GACrBD,YAAY,CAACc,IAAI,CAACb,KAAK,EAAEa,IAAI,CAAC5C,QAAQ,CAAC,GACvCL,SAAS;IACbqE,WAAW,EAAEvB,iBAAiB,CAACG,IAAI,CAACqB,UAAU;EAChD,CAAC;EACD,IAAIlB,SAAS,EAAE;IACb,OAAO;MAAE,GAAGC,MAAM;MAAEkB,MAAM,EAAE,IAAI;MAAEC,cAAc,EAAE;QAAEC,aAAa,EAAE;MAAK;IAAE,CAAC;EAC7E;EACA,OAAO;IAAE,GAAGpB,MAAM;IAAEkB,MAAM,EAAE;EAAM,CAAC;AACrC;AAEA,OAAO,SAASG,qBAAqBA,CACnC/D,IAAY,EACZE,SAAkC,EACA;EAClC,IAAIA,SAAS,CAACU,MAAM,KAAK,CAAC,EAAE;IAC1B,OAAOZ,IAAI;EACb;EACA,MAAMH,KAOL,GAAG,EAAE;EACN,IAAIG,IAAI,CAACY,MAAM,GAAG,CAAC,EAAE;IACnBf,KAAK,CAACE,IAAI,CAAC;MAAE5B,IAAI,EAAE,MAAM;MAAE6B;IAAK,CAAC,CAAC;EACpC;EACA,KAAK,MAAMgE,EAAE,IAAI9D,SAAS,EAAE;IAC1BL,KAAK,CAACE,IAAI,CAAC;MACT5B,IAAI,EAAE,WAAW;MACjBiC,UAAU,EAAE4D,EAAE,CAAC5D,UAAU;MACzBG,QAAQ,EAAEyD,EAAE,CAACzD,QAAQ;MACrBI,KAAK,EAAEqD,EAAE,CAACrD;IACZ,CAAC,CAAC;EACJ;EACA,OAAOd,KAAK;AACd;AAEA,OAAO,SAASoE,aAAaA,CAC3BtD,KAAyC,EACjB;EACxB,MAAMuD,GAA2B,GAAG,CAAC,CAAC;EACtC,KAAK,MAAM,CAACC,CAAC,EAAEC,CAAC,CAAC,IAAI1C,MAAM,CAACC,OAAO,CAAChB,KAAK,CAAC,EAAE;IAC1C,IAAIyD,CAAC,IAAI,IAAI,EAAE;MACbF,GAAG,CAACC,CAAC,CAAC,GAAGC,CAAC;IACZ;EACF;EACA,OAAOF,GAAG;AACZ;AAEA,OAAO,eAAeG,YAAYA,CAChCC,GAAa,EACgB;EAC7B,IAAI;IACF,OAAO,MAAMA,GAAG,CAACtE,IAAI,CAAC,CAAC;EACzB,CAAC,CAAC,MAAM;IACN,OAAOX,SAAS;EAClB;AACF;AAUA;AACA;AACA;AACA;AACA,OAAO,eAAekF,mBAAmBA,CACvCjC,IAA6B,EACV;EACnB,MAAMkC,MAAM,GAAGxG,mBAAmB,CAACsE,IAAI,CAACK,KAAK,CAAC;EAC9C,MAAM8B,OAAO,GAAG1G,gBAAgB,CAACyG,MAAM,CAACE,MAAM,CAAC;EAC/C,MAAMC,GAAG,GAAG,IAAIC,GAAG,CAAC,kBAAkB,EAAE,GAAGH,OAAO,GAAG,CAAC,CAACI,QAAQ,CAAC,CAAC;EAEjE,MAAMP,GAAG,GAAG,MAAME,MAAM,CAACE,MAAM,CAACI,KAAK,CAACH,GAAG,EAAE;IACzCI,MAAM,EAAE,MAAM;IACdC,OAAO,EAAEf,aAAa,CAAC;MACrB,cAAc,EAAE,kBAAkB;MAClCgB,MAAM,EAAE3C,IAAI,CAAC4C,MAAM;MACnB,IAAI5C,IAAI,CAAC0C,OAAO,IAAI,CAAC,CAAC;IACxB,CAAC,CAAC;IACFG,IAAI,EAAE1E,IAAI,CAACC,SAAS,CAAC4B,IAAI,CAAC6C,IAAI,CAAC;IAC/BC,MAAM,EAAE9C,IAAI,CAAC+C;EACf,CAAC,CAAC;EAEF,IAAI,CAACf,GAAG,CAACgB,EAAE,EAAE;IACX,MAAMC,OAAO,GAAG,MAAMlB,YAAY,CAACC,GAAG,CAAC;IACvC,MAAM,IAAIkB,KAAK,CACb,wCAAwClB,GAAG,CAACmB,MAAM,IAAInB,GAAG,CAACoB,UAAU,EAAE,IACjEH,OAAO,GAAG,KAAKA,OAAO,EAAE,GAAG,EAAE,CACpC,CAAC;EACH;EAEA,OAAOjB,GAAG;AACZ","ignoreList":[]}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
/*
|
|
2
|
+
* Copyright 2026 Palantir Technologies, Inc. All rights reserved.
|
|
3
|
+
*
|
|
4
|
+
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
+
* you may not use this file except in compliance with the License.
|
|
6
|
+
* You may obtain a copy of the License at
|
|
7
|
+
*
|
|
8
|
+
* http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
+
*
|
|
10
|
+
* Unless required by applicable law or agreed to in writing, software
|
|
11
|
+
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
+
* See the License for the specific language governing permissions and
|
|
14
|
+
* limitations under the License.
|
|
15
|
+
*/
|
|
16
|
+
|
|
17
|
+
export function resolveMessages(callerLabel, system, prompt, messages) {
|
|
18
|
+
if (prompt != null && messages != null) {
|
|
19
|
+
throw new Error(`${callerLabel}: cannot specify both \`prompt\` and \`messages\`: choose one.`);
|
|
20
|
+
}
|
|
21
|
+
const sys = resolveSystem(system);
|
|
22
|
+
const body = resolveBody(prompt, messages);
|
|
23
|
+
if (sys.length === 0 && body.length === 0) {
|
|
24
|
+
throw new Error(`${callerLabel}: must provide at least one of \`system\`, \`prompt\`, or \`messages\`.`);
|
|
25
|
+
}
|
|
26
|
+
return [...sys, ...body];
|
|
27
|
+
}
|
|
28
|
+
function resolveSystem(system) {
|
|
29
|
+
if (system == null) {
|
|
30
|
+
return [];
|
|
31
|
+
}
|
|
32
|
+
if (typeof system === "string") {
|
|
33
|
+
return system === "" ? [] : [{
|
|
34
|
+
role: "system",
|
|
35
|
+
content: system
|
|
36
|
+
}];
|
|
37
|
+
}
|
|
38
|
+
if (Array.isArray(system)) {
|
|
39
|
+
return system;
|
|
40
|
+
}
|
|
41
|
+
return [system];
|
|
42
|
+
}
|
|
43
|
+
function resolveBody(prompt, messages) {
|
|
44
|
+
if (messages != null) {
|
|
45
|
+
return messages;
|
|
46
|
+
}
|
|
47
|
+
if (prompt == null) {
|
|
48
|
+
return [];
|
|
49
|
+
}
|
|
50
|
+
if (typeof prompt === "string") {
|
|
51
|
+
return [{
|
|
52
|
+
role: "user",
|
|
53
|
+
content: prompt
|
|
54
|
+
}];
|
|
55
|
+
}
|
|
56
|
+
return prompt;
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=options.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"options.js","names":["resolveMessages","callerLabel","system","prompt","messages","Error","sys","resolveSystem","body","resolveBody","length","role","content","Array","isArray"],"sources":["options.ts"],"sourcesContent":["/*\n * Copyright 2026 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ModelMessage, SystemModelMessage } from \"../types.js\";\n\nexport type SystemOption =\n | string\n | SystemModelMessage\n | Array<SystemModelMessage>\n | undefined;\n\nexport type PromptOption = string | Array<ModelMessage> | undefined;\n\nexport function resolveMessages(\n callerLabel: string,\n system: SystemOption,\n prompt: PromptOption,\n messages: Array<ModelMessage> | undefined,\n): Array<ModelMessage> {\n if (prompt != null && messages != null) {\n throw new Error(\n `${callerLabel}: cannot specify both \\`prompt\\` and \\`messages\\`: choose one.`,\n );\n }\n\n const sys: Array<SystemModelMessage> = resolveSystem(system);\n const body: Array<ModelMessage> = resolveBody(prompt, messages);\n\n if (sys.length === 0 && body.length === 0) {\n throw new Error(\n `${callerLabel}: must provide at least one of \\`system\\`, \\`prompt\\`, or \\`messages\\`.`,\n );\n }\n\n return [...sys, ...body];\n}\n\nfunction resolveSystem(system: SystemOption): Array<SystemModelMessage> {\n if (system == null) {\n return [];\n }\n if (typeof system === \"string\") {\n return system === \"\" ? [] : [{ role: \"system\", content: system }];\n }\n if (Array.isArray(system)) {\n return system;\n }\n return [system];\n}\n\nfunction resolveBody(\n prompt: PromptOption,\n messages: Array<ModelMessage> | undefined,\n): Array<ModelMessage> {\n if (messages != null) {\n return messages;\n }\n if (prompt == null) {\n return [];\n }\n if (typeof prompt === \"string\") {\n return [{ role: \"user\", content: prompt }];\n }\n return prompt;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAYA,OAAO,SAASA,eAAeA,CAC7BC,WAAmB,EACnBC,MAAoB,EACpBC,MAAoB,EACpBC,QAAyC,EACpB;EACrB,IAAID,MAAM,IAAI,IAAI,IAAIC,QAAQ,IAAI,IAAI,EAAE;IACtC,MAAM,IAAIC,KAAK,CACb,GAAGJ,WAAW,gEAChB,CAAC;EACH;EAEA,MAAMK,GAA8B,GAAGC,aAAa,CAACL,MAAM,CAAC;EAC5D,MAAMM,IAAyB,GAAGC,WAAW,CAACN,MAAM,EAAEC,QAAQ,CAAC;EAE/D,IAAIE,GAAG,CAACI,MAAM,KAAK,CAAC,IAAIF,IAAI,CAACE,MAAM,KAAK,CAAC,EAAE;IACzC,MAAM,IAAIL,KAAK,CACb,GAAGJ,WAAW,yEAChB,CAAC;EACH;EAEA,OAAO,CAAC,GAAGK,GAAG,EAAE,GAAGE,IAAI,CAAC;AAC1B;AAEA,SAASD,aAAaA,CAACL,MAAoB,EAA6B;EACtE,IAAIA,MAAM,IAAI,IAAI,EAAE;IAClB,OAAO,EAAE;EACX;EACA,IAAI,OAAOA,MAAM,KAAK,QAAQ,EAAE;IAC9B,OAAOA,MAAM,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;MAAES,IAAI,EAAE,QAAQ;MAAEC,OAAO,EAAEV;IAAO,CAAC,CAAC;EACnE;EACA,IAAIW,KAAK,CAACC,OAAO,CAACZ,MAAM,CAAC,EAAE;IACzB,OAAOA,MAAM;EACf;EACA,OAAO,CAACA,MAAM,CAAC;AACjB;AAEA,SAASO,WAAWA,CAClBN,MAAoB,EACpBC,QAAyC,EACpB;EACrB,IAAIA,QAAQ,IAAI,IAAI,EAAE;IACpB,OAAOA,QAAQ;EACjB;EACA,IAAID,MAAM,IAAI,IAAI,EAAE;IAClB,OAAO,EAAE;EACX;EACA,IAAI,OAAOA,MAAM,KAAK,QAAQ,EAAE;IAC9B,OAAO,CAAC;MAAEQ,IAAI,EAAE,MAAM;MAAEC,OAAO,EAAET;IAAO,CAAC,CAAC;EAC5C;EACA,OAAOA,MAAM;AACf","ignoreList":[]}
|
|
@@ -0,0 +1,107 @@
|
|
|
1
|
+
/*
|
|
2
|
+
* Copyright 2026 Palantir Technologies, Inc. All rights reserved.
|
|
3
|
+
*
|
|
4
|
+
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
+
* you may not use this file except in compliance with the License.
|
|
6
|
+
* You may obtain a copy of the License at
|
|
7
|
+
*
|
|
8
|
+
* http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
+
*
|
|
10
|
+
* Unless required by applicable law or agreed to in writing, software
|
|
11
|
+
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
+
* See the License for the specific language governing permissions and
|
|
14
|
+
* limitations under the License.
|
|
15
|
+
*/
|
|
16
|
+
|
|
17
|
+
import { _getFoundryInternal } from "../model.js";
|
|
18
|
+
import { buildAssistantContent, buildOpenAiRequestBody, EMPTY_USAGE, getModelApiName, mapFinishReason, mapUsage, parseToolArguments, postChatCompletions } from "./openAi.js";
|
|
19
|
+
/**
|
|
20
|
+
* Performs a single non-streaming chat completion against the LMS OpenAI proxy
|
|
21
|
+
* and returns a {@link StepResult}. Caller is responsible for prepending any
|
|
22
|
+
* system messages before invoking.
|
|
23
|
+
*/
|
|
24
|
+
export async function runStep(input) {
|
|
25
|
+
const apiName = getModelApiName(_getFoundryInternal(input.model).identifier);
|
|
26
|
+
const body = buildOpenAiRequestBody({
|
|
27
|
+
apiName,
|
|
28
|
+
...input
|
|
29
|
+
}, false);
|
|
30
|
+
const res = await postChatCompletions({
|
|
31
|
+
model: input.model,
|
|
32
|
+
body,
|
|
33
|
+
accept: "application/json",
|
|
34
|
+
headers: input.headers,
|
|
35
|
+
abortSignal: input.abortSignal
|
|
36
|
+
});
|
|
37
|
+
const data = await res.json();
|
|
38
|
+
return parseResponse(data, body, input.warnings);
|
|
39
|
+
}
|
|
40
|
+
function parseResponse(res, request, warnings) {
|
|
41
|
+
const choice = res.choices[0];
|
|
42
|
+
if (choice == null) {
|
|
43
|
+
throw new Error("LMS chat/completions response did not include any choices");
|
|
44
|
+
}
|
|
45
|
+
const message = choice.message;
|
|
46
|
+
const text = message.content ?? "";
|
|
47
|
+
const reasoningText = message.reasoning ?? message.reasoning_content;
|
|
48
|
+
const toolCalls = (message.tool_calls ?? []).map(c => ({
|
|
49
|
+
type: "tool-call",
|
|
50
|
+
toolCallId: c.id,
|
|
51
|
+
toolName: c.function.name,
|
|
52
|
+
input: parseToolArguments(c.function.arguments, warnings)
|
|
53
|
+
}));
|
|
54
|
+
const reasoning = reasoningText != null ? [{
|
|
55
|
+
type: "reasoning",
|
|
56
|
+
text: reasoningText
|
|
57
|
+
}] : [];
|
|
58
|
+
const content = [];
|
|
59
|
+
if (text.length > 0) {
|
|
60
|
+
content.push({
|
|
61
|
+
type: "text",
|
|
62
|
+
text
|
|
63
|
+
});
|
|
64
|
+
}
|
|
65
|
+
for (const r of reasoning) {
|
|
66
|
+
content.push({
|
|
67
|
+
type: "reasoning",
|
|
68
|
+
text: r.text
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
for (const tc of toolCalls) {
|
|
72
|
+
content.push({
|
|
73
|
+
...tc,
|
|
74
|
+
type: "tool-call"
|
|
75
|
+
});
|
|
76
|
+
}
|
|
77
|
+
const usage = res.usage != null ? mapUsage(res.usage) : EMPTY_USAGE;
|
|
78
|
+
const finishReason = mapFinishReason(choice.finish_reason);
|
|
79
|
+
const responseMessages = [{
|
|
80
|
+
role: "assistant",
|
|
81
|
+
content: buildAssistantContent(text, toolCalls)
|
|
82
|
+
}];
|
|
83
|
+
return {
|
|
84
|
+
content,
|
|
85
|
+
text,
|
|
86
|
+
reasoning,
|
|
87
|
+
reasoningText,
|
|
88
|
+
toolCalls,
|
|
89
|
+
toolResults: [],
|
|
90
|
+
finishReason,
|
|
91
|
+
rawFinishReason: choice.finish_reason,
|
|
92
|
+
usage,
|
|
93
|
+
warnings: warnings.length > 0 ? warnings : undefined,
|
|
94
|
+
request: {
|
|
95
|
+
body: request
|
|
96
|
+
},
|
|
97
|
+
response: {
|
|
98
|
+
id: res.id,
|
|
99
|
+
modelId: res.model,
|
|
100
|
+
timestamp: new Date(res.created * 1000),
|
|
101
|
+
body: res,
|
|
102
|
+
messages: responseMessages
|
|
103
|
+
},
|
|
104
|
+
providerMetadata: undefined
|
|
105
|
+
};
|
|
106
|
+
}
|
|
107
|
+
//# sourceMappingURL=runStep.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runStep.js","names":["_getFoundryInternal","buildAssistantContent","buildOpenAiRequestBody","EMPTY_USAGE","getModelApiName","mapFinishReason","mapUsage","parseToolArguments","postChatCompletions","runStep","input","apiName","model","identifier","body","res","accept","headers","abortSignal","data","json","parseResponse","warnings","request","choice","choices","Error","message","text","content","reasoningText","reasoning","reasoning_content","toolCalls","tool_calls","map","c","type","toolCallId","id","toolName","function","name","arguments","length","push","r","tc","usage","finishReason","finish_reason","responseMessages","role","toolResults","rawFinishReason","undefined","response","modelId","timestamp","Date","created","messages","providerMetadata"],"sources":["runStep.ts"],"sourcesContent":["/*\n * Copyright 2026 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { LanguageModel } from \"../model.js\";\nimport { _getFoundryInternal } from \"../model.js\";\nimport type {\n AssistantModelMessage,\n ContentPart,\n LanguageModelUsage,\n ModelMessage,\n ReasoningOutput,\n StepResult,\n ToolCall,\n ToolChoice,\n ToolSet,\n Warning,\n} from \"../types.js\";\nimport {\n buildAssistantContent,\n buildOpenAiRequestBody,\n EMPTY_USAGE,\n getModelApiName,\n mapFinishReason,\n mapUsage,\n type OpenAiAssistantToolCall,\n type OpenAiChatRequestNonStreaming,\n parseToolArguments,\n postChatCompletions,\n} from \"./openAi.js\";\n\nexport interface RunStepInput<TOOLS extends ToolSet> {\n model: LanguageModel;\n /** System messages flattened into the head of this list. */\n messages: ReadonlyArray<ModelMessage>;\n tools: TOOLS | undefined;\n toolChoice: ToolChoice<TOOLS> | undefined;\n maxOutputTokens: number | undefined;\n temperature: number | undefined;\n topP: number | undefined;\n stopSequences: ReadonlyArray<string> | undefined;\n seed: number | undefined;\n presencePenalty: number | undefined;\n frequencyPenalty: number | undefined;\n abortSignal: AbortSignal | undefined;\n headers: Record<string, string | undefined> | undefined;\n /**\n * Mutated in place — additional warnings discovered while building the\n * request are appended here.\n */\n warnings: Array<Warning>;\n}\n\n/**\n * Performs a single non-streaming chat completion against the LMS OpenAI proxy\n * and returns a {@link StepResult}. Caller is responsible for prepending any\n * system messages before invoking.\n */\nexport async function runStep<TOOLS extends ToolSet>(\n input: RunStepInput<TOOLS>,\n): Promise<StepResult<TOOLS>> {\n const apiName = getModelApiName(_getFoundryInternal(input.model).identifier);\n\n const body = buildOpenAiRequestBody<TOOLS>({ apiName, ...input }, false);\n\n const res = await postChatCompletions({\n model: input.model,\n body,\n accept: \"application/json\",\n headers: input.headers,\n abortSignal: input.abortSignal,\n });\n\n const data = (await res.json()) as OpenAiChatCompletionResponse;\n\n return parseResponse<TOOLS>(data, body, input.warnings);\n}\n\ninterface OpenAiChatCompletionResponse {\n id: string;\n object: string;\n created: number;\n model: string;\n choices: Array<{\n index: number;\n message: {\n role: \"assistant\";\n content: string | null;\n tool_calls?: Array<OpenAiAssistantToolCall>;\n reasoning?: string;\n reasoning_content?: string;\n };\n finish_reason: string;\n }>;\n usage?: {\n prompt_tokens: number;\n completion_tokens: number;\n total_tokens: number;\n completion_tokens_details?: {\n reasoning_tokens?: number;\n };\n prompt_tokens_details?: {\n cached_tokens?: number;\n };\n };\n}\n\nfunction parseResponse<TOOLS extends ToolSet>(\n res: OpenAiChatCompletionResponse,\n request: OpenAiChatRequestNonStreaming,\n warnings: Array<Warning>,\n): StepResult<TOOLS> {\n const choice = res.choices[0];\n if (choice == null) {\n throw new Error(\n \"LMS chat/completions response did not include any choices\",\n );\n }\n\n const message = choice.message;\n const text = message.content ?? \"\";\n const reasoningText = message.reasoning ?? message.reasoning_content;\n\n const toolCalls: Array<ToolCall<keyof TOOLS & string>> = (message.tool_calls\n ?? [])\n .map((c) => ({\n type: \"tool-call\",\n toolCallId: c.id,\n toolName: c.function.name as keyof TOOLS & string,\n input: parseToolArguments(c.function.arguments, warnings),\n }));\n\n const reasoning: Array<ReasoningOutput> = reasoningText != null\n ? [{ type: \"reasoning\", text: reasoningText }]\n : [];\n\n const content: Array<ContentPart<TOOLS>> = [];\n if (text.length > 0) {\n content.push({ type: \"text\", text });\n }\n for (const r of reasoning) {\n content.push({ type: \"reasoning\", text: r.text });\n }\n for (const tc of toolCalls) {\n content.push({ ...tc, type: \"tool-call\" });\n }\n\n const usage: LanguageModelUsage = res.usage != null\n ? mapUsage(res.usage)\n : EMPTY_USAGE;\n\n const finishReason = mapFinishReason(choice.finish_reason);\n\n const responseMessages: Array<AssistantModelMessage> = [{\n role: \"assistant\",\n content: buildAssistantContent(text, toolCalls),\n }];\n\n return {\n content,\n text,\n reasoning,\n reasoningText,\n toolCalls,\n toolResults: [],\n finishReason,\n rawFinishReason: choice.finish_reason,\n usage,\n warnings: warnings.length > 0 ? warnings : undefined,\n request: { body: request },\n response: {\n id: res.id,\n modelId: res.model,\n timestamp: new Date(res.created * 1000),\n body: res,\n messages: responseMessages,\n },\n providerMetadata: undefined,\n };\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAGA,SAASA,mBAAmB,QAAQ,aAAa;AAajD,SACEC,qBAAqB,EACrBC,sBAAsB,EACtBC,WAAW,EACXC,eAAe,EACfC,eAAe,EACfC,QAAQ,EAGRC,kBAAkB,EAClBC,mBAAmB,QACd,aAAa;AAwBpB;AACA;AACA;AACA;AACA;AACA,OAAO,eAAeC,OAAOA,CAC3BC,KAA0B,EACE;EAC5B,MAAMC,OAAO,GAAGP,eAAe,CAACJ,mBAAmB,CAACU,KAAK,CAACE,KAAK,CAAC,CAACC,UAAU,CAAC;EAE5E,MAAMC,IAAI,GAAGZ,sBAAsB,CAAQ;IAAES,OAAO;IAAE,GAAGD;EAAM,CAAC,EAAE,KAAK,CAAC;EAExE,MAAMK,GAAG,GAAG,MAAMP,mBAAmB,CAAC;IACpCI,KAAK,EAAEF,KAAK,CAACE,KAAK;IAClBE,IAAI;IACJE,MAAM,EAAE,kBAAkB;IAC1BC,OAAO,EAAEP,KAAK,CAACO,OAAO;IACtBC,WAAW,EAAER,KAAK,CAACQ;EACrB,CAAC,CAAC;EAEF,MAAMC,IAAI,GAAI,MAAMJ,GAAG,CAACK,IAAI,CAAC,CAAkC;EAE/D,OAAOC,aAAa,CAAQF,IAAI,EAAEL,IAAI,EAAEJ,KAAK,CAACY,QAAQ,CAAC;AACzD;AA+BA,SAASD,aAAaA,CACpBN,GAAiC,EACjCQ,OAAsC,EACtCD,QAAwB,EACL;EACnB,MAAME,MAAM,GAAGT,GAAG,CAACU,OAAO,CAAC,CAAC,CAAC;EAC7B,IAAID,MAAM,IAAI,IAAI,EAAE;IAClB,MAAM,IAAIE,KAAK,CACb,2DACF,CAAC;EACH;EAEA,MAAMC,OAAO,GAAGH,MAAM,CAACG,OAAO;EAC9B,MAAMC,IAAI,GAAGD,OAAO,CAACE,OAAO,IAAI,EAAE;EAClC,MAAMC,aAAa,GAAGH,OAAO,CAACI,SAAS,IAAIJ,OAAO,CAACK,iBAAiB;EAEpE,MAAMC,SAAgD,GAAG,CAACN,OAAO,CAACO,UAAU,IACvE,EAAE,EACJC,GAAG,CAAEC,CAAC,KAAM;IACXC,IAAI,EAAE,WAAW;IACjBC,UAAU,EAAEF,CAAC,CAACG,EAAE;IAChBC,QAAQ,EAAEJ,CAAC,CAACK,QAAQ,CAACC,IAA4B;IACjDhC,KAAK,EAAEH,kBAAkB,CAAC6B,CAAC,CAACK,QAAQ,CAACE,SAAS,EAAErB,QAAQ;EAC1D,CAAC,CAAC,CAAC;EAEL,MAAMS,SAAiC,GAAGD,aAAa,IAAI,IAAI,GAC3D,CAAC;IAAEO,IAAI,EAAE,WAAW;IAAET,IAAI,EAAEE;EAAc,CAAC,CAAC,GAC5C,EAAE;EAEN,MAAMD,OAAkC,GAAG,EAAE;EAC7C,IAAID,IAAI,CAACgB,MAAM,GAAG,CAAC,EAAE;IACnBf,OAAO,CAACgB,IAAI,CAAC;MAAER,IAAI,EAAE,MAAM;MAAET;IAAK,CAAC,CAAC;EACtC;EACA,KAAK,MAAMkB,CAAC,IAAIf,SAAS,EAAE;IACzBF,OAAO,CAACgB,IAAI,CAAC;MAAER,IAAI,EAAE,WAAW;MAAET,IAAI,EAAEkB,CAAC,CAAClB;IAAK,CAAC,CAAC;EACnD;EACA,KAAK,MAAMmB,EAAE,IAAId,SAAS,EAAE;IAC1BJ,OAAO,CAACgB,IAAI,CAAC;MAAE,GAAGE,EAAE;MAAEV,IAAI,EAAE;IAAY,CAAC,CAAC;EAC5C;EAEA,MAAMW,KAAyB,GAAGjC,GAAG,CAACiC,KAAK,IAAI,IAAI,GAC/C1C,QAAQ,CAACS,GAAG,CAACiC,KAAK,CAAC,GACnB7C,WAAW;EAEf,MAAM8C,YAAY,GAAG5C,eAAe,CAACmB,MAAM,CAAC0B,aAAa,CAAC;EAE1D,MAAMC,gBAA8C,GAAG,CAAC;IACtDC,IAAI,EAAE,WAAW;IACjBvB,OAAO,EAAE5B,qBAAqB,CAAC2B,IAAI,EAAEK,SAAS;EAChD,CAAC,CAAC;EAEF,OAAO;IACLJ,OAAO;IACPD,IAAI;IACJG,SAAS;IACTD,aAAa;IACbG,SAAS;IACToB,WAAW,EAAE,EAAE;IACfJ,YAAY;IACZK,eAAe,EAAE9B,MAAM,CAAC0B,aAAa;IACrCF,KAAK;IACL1B,QAAQ,EAAEA,QAAQ,CAACsB,MAAM,GAAG,CAAC,GAAGtB,QAAQ,GAAGiC,SAAS;IACpDhC,OAAO,EAAE;MAAET,IAAI,EAAES;IAAQ,CAAC;IAC1BiC,QAAQ,EAAE;MACRjB,EAAE,EAAExB,GAAG,CAACwB,EAAE;MACVkB,OAAO,EAAE1C,GAAG,CAACH,KAAK;MAClB8C,SAAS,EAAE,IAAIC,IAAI,CAAC5C,GAAG,CAAC6C,OAAO,GAAG,IAAI,CAAC;MACvC9C,IAAI,EAAEC,GAAG;MACT8C,QAAQ,EAAEV;IACZ,CAAC;IACDW,gBAAgB,EAAEP;EACpB,CAAC;AACH","ignoreList":[]}
|
|
@@ -0,0 +1,223 @@
|
|
|
1
|
+
/*
|
|
2
|
+
* Copyright 2026 Palantir Technologies, Inc. All rights reserved.
|
|
3
|
+
*
|
|
4
|
+
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
+
* you may not use this file except in compliance with the License.
|
|
6
|
+
* You may obtain a copy of the License at
|
|
7
|
+
*
|
|
8
|
+
* http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
+
*
|
|
10
|
+
* Unless required by applicable law or agreed to in writing, software
|
|
11
|
+
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
+
* See the License for the specific language governing permissions and
|
|
14
|
+
* limitations under the License.
|
|
15
|
+
*/
|
|
16
|
+
|
|
17
|
+
import { _getFoundryInternal } from "../model.js";
|
|
18
|
+
import { buildAssistantContent, buildOpenAiRequestBody, EMPTY_USAGE, getModelApiName, mapFinishReason, mapUsage, parseToolArguments, postChatCompletions } from "./openAi.js";
|
|
19
|
+
export async function streamStep(input) {
|
|
20
|
+
const apiName = getModelApiName(_getFoundryInternal(input.model).identifier);
|
|
21
|
+
const body = buildOpenAiRequestBody({
|
|
22
|
+
apiName,
|
|
23
|
+
...input
|
|
24
|
+
}, true);
|
|
25
|
+
const res = await postChatCompletions({
|
|
26
|
+
model: input.model,
|
|
27
|
+
body,
|
|
28
|
+
accept: "text/event-stream",
|
|
29
|
+
headers: input.headers,
|
|
30
|
+
abortSignal: input.abortSignal
|
|
31
|
+
});
|
|
32
|
+
if (res.body == null) {
|
|
33
|
+
throw new Error("LMS chat/completions response had no streaming body");
|
|
34
|
+
}
|
|
35
|
+
return parseSseStream({
|
|
36
|
+
body: res.body,
|
|
37
|
+
onChunk: input.onChunk,
|
|
38
|
+
warnings: input.warnings,
|
|
39
|
+
request: {
|
|
40
|
+
body
|
|
41
|
+
}
|
|
42
|
+
});
|
|
43
|
+
}
|
|
44
|
+
const TEXT_PART_ID = "text-0";
|
|
45
|
+
const REASONING_PART_ID = "reasoning-0";
|
|
46
|
+
async function parseSseStream(args) {
|
|
47
|
+
const reader = args.body.pipeThrough(new TextDecoderStream()).getReader();
|
|
48
|
+
let buffer = "";
|
|
49
|
+
let textBuf = "";
|
|
50
|
+
let reasoningStarted = false;
|
|
51
|
+
let reasoningBuf = "";
|
|
52
|
+
let finishReason = "other";
|
|
53
|
+
let rawFinishReason;
|
|
54
|
+
let usage;
|
|
55
|
+
const toolCallAcc = new Map();
|
|
56
|
+
let responseInitialized = false;
|
|
57
|
+
let responseId;
|
|
58
|
+
let responseModel;
|
|
59
|
+
let responseCreated;
|
|
60
|
+
try {
|
|
61
|
+
while (true) {
|
|
62
|
+
const {
|
|
63
|
+
value,
|
|
64
|
+
done
|
|
65
|
+
} = await reader.read();
|
|
66
|
+
if (done) {
|
|
67
|
+
break;
|
|
68
|
+
}
|
|
69
|
+
buffer += value.replace(/\r\n/g, "\n");
|
|
70
|
+
let frameEnd = buffer.indexOf("\n\n");
|
|
71
|
+
while (frameEnd !== -1) {
|
|
72
|
+
const frame = buffer.slice(0, frameEnd);
|
|
73
|
+
buffer = buffer.slice(frameEnd + 2);
|
|
74
|
+
for (const line of frame.split("\n")) {
|
|
75
|
+
if (!line.startsWith("data:")) {
|
|
76
|
+
continue;
|
|
77
|
+
}
|
|
78
|
+
const payload = line.slice(5).trim();
|
|
79
|
+
if (payload === "") {
|
|
80
|
+
continue;
|
|
81
|
+
}
|
|
82
|
+
if (payload === "[DONE]") {
|
|
83
|
+
continue;
|
|
84
|
+
}
|
|
85
|
+
let parsed;
|
|
86
|
+
try {
|
|
87
|
+
parsed = JSON.parse(payload);
|
|
88
|
+
} catch {
|
|
89
|
+
args.warnings.push({
|
|
90
|
+
type: "other",
|
|
91
|
+
message: `Skipped malformed SSE frame: ${payload.slice(0, 80)}`
|
|
92
|
+
});
|
|
93
|
+
continue;
|
|
94
|
+
}
|
|
95
|
+
if (isErrorFrame(parsed)) {
|
|
96
|
+
throw new Error(extractErrorMessage(parsed));
|
|
97
|
+
}
|
|
98
|
+
const chunk = parsed;
|
|
99
|
+
if (!responseInitialized) {
|
|
100
|
+
responseId = chunk.id;
|
|
101
|
+
responseModel = chunk.model;
|
|
102
|
+
responseCreated = chunk.created;
|
|
103
|
+
responseInitialized = true;
|
|
104
|
+
}
|
|
105
|
+
if (chunk.usage != null) {
|
|
106
|
+
usage = mapUsage(chunk.usage);
|
|
107
|
+
}
|
|
108
|
+
for (const choice of chunk.choices ?? []) {
|
|
109
|
+
const delta = choice.delta;
|
|
110
|
+
const contentDelta = delta.content;
|
|
111
|
+
if (typeof contentDelta === "string" && contentDelta.length > 0) {
|
|
112
|
+
textBuf += contentDelta;
|
|
113
|
+
await args.onChunk({
|
|
114
|
+
type: "text-delta",
|
|
115
|
+
id: TEXT_PART_ID,
|
|
116
|
+
delta: contentDelta
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
const reasoningDelta = delta.reasoning ?? delta.reasoning_content;
|
|
120
|
+
if (typeof reasoningDelta === "string" && reasoningDelta.length > 0) {
|
|
121
|
+
reasoningStarted = true;
|
|
122
|
+
reasoningBuf += reasoningDelta;
|
|
123
|
+
await args.onChunk({
|
|
124
|
+
type: "reasoning-delta",
|
|
125
|
+
id: REASONING_PART_ID,
|
|
126
|
+
delta: reasoningDelta
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
if (delta.tool_calls != null) {
|
|
130
|
+
for (const tc of delta.tool_calls) {
|
|
131
|
+
if (typeof tc.index !== "number") {
|
|
132
|
+
continue;
|
|
133
|
+
}
|
|
134
|
+
const acc = toolCallAcc.get(tc.index) ?? {
|
|
135
|
+
id: "",
|
|
136
|
+
name: "",
|
|
137
|
+
args: ""
|
|
138
|
+
};
|
|
139
|
+
if (tc.id != null) {
|
|
140
|
+
acc.id = tc.id;
|
|
141
|
+
}
|
|
142
|
+
if (tc.function?.name != null) {
|
|
143
|
+
acc.name = tc.function.name;
|
|
144
|
+
}
|
|
145
|
+
if (tc.function?.arguments != null) {
|
|
146
|
+
acc.args += tc.function.arguments;
|
|
147
|
+
}
|
|
148
|
+
toolCallAcc.set(tc.index, acc);
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
if (choice.finish_reason != null) {
|
|
152
|
+
rawFinishReason = choice.finish_reason;
|
|
153
|
+
finishReason = mapFinishReason(choice.finish_reason);
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
frameEnd = buffer.indexOf("\n\n");
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
} finally {
|
|
161
|
+
reader.releaseLock();
|
|
162
|
+
}
|
|
163
|
+
const finalUsage = usage ?? EMPTY_USAGE;
|
|
164
|
+
|
|
165
|
+
// A connection drop or non-conformant provider that closes the stream
|
|
166
|
+
// without finish_reason should still surface tool calls to fullStream.
|
|
167
|
+
const toolCalls = Array.from(toolCallAcc.entries()).map(([index, acc]) => ({
|
|
168
|
+
type: "tool-call",
|
|
169
|
+
toolCallId: acc.id !== "" ? acc.id : `tool_call_${index}`,
|
|
170
|
+
toolName: acc.name,
|
|
171
|
+
input: parseToolArguments(acc.args, args.warnings)
|
|
172
|
+
}));
|
|
173
|
+
for (const tc of toolCalls) {
|
|
174
|
+
await args.onChunk({
|
|
175
|
+
type: "tool-call",
|
|
176
|
+
toolCallId: tc.toolCallId,
|
|
177
|
+
toolName: tc.toolName,
|
|
178
|
+
input: tc.input
|
|
179
|
+
});
|
|
180
|
+
}
|
|
181
|
+
await args.onChunk({
|
|
182
|
+
type: "finish",
|
|
183
|
+
finishReason,
|
|
184
|
+
rawFinishReason,
|
|
185
|
+
usage: finalUsage
|
|
186
|
+
});
|
|
187
|
+
const response = {
|
|
188
|
+
id: responseId,
|
|
189
|
+
modelId: responseModel,
|
|
190
|
+
timestamp: responseCreated != null ? new Date(responseCreated * 1000) : undefined,
|
|
191
|
+
messages: [{
|
|
192
|
+
role: "assistant",
|
|
193
|
+
content: buildAssistantContent(textBuf, toolCalls)
|
|
194
|
+
}]
|
|
195
|
+
};
|
|
196
|
+
return {
|
|
197
|
+
text: textBuf,
|
|
198
|
+
reasoningText: reasoningStarted ? reasoningBuf : undefined,
|
|
199
|
+
toolCalls,
|
|
200
|
+
finishReason,
|
|
201
|
+
rawFinishReason,
|
|
202
|
+
usage: finalUsage,
|
|
203
|
+
request: args.request,
|
|
204
|
+
response
|
|
205
|
+
};
|
|
206
|
+
}
|
|
207
|
+
function isErrorFrame(payload) {
|
|
208
|
+
return typeof payload === "object" && payload != null && "error" in payload && payload.error != null;
|
|
209
|
+
}
|
|
210
|
+
function extractErrorMessage(payload) {
|
|
211
|
+
const err = payload.error;
|
|
212
|
+
if (typeof err === "string") {
|
|
213
|
+
return err;
|
|
214
|
+
}
|
|
215
|
+
if (typeof err === "object" && err != null) {
|
|
216
|
+
const message = err.message;
|
|
217
|
+
if (typeof message === "string") {
|
|
218
|
+
return message;
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
return "LMS chat/completions stream emitted an error frame";
|
|
222
|
+
}
|
|
223
|
+
//# sourceMappingURL=streamStep.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"streamStep.js","names":["_getFoundryInternal","buildAssistantContent","buildOpenAiRequestBody","EMPTY_USAGE","getModelApiName","mapFinishReason","mapUsage","parseToolArguments","postChatCompletions","streamStep","input","apiName","model","identifier","body","res","accept","headers","abortSignal","Error","parseSseStream","onChunk","warnings","request","TEXT_PART_ID","REASONING_PART_ID","args","reader","pipeThrough","TextDecoderStream","getReader","buffer","textBuf","reasoningStarted","reasoningBuf","finishReason","rawFinishReason","usage","toolCallAcc","Map","responseInitialized","responseId","responseModel","responseCreated","value","done","read","replace","frameEnd","indexOf","frame","slice","line","split","startsWith","payload","trim","parsed","JSON","parse","push","type","message","isErrorFrame","extractErrorMessage","chunk","id","created","choice","choices","delta","contentDelta","content","length","reasoningDelta","reasoning","reasoning_content","tool_calls","tc","index","acc","get","name","function","arguments","set","finish_reason","releaseLock","finalUsage","toolCalls","Array","from","entries","map","toolCallId","toolName","response","modelId","timestamp","Date","undefined","messages","role","text","reasoningText","error","err"],"sources":["streamStep.ts"],"sourcesContent":["/*\n * Copyright 2026 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { _getFoundryInternal, type LanguageModel } from \"../model.js\";\nimport type {\n FinishReason,\n LanguageModelUsage,\n ModelMessage,\n RequestMetadata,\n ResponseMetadata,\n ToolCall,\n ToolChoice,\n ToolSet,\n Warning,\n} from \"../types.js\";\nimport {\n buildAssistantContent,\n buildOpenAiRequestBody,\n EMPTY_USAGE,\n getModelApiName,\n mapFinishReason,\n mapUsage,\n parseToolArguments,\n postChatCompletions,\n} from \"./openAi.js\";\n\nexport type StreamChunkEvent =\n | { type: \"text-delta\"; id: string; delta: string }\n | { type: \"reasoning-delta\"; id: string; delta: string }\n | {\n type: \"tool-call\";\n toolCallId: string;\n toolName: string;\n input: unknown;\n }\n | {\n type: \"finish\";\n finishReason: FinishReason;\n rawFinishReason: string | undefined;\n usage: LanguageModelUsage;\n }\n | { type: \"error\"; error: Error };\n\nexport interface StreamStepInput<TOOLS extends ToolSet> {\n model: LanguageModel;\n messages: ReadonlyArray<ModelMessage>;\n tools: TOOLS | undefined;\n toolChoice: ToolChoice<TOOLS> | undefined;\n maxOutputTokens: number | undefined;\n temperature: number | undefined;\n topP: number | undefined;\n stopSequences: ReadonlyArray<string> | undefined;\n seed: number | undefined;\n presencePenalty: number | undefined;\n frequencyPenalty: number | undefined;\n abortSignal: AbortSignal | undefined;\n headers: Record<string, string | undefined> | undefined;\n warnings: Array<Warning>;\n onChunk: (chunk: StreamChunkEvent) => void | Promise<void>;\n}\n\nexport interface StreamStepResult {\n text: string;\n reasoningText: string | undefined;\n toolCalls: Array<ToolCall>;\n finishReason: FinishReason;\n rawFinishReason: string | undefined;\n usage: LanguageModelUsage;\n request: RequestMetadata;\n response: ResponseMetadata;\n}\n\ninterface OpenAiStreamChunk {\n id: string;\n object: \"chat.completion.chunk\";\n created: number;\n model: string;\n choices: Array<{\n index: number;\n delta: {\n role?: \"assistant\";\n content?: string | null;\n reasoning?: string;\n reasoning_content?: string;\n tool_calls?: Array<{\n index: number;\n id?: string;\n type?: \"function\";\n function?: { name?: string; arguments?: string };\n }>;\n };\n finish_reason: string | null;\n }>;\n usage?: {\n prompt_tokens: number;\n completion_tokens: number;\n total_tokens: number;\n completion_tokens_details?: { reasoning_tokens?: number };\n prompt_tokens_details?: { cached_tokens?: number };\n };\n}\n\ninterface ToolCallAccumulator {\n id: string;\n name: string;\n args: string;\n}\n\nexport async function streamStep<TOOLS extends ToolSet>(\n input: StreamStepInput<TOOLS>,\n): Promise<StreamStepResult> {\n const apiName = getModelApiName(_getFoundryInternal(input.model).identifier);\n const body = buildOpenAiRequestBody<TOOLS>({ apiName, ...input }, true);\n\n const res = await postChatCompletions({\n model: input.model,\n body,\n accept: \"text/event-stream\",\n headers: input.headers,\n abortSignal: input.abortSignal,\n });\n\n if (res.body == null) {\n throw new Error(\"LMS chat/completions response had no streaming body\");\n }\n\n return parseSseStream({\n body: res.body,\n onChunk: input.onChunk,\n warnings: input.warnings,\n request: { body },\n });\n}\n\ninterface ParseSseStreamArgs {\n body: ReadableStream<Uint8Array>;\n onChunk: (chunk: StreamChunkEvent) => void | Promise<void>;\n warnings: Array<Warning>;\n request: RequestMetadata;\n}\n\nconst TEXT_PART_ID = \"text-0\";\nconst REASONING_PART_ID = \"reasoning-0\";\n\nasync function parseSseStream(\n args: ParseSseStreamArgs,\n): Promise<StreamStepResult> {\n const reader = args.body\n .pipeThrough(new TextDecoderStream())\n .getReader();\n\n let buffer = \"\";\n let textBuf = \"\";\n let reasoningStarted = false;\n let reasoningBuf = \"\";\n let finishReason: FinishReason = \"other\";\n let rawFinishReason: string | undefined;\n let usage: LanguageModelUsage | undefined;\n const toolCallAcc = new Map<number, ToolCallAccumulator>();\n let responseInitialized = false;\n let responseId: string | undefined;\n let responseModel: string | undefined;\n let responseCreated: number | undefined;\n\n try {\n while (true) {\n const { value, done } = await reader.read();\n if (done) {\n break;\n }\n buffer += value.replace(/\\r\\n/g, \"\\n\");\n\n let frameEnd = buffer.indexOf(\"\\n\\n\");\n while (frameEnd !== -1) {\n const frame = buffer.slice(0, frameEnd);\n buffer = buffer.slice(frameEnd + 2);\n\n for (const line of frame.split(\"\\n\")) {\n if (!line.startsWith(\"data:\")) {\n continue;\n }\n const payload = line.slice(5).trim();\n if (payload === \"\") {\n continue;\n }\n if (payload === \"[DONE]\") {\n continue;\n }\n\n let parsed: unknown;\n try {\n parsed = JSON.parse(payload);\n } catch {\n args.warnings.push({\n type: \"other\",\n message: `Skipped malformed SSE frame: ${payload.slice(0, 80)}`,\n });\n continue;\n }\n\n if (isErrorFrame(parsed)) {\n throw new Error(extractErrorMessage(parsed));\n }\n\n const chunk = parsed as OpenAiStreamChunk;\n\n if (!responseInitialized) {\n responseId = chunk.id;\n responseModel = chunk.model;\n responseCreated = chunk.created;\n responseInitialized = true;\n }\n\n if (chunk.usage != null) {\n usage = mapUsage(chunk.usage);\n }\n\n for (const choice of chunk.choices ?? []) {\n const delta = choice.delta;\n const contentDelta = delta.content;\n if (typeof contentDelta === \"string\" && contentDelta.length > 0) {\n textBuf += contentDelta;\n await args.onChunk({\n type: \"text-delta\",\n id: TEXT_PART_ID,\n delta: contentDelta,\n });\n }\n\n const reasoningDelta = delta.reasoning ?? delta.reasoning_content;\n if (\n typeof reasoningDelta === \"string\" && reasoningDelta.length > 0\n ) {\n reasoningStarted = true;\n reasoningBuf += reasoningDelta;\n await args.onChunk({\n type: \"reasoning-delta\",\n id: REASONING_PART_ID,\n delta: reasoningDelta,\n });\n }\n\n if (delta.tool_calls != null) {\n for (const tc of delta.tool_calls) {\n if (typeof tc.index !== \"number\") {\n continue;\n }\n const acc = toolCallAcc.get(tc.index)\n ?? { id: \"\", name: \"\", args: \"\" };\n if (tc.id != null) {\n acc.id = tc.id;\n }\n if (tc.function?.name != null) {\n acc.name = tc.function.name;\n }\n if (tc.function?.arguments != null) {\n acc.args += tc.function.arguments;\n }\n toolCallAcc.set(tc.index, acc);\n }\n }\n\n if (choice.finish_reason != null) {\n rawFinishReason = choice.finish_reason;\n finishReason = mapFinishReason(choice.finish_reason);\n }\n }\n }\n frameEnd = buffer.indexOf(\"\\n\\n\");\n }\n }\n } finally {\n reader.releaseLock();\n }\n\n const finalUsage: LanguageModelUsage = usage ?? EMPTY_USAGE;\n\n // A connection drop or non-conformant provider that closes the stream\n // without finish_reason should still surface tool calls to fullStream.\n const toolCalls: Array<ToolCall> = Array.from(toolCallAcc.entries()).map(\n ([index, acc]) => ({\n type: \"tool-call\" as const,\n toolCallId: acc.id !== \"\" ? acc.id : `tool_call_${index}`,\n toolName: acc.name,\n input: parseToolArguments(acc.args, args.warnings),\n }),\n );\n for (const tc of toolCalls) {\n await args.onChunk({\n type: \"tool-call\",\n toolCallId: tc.toolCallId,\n toolName: tc.toolName,\n input: tc.input,\n });\n }\n\n await args.onChunk({\n type: \"finish\",\n finishReason,\n rawFinishReason,\n usage: finalUsage,\n });\n\n const response: ResponseMetadata = {\n id: responseId,\n modelId: responseModel,\n timestamp: responseCreated != null\n ? new Date(responseCreated * 1000)\n : undefined,\n messages: [{\n role: \"assistant\",\n content: buildAssistantContent(textBuf, toolCalls),\n }],\n };\n\n return {\n text: textBuf,\n reasoningText: reasoningStarted ? reasoningBuf : undefined,\n toolCalls,\n finishReason,\n rawFinishReason,\n usage: finalUsage,\n request: args.request,\n response,\n };\n}\n\nfunction isErrorFrame(payload: unknown): payload is { error: unknown } {\n return (\n typeof payload === \"object\"\n && payload != null\n && \"error\" in payload\n && (payload as { error: unknown }).error != null\n );\n}\n\nfunction extractErrorMessage(payload: { error: unknown }): string {\n const err = payload.error;\n if (typeof err === \"string\") {\n return err;\n }\n if (typeof err === \"object\" && err != null) {\n const message = (err as { message?: unknown }).message;\n if (typeof message === \"string\") {\n return message;\n }\n }\n return \"LMS chat/completions stream emitted an error frame\";\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,SAASA,mBAAmB,QAA4B,aAAa;AAYrE,SACEC,qBAAqB,EACrBC,sBAAsB,EACtBC,WAAW,EACXC,eAAe,EACfC,eAAe,EACfC,QAAQ,EACRC,kBAAkB,EAClBC,mBAAmB,QACd,aAAa;AAoFpB,OAAO,eAAeC,UAAUA,CAC9BC,KAA6B,EACF;EAC3B,MAAMC,OAAO,GAAGP,eAAe,CAACJ,mBAAmB,CAACU,KAAK,CAACE,KAAK,CAAC,CAACC,UAAU,CAAC;EAC5E,MAAMC,IAAI,GAAGZ,sBAAsB,CAAQ;IAAES,OAAO;IAAE,GAAGD;EAAM,CAAC,EAAE,IAAI,CAAC;EAEvE,MAAMK,GAAG,GAAG,MAAMP,mBAAmB,CAAC;IACpCI,KAAK,EAAEF,KAAK,CAACE,KAAK;IAClBE,IAAI;IACJE,MAAM,EAAE,mBAAmB;IAC3BC,OAAO,EAAEP,KAAK,CAACO,OAAO;IACtBC,WAAW,EAAER,KAAK,CAACQ;EACrB,CAAC,CAAC;EAEF,IAAIH,GAAG,CAACD,IAAI,IAAI,IAAI,EAAE;IACpB,MAAM,IAAIK,KAAK,CAAC,qDAAqD,CAAC;EACxE;EAEA,OAAOC,cAAc,CAAC;IACpBN,IAAI,EAAEC,GAAG,CAACD,IAAI;IACdO,OAAO,EAAEX,KAAK,CAACW,OAAO;IACtBC,QAAQ,EAAEZ,KAAK,CAACY,QAAQ;IACxBC,OAAO,EAAE;MAAET;IAAK;EAClB,CAAC,CAAC;AACJ;AASA,MAAMU,YAAY,GAAG,QAAQ;AAC7B,MAAMC,iBAAiB,GAAG,aAAa;AAEvC,eAAeL,cAAcA,CAC3BM,IAAwB,EACG;EAC3B,MAAMC,MAAM,GAAGD,IAAI,CAACZ,IAAI,CACrBc,WAAW,CAAC,IAAIC,iBAAiB,CAAC,CAAC,CAAC,CACpCC,SAAS,CAAC,CAAC;EAEd,IAAIC,MAAM,GAAG,EAAE;EACf,IAAIC,OAAO,GAAG,EAAE;EAChB,IAAIC,gBAAgB,GAAG,KAAK;EAC5B,IAAIC,YAAY,GAAG,EAAE;EACrB,IAAIC,YAA0B,GAAG,OAAO;EACxC,IAAIC,eAAmC;EACvC,IAAIC,KAAqC;EACzC,MAAMC,WAAW,GAAG,IAAIC,GAAG,CAA8B,CAAC;EAC1D,IAAIC,mBAAmB,GAAG,KAAK;EAC/B,IAAIC,UAA8B;EAClC,IAAIC,aAAiC;EACrC,IAAIC,eAAmC;EAEvC,IAAI;IACF,OAAO,IAAI,EAAE;MACX,MAAM;QAAEC,KAAK;QAAEC;MAAK,CAAC,GAAG,MAAMlB,MAAM,CAACmB,IAAI,CAAC,CAAC;MAC3C,IAAID,IAAI,EAAE;QACR;MACF;MACAd,MAAM,IAAIa,KAAK,CAACG,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC;MAEtC,IAAIC,QAAQ,GAAGjB,MAAM,CAACkB,OAAO,CAAC,MAAM,CAAC;MACrC,OAAOD,QAAQ,KAAK,CAAC,CAAC,EAAE;QACtB,MAAME,KAAK,GAAGnB,MAAM,CAACoB,KAAK,CAAC,CAAC,EAAEH,QAAQ,CAAC;QACvCjB,MAAM,GAAGA,MAAM,CAACoB,KAAK,CAACH,QAAQ,GAAG,CAAC,CAAC;QAEnC,KAAK,MAAMI,IAAI,IAAIF,KAAK,CAACG,KAAK,CAAC,IAAI,CAAC,EAAE;UACpC,IAAI,CAACD,IAAI,CAACE,UAAU,CAAC,OAAO,CAAC,EAAE;YAC7B;UACF;UACA,MAAMC,OAAO,GAAGH,IAAI,CAACD,KAAK,CAAC,CAAC,CAAC,CAACK,IAAI,CAAC,CAAC;UACpC,IAAID,OAAO,KAAK,EAAE,EAAE;YAClB;UACF;UACA,IAAIA,OAAO,KAAK,QAAQ,EAAE;YACxB;UACF;UAEA,IAAIE,MAAe;UACnB,IAAI;YACFA,MAAM,GAAGC,IAAI,CAACC,KAAK,CAACJ,OAAO,CAAC;UAC9B,CAAC,CAAC,MAAM;YACN7B,IAAI,CAACJ,QAAQ,CAACsC,IAAI,CAAC;cACjBC,IAAI,EAAE,OAAO;cACbC,OAAO,EAAE,gCAAgCP,OAAO,CAACJ,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;YAC/D,CAAC,CAAC;YACF;UACF;UAEA,IAAIY,YAAY,CAACN,MAAM,CAAC,EAAE;YACxB,MAAM,IAAItC,KAAK,CAAC6C,mBAAmB,CAACP,MAAM,CAAC,CAAC;UAC9C;UAEA,MAAMQ,KAAK,GAAGR,MAA2B;UAEzC,IAAI,CAACjB,mBAAmB,EAAE;YACxBC,UAAU,GAAGwB,KAAK,CAACC,EAAE;YACrBxB,aAAa,GAAGuB,KAAK,CAACrD,KAAK;YAC3B+B,eAAe,GAAGsB,KAAK,CAACE,OAAO;YAC/B3B,mBAAmB,GAAG,IAAI;UAC5B;UAEA,IAAIyB,KAAK,CAAC5B,KAAK,IAAI,IAAI,EAAE;YACvBA,KAAK,GAAG/B,QAAQ,CAAC2D,KAAK,CAAC5B,KAAK,CAAC;UAC/B;UAEA,KAAK,MAAM+B,MAAM,IAAIH,KAAK,CAACI,OAAO,IAAI,EAAE,EAAE;YACxC,MAAMC,KAAK,GAAGF,MAAM,CAACE,KAAK;YAC1B,MAAMC,YAAY,GAAGD,KAAK,CAACE,OAAO;YAClC,IAAI,OAAOD,YAAY,KAAK,QAAQ,IAAIA,YAAY,CAACE,MAAM,GAAG,CAAC,EAAE;cAC/DzC,OAAO,IAAIuC,YAAY;cACvB,MAAM7C,IAAI,CAACL,OAAO,CAAC;gBACjBwC,IAAI,EAAE,YAAY;gBAClBK,EAAE,EAAE1C,YAAY;gBAChB8C,KAAK,EAAEC;cACT,CAAC,CAAC;YACJ;YAEA,MAAMG,cAAc,GAAGJ,KAAK,CAACK,SAAS,IAAIL,KAAK,CAACM,iBAAiB;YACjE,IACE,OAAOF,cAAc,KAAK,QAAQ,IAAIA,cAAc,CAACD,MAAM,GAAG,CAAC,EAC/D;cACAxC,gBAAgB,GAAG,IAAI;cACvBC,YAAY,IAAIwC,cAAc;cAC9B,MAAMhD,IAAI,CAACL,OAAO,CAAC;gBACjBwC,IAAI,EAAE,iBAAiB;gBACvBK,EAAE,EAAEzC,iBAAiB;gBACrB6C,KAAK,EAAEI;cACT,CAAC,CAAC;YACJ;YAEA,IAAIJ,KAAK,CAACO,UAAU,IAAI,IAAI,EAAE;cAC5B,KAAK,MAAMC,EAAE,IAAIR,KAAK,CAACO,UAAU,EAAE;gBACjC,IAAI,OAAOC,EAAE,CAACC,KAAK,KAAK,QAAQ,EAAE;kBAChC;gBACF;gBACA,MAAMC,GAAG,GAAG1C,WAAW,CAAC2C,GAAG,CAACH,EAAE,CAACC,KAAK,CAAC,IAChC;kBAAEb,EAAE,EAAE,EAAE;kBAAEgB,IAAI,EAAE,EAAE;kBAAExD,IAAI,EAAE;gBAAG,CAAC;gBACnC,IAAIoD,EAAE,CAACZ,EAAE,IAAI,IAAI,EAAE;kBACjBc,GAAG,CAACd,EAAE,GAAGY,EAAE,CAACZ,EAAE;gBAChB;gBACA,IAAIY,EAAE,CAACK,QAAQ,EAAED,IAAI,IAAI,IAAI,EAAE;kBAC7BF,GAAG,CAACE,IAAI,GAAGJ,EAAE,CAACK,QAAQ,CAACD,IAAI;gBAC7B;gBACA,IAAIJ,EAAE,CAACK,QAAQ,EAAEC,SAAS,IAAI,IAAI,EAAE;kBAClCJ,GAAG,CAACtD,IAAI,IAAIoD,EAAE,CAACK,QAAQ,CAACC,SAAS;gBACnC;gBACA9C,WAAW,CAAC+C,GAAG,CAACP,EAAE,CAACC,KAAK,EAAEC,GAAG,CAAC;cAChC;YACF;YAEA,IAAIZ,MAAM,CAACkB,aAAa,IAAI,IAAI,EAAE;cAChClD,eAAe,GAAGgC,MAAM,CAACkB,aAAa;cACtCnD,YAAY,GAAG9B,eAAe,CAAC+D,MAAM,CAACkB,aAAa,CAAC;YACtD;UACF;QACF;QACAtC,QAAQ,GAAGjB,MAAM,CAACkB,OAAO,CAAC,MAAM,CAAC;MACnC;IACF;EACF,CAAC,SAAS;IACRtB,MAAM,CAAC4D,WAAW,CAAC,CAAC;EACtB;EAEA,MAAMC,UAA8B,GAAGnD,KAAK,IAAIlC,WAAW;;EAE3D;EACA;EACA,MAAMsF,SAA0B,GAAGC,KAAK,CAACC,IAAI,CAACrD,WAAW,CAACsD,OAAO,CAAC,CAAC,CAAC,CAACC,GAAG,CACtE,CAAC,CAACd,KAAK,EAAEC,GAAG,CAAC,MAAM;IACjBnB,IAAI,EAAE,WAAoB;IAC1BiC,UAAU,EAAEd,GAAG,CAACd,EAAE,KAAK,EAAE,GAAGc,GAAG,CAACd,EAAE,GAAG,aAAaa,KAAK,EAAE;IACzDgB,QAAQ,EAAEf,GAAG,CAACE,IAAI;IAClBxE,KAAK,EAAEH,kBAAkB,CAACyE,GAAG,CAACtD,IAAI,EAAEA,IAAI,CAACJ,QAAQ;EACnD,CAAC,CACH,CAAC;EACD,KAAK,MAAMwD,EAAE,IAAIW,SAAS,EAAE;IAC1B,MAAM/D,IAAI,CAACL,OAAO,CAAC;MACjBwC,IAAI,EAAE,WAAW;MACjBiC,UAAU,EAAEhB,EAAE,CAACgB,UAAU;MACzBC,QAAQ,EAAEjB,EAAE,CAACiB,QAAQ;MACrBrF,KAAK,EAAEoE,EAAE,CAACpE;IACZ,CAAC,CAAC;EACJ;EAEA,MAAMgB,IAAI,CAACL,OAAO,CAAC;IACjBwC,IAAI,EAAE,QAAQ;IACd1B,YAAY;IACZC,eAAe;IACfC,KAAK,EAAEmD;EACT,CAAC,CAAC;EAEF,MAAMQ,QAA0B,GAAG;IACjC9B,EAAE,EAAEzB,UAAU;IACdwD,OAAO,EAAEvD,aAAa;IACtBwD,SAAS,EAAEvD,eAAe,IAAI,IAAI,GAC9B,IAAIwD,IAAI,CAACxD,eAAe,GAAG,IAAI,CAAC,GAChCyD,SAAS;IACbC,QAAQ,EAAE,CAAC;MACTC,IAAI,EAAE,WAAW;MACjB9B,OAAO,EAAEvE,qBAAqB,CAAC+B,OAAO,EAAEyD,SAAS;IACnD,CAAC;EACH,CAAC;EAED,OAAO;IACLc,IAAI,EAAEvE,OAAO;IACbwE,aAAa,EAAEvE,gBAAgB,GAAGC,YAAY,GAAGkE,SAAS;IAC1DX,SAAS;IACTtD,YAAY;IACZC,eAAe;IACfC,KAAK,EAAEmD,UAAU;IACjBjE,OAAO,EAAEG,IAAI,CAACH,OAAO;IACrByE;EACF,CAAC;AACH;AAEA,SAASjC,YAAYA,CAACR,OAAgB,EAAiC;EACrE,OACE,OAAOA,OAAO,KAAK,QAAQ,IACxBA,OAAO,IAAI,IAAI,IACf,OAAO,IAAIA,OAAO,IACjBA,OAAO,CAAwBkD,KAAK,IAAI,IAAI;AAEpD;AAEA,SAASzC,mBAAmBA,CAACT,OAA2B,EAAU;EAChE,MAAMmD,GAAG,GAAGnD,OAAO,CAACkD,KAAK;EACzB,IAAI,OAAOC,GAAG,KAAK,QAAQ,EAAE;IAC3B,OAAOA,GAAG;EACZ;EACA,IAAI,OAAOA,GAAG,KAAK,QAAQ,IAAIA,GAAG,IAAI,IAAI,EAAE;IAC1C,MAAM5C,OAAO,GAAI4C,GAAG,CAA2B5C,OAAO;IACtD,IAAI,OAAOA,OAAO,KAAK,QAAQ,EAAE;MAC/B,OAAOA,OAAO;IAChB;EACF;EACA,OAAO,oDAAoD;AAC7D","ignoreList":[]}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
/*
|
|
2
|
+
* Copyright 2026 Palantir Technologies, Inc. All rights reserved.
|
|
3
|
+
*
|
|
4
|
+
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
+
* you may not use this file except in compliance with the License.
|
|
6
|
+
* You may obtain a copy of the License at
|
|
7
|
+
*
|
|
8
|
+
* http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
+
*
|
|
10
|
+
* Unless required by applicable law or agreed to in writing, software
|
|
11
|
+
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
+
* See the License for the specific language governing permissions and
|
|
14
|
+
* limitations under the License.
|
|
15
|
+
*/
|
|
16
|
+
|
|
17
|
+
import { getUIMessageText } from "../uiMessage.js";
|
|
18
|
+
|
|
19
|
+
/**
|
|
20
|
+
* Convert `UIMessage[]` (the conversation shape rendered in the UI) into
|
|
21
|
+
* `ModelMessage[]` (the shape `streamText` accepts).
|
|
22
|
+
*
|
|
23
|
+
* v0: text-only. File parts, reasoning parts, source parts, and tool-related
|
|
24
|
+
* parts are dropped. Tool-result UIMessages (none in v0 either) would map to
|
|
25
|
+
* `ToolModelMessage` once tool round-trips are wired up.
|
|
26
|
+
*
|
|
27
|
+
* When `system` is provided, any system-role messages already in `ui` are
|
|
28
|
+
* dropped so the prompt isn't double-prepended — the transport-level
|
|
29
|
+
* `system` wins.
|
|
30
|
+
*/
|
|
31
|
+
export function uiMessagesToModelMessages(ui, system) {
|
|
32
|
+
const out = [];
|
|
33
|
+
const transportSystemSet = system != null && system.length > 0;
|
|
34
|
+
if (transportSystemSet) {
|
|
35
|
+
out.push({
|
|
36
|
+
role: "system",
|
|
37
|
+
content: system
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
for (const m of ui) {
|
|
41
|
+
const text = getUIMessageText(m);
|
|
42
|
+
if (text.length === 0) {
|
|
43
|
+
continue;
|
|
44
|
+
}
|
|
45
|
+
if (m.role === "system" && transportSystemSet) {
|
|
46
|
+
continue;
|
|
47
|
+
}
|
|
48
|
+
out.push({
|
|
49
|
+
role: m.role,
|
|
50
|
+
content: text
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
return out;
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=uiMessageBridge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"uiMessageBridge.js","names":["getUIMessageText","uiMessagesToModelMessages","ui","system","out","transportSystemSet","length","push","role","content","m","text"],"sources":["uiMessageBridge.ts"],"sourcesContent":["/*\n * Copyright 2026 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ModelMessage } from \"../types.js\";\nimport { getUIMessageText, type UIMessage } from \"../uiMessage.js\";\n\n/**\n * Convert `UIMessage[]` (the conversation shape rendered in the UI) into\n * `ModelMessage[]` (the shape `streamText` accepts).\n *\n * v0: text-only. File parts, reasoning parts, source parts, and tool-related\n * parts are dropped. Tool-result UIMessages (none in v0 either) would map to\n * `ToolModelMessage` once tool round-trips are wired up.\n *\n * When `system` is provided, any system-role messages already in `ui` are\n * dropped so the prompt isn't double-prepended — the transport-level\n * `system` wins.\n */\nexport function uiMessagesToModelMessages(\n ui: ReadonlyArray<UIMessage>,\n system?: string,\n): Array<ModelMessage> {\n const out: Array<ModelMessage> = [];\n const transportSystemSet = system != null && system.length > 0;\n\n if (transportSystemSet) {\n out.push({ role: \"system\", content: system });\n }\n\n for (const m of ui) {\n const text = getUIMessageText(m);\n if (text.length === 0) {\n continue;\n }\n if (m.role === \"system\" && transportSystemSet) {\n continue;\n }\n out.push({ role: m.role, content: text });\n }\n\n return out;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAGA,SAASA,gBAAgB,QAAwB,iBAAiB;;AAElE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,yBAAyBA,CACvCC,EAA4B,EAC5BC,MAAe,EACM;EACrB,MAAMC,GAAwB,GAAG,EAAE;EACnC,MAAMC,kBAAkB,GAAGF,MAAM,IAAI,IAAI,IAAIA,MAAM,CAACG,MAAM,GAAG,CAAC;EAE9D,IAAID,kBAAkB,EAAE;IACtBD,GAAG,CAACG,IAAI,CAAC;MAAEC,IAAI,EAAE,QAAQ;MAAEC,OAAO,EAAEN;IAAO,CAAC,CAAC;EAC/C;EAEA,KAAK,MAAMO,CAAC,IAAIR,EAAE,EAAE;IAClB,MAAMS,IAAI,GAAGX,gBAAgB,CAACU,CAAC,CAAC;IAChC,IAAIC,IAAI,CAACL,MAAM,KAAK,CAAC,EAAE;MACrB;IACF;IACA,IAAII,CAAC,CAACF,IAAI,KAAK,QAAQ,IAAIH,kBAAkB,EAAE;MAC7C;IACF;IACAD,GAAG,CAACG,IAAI,CAAC;MAAEC,IAAI,EAAEE,CAAC,CAACF,IAAI;MAAEC,OAAO,EAAEE;IAAK,CAAC,CAAC;EAC3C;EAEA,OAAOP,GAAG;AACZ","ignoreList":[]}
|