modelfusion 0.104.0 → 0.105.0

package/model-provider/openai/{chat/OpenAIChatModel.d.ts → OpenAIChatModel.d.ts}

@@ -1,13 +1,13 @@
-import { StructureFromTextPromptTemplate } from "../../../model-function/generate-structure/StructureFromTextPromptTemplate.js";
-import { StructureFromTextStreamingModel } from "../../../model-function/generate-structure/StructureFromTextStreamingModel.js";
-import { PromptTemplateTextStreamingModel } from "../../../model-function/generate-text/PromptTemplateTextStreamingModel.js";
-import { TextGenerationModelSettings, TextStreamingModel } from "../../../model-function/generate-text/TextGenerationModel.js";
-import { TextGenerationPromptTemplate } from "../../../model-function/generate-text/TextGenerationPromptTemplate.js";
-import { ToolCallGenerationModel } from "../../../tool/generate-tool-call/ToolCallGenerationModel.js";
-import { ToolCallsOrTextGenerationModel } from "../../../tool/generate-tool-calls-or-text/ToolCallsOrTextGenerationModel.js";
-import { TikTokenTokenizer } from "../TikTokenTokenizer.js";
+import { StructureFromTextPromptTemplate } from "../../model-function/generate-structure/StructureFromTextPromptTemplate.js";
+import { StructureFromTextStreamingModel } from "../../model-function/generate-structure/StructureFromTextStreamingModel.js";
+import { PromptTemplateFullTextModel } from "../../model-function/generate-text/PromptTemplateFullTextModel.js";
+import { TextGenerationModelSettings, TextStreamingModel } from "../../model-function/generate-text/TextGenerationModel.js";
+import { TextGenerationPromptTemplate } from "../../model-function/generate-text/TextGenerationPromptTemplate.js";
+import { ToolCallGenerationModel } from "../../tool/generate-tool-call/ToolCallGenerationModel.js";
+import { ToolCallsOrTextGenerationModel } from "../../tool/generate-tool-calls-or-text/ToolCallsOrTextGenerationModel.js";
 import { AbstractOpenAIChatCallSettings, AbstractOpenAIChatModel, OpenAIChatPrompt, OpenAIChatResponse } from "./AbstractOpenAIChatModel.js";
 import { OpenAIChatFunctionCallStructureGenerationModel } from "./OpenAIChatFunctionCallStructureGenerationModel.js";
+import { TikTokenTokenizer } from "./TikTokenTokenizer.js";
 export declare const OPENAI_CHAT_MODELS: {
     "gpt-4": {
         contextWindowSize: number;
@@ -147,16 +147,16 @@ export declare class OpenAIChatModel extends AbstractOpenAIChatModel<OpenAIChatS
     /**
      * Returns this model with a text prompt template.
      */
-    withTextPrompt(): PromptTemplateTextStreamingModel<string, OpenAIChatPrompt, OpenAIChatSettings, this>;
+    withTextPrompt(): PromptTemplateFullTextModel<string, OpenAIChatPrompt, OpenAIChatSettings, this>;
     /**
      * Returns this model with an instruction prompt template.
      */
-    withInstructionPrompt(): PromptTemplateTextStreamingModel<import("../../../index.js").InstructionPrompt, OpenAIChatPrompt, OpenAIChatSettings, this>;
+    withInstructionPrompt(): PromptTemplateFullTextModel<import("../../index.js").InstructionPrompt, OpenAIChatPrompt, OpenAIChatSettings, this>;
     /**
      * Returns this model with a chat prompt template.
      */
-    withChatPrompt(): PromptTemplateTextStreamingModel<import("../../../index.js").ChatPrompt, OpenAIChatPrompt, OpenAIChatSettings, this>;
-    withPromptTemplate<INPUT_PROMPT>(promptTemplate: TextGenerationPromptTemplate<INPUT_PROMPT, OpenAIChatPrompt>): PromptTemplateTextStreamingModel<INPUT_PROMPT, OpenAIChatPrompt, OpenAIChatSettings, this>;
+    withChatPrompt(): PromptTemplateFullTextModel<import("../../index.js").ChatPrompt, OpenAIChatPrompt, OpenAIChatSettings, this>;
+    withPromptTemplate<INPUT_PROMPT>(promptTemplate: TextGenerationPromptTemplate<INPUT_PROMPT, OpenAIChatPrompt>): PromptTemplateFullTextModel<INPUT_PROMPT, OpenAIChatPrompt, OpenAIChatSettings, this>;
     withSettings(additionalSettings: Partial<OpenAIChatSettings>): this;
 }
 export {};

package/model-provider/openai/{chat/OpenAIChatModel.js → OpenAIChatModel.js}

@@ -1,10 +1,10 @@
-import { StructureFromTextStreamingModel } from "../../../model-function/generate-structure/StructureFromTextStreamingModel.js";
-import { PromptTemplateTextStreamingModel } from "../../../model-function/generate-text/PromptTemplateTextStreamingModel.js";
-import { textGenerationModelProperties, } from "../../../model-function/generate-text/TextGenerationModel.js";
-import { TikTokenTokenizer } from "../TikTokenTokenizer.js";
+import { StructureFromTextStreamingModel } from "../../model-function/generate-structure/StructureFromTextStreamingModel.js";
+import { PromptTemplateFullTextModel } from "../../model-function/generate-text/PromptTemplateFullTextModel.js";
+import { textGenerationModelProperties, } from "../../model-function/generate-text/TextGenerationModel.js";
 import { AbstractOpenAIChatModel, } from "./AbstractOpenAIChatModel.js";
 import { OpenAIChatFunctionCallStructureGenerationModel } from "./OpenAIChatFunctionCallStructureGenerationModel.js";
 import { chat, identity, instruction, text, } from "./OpenAIChatPromptTemplate.js";
+import { TikTokenTokenizer } from "./TikTokenTokenizer.js";
 import { countOpenAIChatPromptTokens } from "./countOpenAIChatMessageTokens.js";
 /*
  * Available OpenAI chat models, their token limits, and pricing.
@@ -237,7 +237,7 @@ export class OpenAIChatModel extends AbstractOpenAIChatModel {
         return this.withPromptTemplate(chat());
     }
     withPromptTemplate(promptTemplate) {
-        return new PromptTemplateTextStreamingModel({
+        return new PromptTemplateFullTextModel({
             model: this.withSettings({
                 stopSequences: [
                     ...(this.settings.stopSequences ?? []),

package/model-provider/openai/OpenAIChatModel.test.cjs

@@ -0,0 +1,94 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const zod_1 = require("zod");
+const ZodSchema_js_1 = require("../../core/schema/ZodSchema.cjs");
+const streamStructure_js_1 = require("../../model-function/generate-structure/streamStructure.cjs");
+const streamText_js_1 = require("../../model-function/generate-text/streamText.cjs");
+const StreamingTestServer_js_1 = require("../../test/StreamingTestServer.cjs");
+const arrayFromAsync_js_1 = require("../../test/arrayFromAsync.cjs");
+const OpenAIApiConfiguration_js_1 = require("./OpenAIApiConfiguration.cjs");
+const OpenAIChatModel_js_1 = require("./OpenAIChatModel.cjs");
+describe("streamText", () => {
+    const server = new StreamingTestServer_js_1.StreamingTestServer("https://api.openai.com/v1/chat/completions");
+    server.setupTestEnvironment();
+    it("should return only values from the first choice when using streamText", async () => {
+        server.responseChunks = [
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":0,"delta":{"role":"assistant","content":""},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"A"},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":1,"delta":{"role":"assistant","content":""},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":1,"delta":{"content":"B"},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":0,"delta":{},"finish_reason":"stop"}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":1,"delta":{},"finish_reason":"stop"}]}\n\n`,
+            "data: [DONE]\n\n",
+        ];
+        const stream = await (0, streamText_js_1.streamText)(new OpenAIChatModel_js_1.OpenAIChatModel({
+            api: new OpenAIApiConfiguration_js_1.OpenAIApiConfiguration({ apiKey: "test-key" }),
+            model: "gpt-3.5-turbo",
+            numberOfGenerations: 2,
+        }).withTextPrompt(), "test prompt");
+        expect(await (0, arrayFromAsync_js_1.arrayFromAsync)(stream)).toStrictEqual(["A"]);
+    });
+});
+describe("streamStructure", () => {
+    const server = new StreamingTestServer_js_1.StreamingTestServer("https://api.openai.com/v1/chat/completions");
+    server.setupTestEnvironment();
+    it("should return a text stream", async () => {
+        server.responseChunks = [
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"role":"assistant","content":null,` +
+                `"function_call":{"name":"generateCharacter","arguments":""}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"{\\n"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":" "}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":" \\""}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"name"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"\\":\\""}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"M"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"ike\\"\\n"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"}"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{},"logprobs":null,"finish_reason":"stop"}]}\n\n`,
+            `data: [DONE]\n\n`,
+        ];
+        const stream = await (0, streamStructure_js_1.streamStructure)(new OpenAIChatModel_js_1.OpenAIChatModel({
+            api: new OpenAIApiConfiguration_js_1.OpenAIApiConfiguration({ apiKey: "test-key" }),
+            model: "gpt-3.5-turbo",
+        })
+            .asFunctionCallStructureGenerationModel({
+            fnName: "generateCharacter",
+            fnDescription: "Generate character descriptions.",
+        })
+            .withTextPrompt(), (0, ZodSchema_js_1.zodSchema)(zod_1.z.object({ name: zod_1.z.string() })), "generate a name");
+        // note: space moved to last chunk bc of trimming
+        expect(await (0, arrayFromAsync_js_1.arrayFromAsync)(stream)).toStrictEqual([
+            { isComplete: false, value: {} },
+            { isComplete: false, value: { name: "" } },
+            { isComplete: false, value: { name: "M" } },
+            { isComplete: false, value: { name: "Mike" } },
+            { isComplete: true, value: { name: "Mike" } },
+        ]);
+    });
+});

package/model-provider/openai/OpenAIChatModel.test.d.ts

@@ -0,0 +1 @@
+export {};

package/model-provider/openai/OpenAIChatModel.test.js

@@ -0,0 +1,92 @@
+import { z } from "zod";
+import { zodSchema } from "../../core/schema/ZodSchema.js";
+import { streamStructure } from "../../model-function/generate-structure/streamStructure.js";
+import { streamText } from "../../model-function/generate-text/streamText.js";
+import { StreamingTestServer } from "../../test/StreamingTestServer.js";
+import { arrayFromAsync } from "../../test/arrayFromAsync.js";
+import { OpenAIApiConfiguration } from "./OpenAIApiConfiguration.js";
+import { OpenAIChatModel } from "./OpenAIChatModel.js";
+describe("streamText", () => {
+    const server = new StreamingTestServer("https://api.openai.com/v1/chat/completions");
+    server.setupTestEnvironment();
+    it("should return only values from the first choice when using streamText", async () => {
+        server.responseChunks = [
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":0,"delta":{"role":"assistant","content":""},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":0,"delta":{"content":"A"},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":1,"delta":{"role":"assistant","content":""},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":1,"delta":{"content":"B"},"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":0,"delta":{},"finish_reason":"stop"}]}\n\n`,
+            `data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1702657020,"model":"gpt-3.5-turbo-0613",` +
+                `"system_fingerprint":null,"choices":[{"index":1,"delta":{},"finish_reason":"stop"}]}\n\n`,
+            "data: [DONE]\n\n",
+        ];
+        const stream = await streamText(new OpenAIChatModel({
+            api: new OpenAIApiConfiguration({ apiKey: "test-key" }),
+            model: "gpt-3.5-turbo",
+            numberOfGenerations: 2,
+        }).withTextPrompt(), "test prompt");
+        expect(await arrayFromAsync(stream)).toStrictEqual(["A"]);
+    });
+});
+describe("streamStructure", () => {
+    const server = new StreamingTestServer("https://api.openai.com/v1/chat/completions");
+    server.setupTestEnvironment();
+    it("should return a text stream", async () => {
+        server.responseChunks = [
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"role":"assistant","content":null,` +
+                `"function_call":{"name":"generateCharacter","arguments":""}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"{\\n"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":" "}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":" \\""}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"name"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"\\":\\""}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"M"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"ike\\"\\n"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{"function_call":{"arguments":"}"}},"logprobs":null,"finish_reason":null}]}\n\n`,
+            `data: {"id":"chatcmpl-8ZhZtizjTCGmZaPRwyUiuDJ1DYUD0","object":"chat.completion.chunk",` +
+                `"created":1703519685,"model":"gpt-3.5-turbo-0613","system_fingerprint":null,` +
+                `"choices":[{"index":0,"delta":{},"logprobs":null,"finish_reason":"stop"}]}\n\n`,
+            `data: [DONE]\n\n`,
+        ];
+        const stream = await streamStructure(new OpenAIChatModel({
+            api: new OpenAIApiConfiguration({ apiKey: "test-key" }),
+            model: "gpt-3.5-turbo",
+        })
+            .asFunctionCallStructureGenerationModel({
+            fnName: "generateCharacter",
+            fnDescription: "Generate character descriptions.",
+        })
+            .withTextPrompt(), zodSchema(z.object({ name: z.string() })), "generate a name");
+        // note: space moved to last chunk bc of trimming
+        expect(await arrayFromAsync(stream)).toStrictEqual([
+            { isComplete: false, value: {} },
+            { isComplete: false, value: { name: "" } },
+            { isComplete: false, value: { name: "M" } },
+            { isComplete: false, value: { name: "Mike" } },
+            { isComplete: true, value: { name: "Mike" } },
+        ]);
+    });
+});

package/model-provider/openai/OpenAIChatPromptTemplate.cjs

@@ -0,0 +1,114 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.chat = exports.instruction = exports.text = exports.identity = void 0;
+const OpenAIChatMessage_js_1 = require("./OpenAIChatMessage.cjs");
+/**
+ * OpenAIMessage[] identity chat format.
+ */
+function identity() {
+    return { format: (prompt) => prompt, stopSequences: [] };
+}
+exports.identity = identity;
+/**
+ * Formats a text prompt as an OpenAI chat prompt.
+ */
+function text() {
+    return {
+        format: (prompt) => [OpenAIChatMessage_js_1.OpenAIChatMessage.user(prompt)],
+        stopSequences: [],
+    };
+}
+exports.text = text;
+/**
+ * Formats an instruction prompt as an OpenAI chat prompt.
+ */
+function instruction() {
+    return {
+        format(prompt) {
+            const messages = [];
+            if (prompt.system != null) {
+                messages.push(OpenAIChatMessage_js_1.OpenAIChatMessage.system(prompt.system));
+            }
+            messages.push(OpenAIChatMessage_js_1.OpenAIChatMessage.user(prompt.instruction));
+            return messages;
+        },
+        stopSequences: [],
+    };
+}
+exports.instruction = instruction;
+/**
+ * Formats a chat prompt as an OpenAI chat prompt.
+ */
+function chat() {
+    return {
+        format(prompt) {
+            const messages = [];
+            if (prompt.system != null) {
+                messages.push(OpenAIChatMessage_js_1.OpenAIChatMessage.system(prompt.system));
+            }
+            for (const { role, content } of prompt.messages) {
+                switch (role) {
+                    case "user": {
+                        messages.push(OpenAIChatMessage_js_1.OpenAIChatMessage.user(content));
+                        break;
+                    }
+                    case "assistant": {
+                        if (typeof content === "string") {
+                            messages.push(OpenAIChatMessage_js_1.OpenAIChatMessage.assistant(content));
+                        }
+                        else {
+                            let text = "";
+                            const toolCalls = [];
+                            for (const part of content) {
+                                switch (part.type) {
+                                    case "text": {
+                                        text += part.text;
+                                        break;
+                                    }
+                                    case "tool-call": {
+                                        toolCalls.push({
+                                            id: part.id,
+                                            type: "function",
+                                            function: {
+                                                name: part.name,
+                                                arguments: JSON.stringify(part.args),
+                                            },
+                                        });
+                                        break;
+                                    }
+                                    default: {
+                                        const _exhaustiveCheck = part;
+                                        throw new Error(`Unsupported part: ${_exhaustiveCheck}`);
+                                    }
+                                }
+                            }
+                            messages.push({
+                                role: "assistant",
+                                content: text,
+                                tool_calls: toolCalls,
+                            });
+                        }
+                        break;
+                    }
+                    case "tool": {
+                        for (const toolResponse of content) {
+                            messages.push({
+                                role: "tool",
+                                tool_call_id: toolResponse.id,
+                                content: JSON.stringify(toolResponse.response),
+                            });
+                        }
+                        break;
+                    }
+                    default: {
+                        const _exhaustiveCheck = role;
+                        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
+                    }
+                }
+            }
+            return messages;
+        },
+        stopSequences: [],
+    };
+}
+exports.chat = chat;

package/model-provider/openai/{chat/OpenAIChatPromptTemplate.d.ts → OpenAIChatPromptTemplate.d.ts}

@@ -1,6 +1,6 @@
-import { TextGenerationPromptTemplate } from "../../../model-function/generate-text/TextGenerationPromptTemplate.js";
-import { ChatPrompt } from "../../../model-function/generate-text/prompt-template/ChatPrompt.js";
-import { InstructionPrompt } from "../../../model-function/generate-text/prompt-template/InstructionPrompt.js";
+import { TextGenerationPromptTemplate } from "../../model-function/generate-text/TextGenerationPromptTemplate.js";
+import { ChatPrompt } from "../../model-function/generate-text/prompt-template/ChatPrompt.js";
+import { InstructionPrompt } from "../../model-function/generate-text/prompt-template/InstructionPrompt.js";
 import { OpenAIChatPrompt } from "./AbstractOpenAIChatModel.js";
 /**
  * OpenAIMessage[] identity chat format.

package/model-provider/openai/OpenAIChatPromptTemplate.js

@@ -0,0 +1,107 @@
+import { OpenAIChatMessage } from "./OpenAIChatMessage.js";
+/**
+ * OpenAIMessage[] identity chat format.
+ */
+export function identity() {
+    return { format: (prompt) => prompt, stopSequences: [] };
+}
+/**
+ * Formats a text prompt as an OpenAI chat prompt.
+ */
+export function text() {
+    return {
+        format: (prompt) => [OpenAIChatMessage.user(prompt)],
+        stopSequences: [],
+    };
+}
+/**
+ * Formats an instruction prompt as an OpenAI chat prompt.
+ */
+export function instruction() {
+    return {
+        format(prompt) {
+            const messages = [];
+            if (prompt.system != null) {
+                messages.push(OpenAIChatMessage.system(prompt.system));
+            }
+            messages.push(OpenAIChatMessage.user(prompt.instruction));
+            return messages;
+        },
+        stopSequences: [],
+    };
+}
+/**
+ * Formats a chat prompt as an OpenAI chat prompt.
+ */
+export function chat() {
+    return {
+        format(prompt) {
+            const messages = [];
+            if (prompt.system != null) {
+                messages.push(OpenAIChatMessage.system(prompt.system));
+            }
+            for (const { role, content } of prompt.messages) {
+                switch (role) {
+                    case "user": {
+                        messages.push(OpenAIChatMessage.user(content));
+                        break;
+                    }
+                    case "assistant": {
+                        if (typeof content === "string") {
+                            messages.push(OpenAIChatMessage.assistant(content));
+                        }
+                        else {
+                            let text = "";
+                            const toolCalls = [];
+                            for (const part of content) {
+                                switch (part.type) {
+                                    case "text": {
+                                        text += part.text;
+                                        break;
+                                    }
+                                    case "tool-call": {
+                                        toolCalls.push({
+                                            id: part.id,
+                                            type: "function",
+                                            function: {
+                                                name: part.name,
+                                                arguments: JSON.stringify(part.args),
+                                            },
+                                        });
+                                        break;
+                                    }
+                                    default: {
+                                        const _exhaustiveCheck = part;
+                                        throw new Error(`Unsupported part: ${_exhaustiveCheck}`);
+                                    }
+                                }
+                            }
+                            messages.push({
+                                role: "assistant",
+                                content: text,
+                                tool_calls: toolCalls,
+                            });
+                        }
+                        break;
+                    }
+                    case "tool": {
+                        for (const toolResponse of content) {
+                            messages.push({
+                                role: "tool",
+                                tool_call_id: toolResponse.id,
+                                content: JSON.stringify(toolResponse.response),
+                            });
+                        }
+                        break;
+                    }
+                    default: {
+                        const _exhaustiveCheck = role;
+                        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
+                    }
+                }
+            }
+            return messages;
+        },
+        stopSequences: [],
+    };
+}