@celljs/ai-openai 3.4.4

package/lib/common/model/openai-chat-model.spec.js

@@ -0,0 +1,82 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const chai_1 = require("chai");
+require("../index");
+const openai_chat_model_1 = require("./openai-chat-model");
+const ai_core_1 = require("@celljs/ai-core");
+const test_container_1 = require("../test/test-container");
+const api_protocol_1 = require("../api/api-protocol");
+const mock_openai_api_1 = require("../test/mock-openai-api");
+const container = (0, test_container_1.createContainer)();
+describe('OpenAIChatModel', () => {
+    let openAIChatModel;
+    let promptTemplate;
+    before(() => {
+        container.rebind(api_protocol_1.OpenAIAPI).to(mock_openai_api_1.MockOpenAIAPI).inSingletonScope();
+        openAIChatModel = container.get(openai_chat_model_1.OpenAIChatModel);
+        promptTemplate = container.get(ai_core_1.PromptTemplate);
+        const functionCallbackRegister = container.get(ai_core_1.FunctionCallbackRegister);
+        functionCallbackRegister.register({
+            name: 'getWeather',
+            description: '',
+            inputTypeSchema: '{}',
+            call: async () => ''
+        });
+    });
+    describe('call', () => {
+        it('should return a ChatResponse for a given prompt', async () => {
+            const prompt = await promptTemplate.create('Hello', { chatOptions: { model: 'gpt-4' } });
+            const response = await openAIChatModel.call(prompt);
+            (0, chai_1.expect)(response).to.have.property('results');
+            (0, chai_1.expect)(response.results.length).to.be.greaterThan(0);
+            (0, chai_1.expect)(response.results[0].output).to.be.instanceOf(ai_core_1.AssistantMessage);
+        });
+        it('should handle tool calls correctly', async () => {
+            const prompt = await promptTemplate.create('What is the weather today?', {
+                chatOptions: {
+                    model: 'gpt-4',
+                    isProxyToolCalls: true,
+                    functions: new Set(['getWeather'])
+                }
+            });
+            const response = await openAIChatModel.call(prompt);
+            (0, chai_1.expect)(response).to.have.property('results');
+            (0, chai_1.expect)(response.results.length).to.be.greaterThan(0);
+            (0, chai_1.expect)(response.metadata).to.have.property('model');
+            (0, chai_1.expect)(response.metadata).to.have.property('usage');
+        });
+    });
+    describe('stream', () => {
+        it('should return an Observable of ChatResponse for a given prompt', done => {
+            (async () => {
+                const prompt = await promptTemplate.create('Hello', { chatOptions: { model: 'gpt-4' } });
+                const response$ = await openAIChatModel.stream(prompt);
+                response$.subscribe({
+                    next: response => {
+                        (0, chai_1.expect)(response).to.have.property('results');
+                        (0, chai_1.expect)(response.results.length).to.be.greaterThan(0);
+                        (0, chai_1.expect)(response.results[0].output).to.be.instanceOf(ai_core_1.AssistantMessage);
+                    },
+                    complete: () => done()
+                });
+            })();
+        });
+        it('should throw an error when trying to stream with audio output', async () => {
+            const prompt = await promptTemplate.create('Hello', {
+                chatOptions: {
+                    model: 'gpt-4',
+                    outputModalities: ['audio']
+                }
+            });
+            try {
+                await openAIChatModel.stream(prompt);
+                chai_1.expect.fail('Should have thrown an error');
+            }
+            catch (e) {
+                (0, chai_1.expect)(e).to.be.instanceOf(Error);
+                (0, chai_1.expect)(e.message).to.include('Audio output is not supported');
+            }
+        });
+    });
+});
+//# sourceMappingURL=openai-chat-model.spec.js.map

package/lib/common/model/openai-chat-model.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-chat-model.spec.js","sourceRoot":"","sources":["../../../src/common/model/openai-chat-model.spec.ts"],"names":[],"mappings":";;AAAA,+BAA8B;AAC9B,oBAAkB;AAClB,2DAAsD;AACtD,6CAA0G;AAC1G,2DAAyD;AACzD,sDAAgD;AAChD,6DAAwD;AAExD,MAAM,SAAS,GAAG,IAAA,gCAAe,GAAE,CAAC;AAEpC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,IAAI,eAAgC,CAAC;IACrC,IAAI,cAA8B,CAAC;IAEnC,MAAM,CAAC,GAAG,EAAE;QACR,SAAS,CAAC,MAAM,CAAC,wBAAS,CAAC,CAAC,EAAE,CAAC,+BAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACjE,eAAe,GAAG,SAAS,CAAC,GAAG,CAAC,mCAAe,CAAC,CAAC;QACjD,cAAc,GAAG,SAAS,CAAC,GAAG,CAAC,wBAAc,CAAC,CAAC;QAC/C,MAAM,wBAAwB,GAAG,SAAS,CAAC,GAAG,CAA2B,kCAAwB,CAAC,CAAC;QACnG,wBAAwB,CAAC,QAAQ,CAAC;YAC9B,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,EAAE;YACf,eAAe,EAAE,IAAI;YACrB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;SACvB,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QAClB,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YACzF,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpD,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAC7C,IAAA,aAAM,EAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACrD,IAAA,aAAM,EAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,0BAAgB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,4BAA4B,EAAE;gBACrE,WAAW,EAAE;oBACT,KAAK,EAAE,OAAO;oBACd,gBAAgB,EAAE,IAAI;oBACtB,SAAS,EAAE,IAAI,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;iBACtB;aACnB,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpD,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAC7C,IAAA,aAAM,EAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YACrD,IAAA,aAAM,EAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACpD,IAAA,aAAM,EAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,gEAAgE,EAAE,IAAI,CAAC,EAAE;YACxE,CAAC,KAAK,IAAI,EAAE;gBACR,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACzF,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACvD,SAAS,CAAC,SAAS,CAAC;oBAChB,IAAI,EAAE,QAAQ,CAAC,EAAE;wBACb,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;wBAC7C,IAAA,aAAM,EAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;wBACrD,IAAA,aAAM,EAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,0BAAgB,CAAC,CAAC;oBAC1E,CAAC;oBACD,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE;iBACzB,CAAC,CAAC;YACP,CAAC,CAAC,EAAE,CAAC;QACT,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,OAAO,EAAE;gBAChD,WAAW,EAAE;oBACT,KAAK,EAAE,OAAO;oBACd,gBAAgB,EAAE,CAAC,OAAO,CAAC;iBACf;aACnB,CAAC,CAAC;YACH,IAAI;gBACA,MAAM,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACrC,aAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;aAC9C;YAAC,OAAO,CAAC,EAAE;gBACR,IAAA,aAAM,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAClC,IAAA,aAAM,EAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;aACjE;QACL,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/lib/common/model/openai-chat-options.d.ts

@@ -0,0 +1,159 @@
+import { FunctionCallback, FunctionCallingOptions } from '@celljs/ai-core';
+import { OpenAIModel } from '../api/openai-model';
+import { AudioParameters, FunctionTool, ResponseFormat, StreamOptions } from '../api';
+/**
+ * Options for the OpenAI Chat API.
+ */
+export declare class OpenAIChatOptions implements FunctionCallingOptions {
+    static readonly DEFAULT_MODEL = OpenAIModel.GPT_3_5_TURBO;
+    /**
+     * ID of the model to use.
+     */
+    model?: string;
+    /**
+     * What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output
+     * more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend
+     * altering this or top_p but not both.
+     */
+    temperature?: number;
+    /**
+     * The maximum number of tokens to generate in the chat completion. The total length of input
+     * tokens and generated tokens is limited by the model's context length.
+     */
+    maxTokens?: number;
+    /**
+     * An alternative to sampling with temperature, called nucleus sampling, where the model considers the
+     * results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10%
+     * probability mass are considered. We generally recommend altering this or temperature but not both.
+     */
+    topP?: number;
+    /**
+     * Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they
+     * appear in the text so far, increasing the model's likelihood to talk about new topics.
+     */
+    presencePenalty?: number;
+    /**
+     * Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing
+     * frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.
+     */
+    frequencyPenalty?: number;
+    /**
+     * How many chat completion choices to generate for each input message. Note that you will be charged based
+     * on the number of generated tokens across all of the choices. Keep n as 1 to minimize costs.
+     */
+    n?: number;
+    /**
+     * Output types that you would like the model to generate for this request.
+     * Most models are capable of generating text, which is the default.
+     */
+    outputModalities?: string[];
+    /**
+     * Audio parameters for the audio generation. Required when audio output is requested with
+     * modalities: ["audio"]
+     */
+    outputAudio?: AudioParameters;
+    /**
+     * Modify the likelihood of specified tokens appearing in the completion. Accepts a JSON object
+     * that maps tokens (specified by their token ID in the tokenizer) to an associated bias value from -100 to 100.
+     */
+    logitBias?: Record<string, number>;
+    /**
+     * Whether to return log probabilities of the output tokens or not. If true, returns the log probabilities
+     * of each output token returned in the 'content' of 'message'.
+     */
+    logprobs?: boolean;
+    /**
+     * An integer between 0 and 5 specifying the number of most likely tokens to return at each token position,
+     * each with an associated log probability. 'logprobs' must be set to 'true' if this parameter is used.
+     */
+    topLogprobs?: number;
+    /**
+     * An object specifying the format that the model must output. Setting to { "type":
+     * "json_object" } enables JSON mode, which guarantees the message the model generates is valid JSON.
+     */
+    responseFormat?: ResponseFormat;
+    /**
+     * Options for streaming response. Included in the API only if streaming-mode completion is requested.
+     */
+    streamOptions?: StreamOptions;
+    /**
+     * This feature is in Beta. If specified, our system will make a best effort to sample
+     * deterministically, such that repeated requests with the same seed and parameters should return the same result.
+     */
+    seed?: number;
+    /**
+     * Up to 4 sequences where the API will stop generating further tokens.
+     */
+    stop?: string[];
+    /**
+     * A list of tools the model may call. Currently, only functions are supported as a tool. Use this to
+     * provide a list of functions the model may generate JSON inputs for.
+     */
+    tools?: FunctionTool[];
+    /**
+     * Controls which (if any) function is called by the model.
+     */
+    toolChoice?: any;
+    /**
+     * A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.
+     */
+    user?: string;
+    /**
+     * Whether to enable parallel function calling during tool use.
+     */
+    parallelToolCalls?: boolean;
+    /**
+     * Tool Function Callbacks to register with the ChatModel.
+     */
+    functionCallbacks: FunctionCallback[];
+    /**
+     * List of functions, identified by their names, to configure for function calling in
+     * the chat completion requests.
+     */
+    functions: Set<string>;
+    /**
+     * If true, the AI will not handle the function calls internally, but will proxy them to the client.
+     */
+    proxyToolCalls?: boolean;
+    /**
+     * Optional HTTP headers to be added to the chat completion request.
+     */
+    httpHeaders: Record<string, string>;
+    /**
+     * Tool context data.
+     */
+    toolContext?: Record<string, any>;
+    copy(): OpenAIChatOptions;
+    static options(fromOptions: OpenAIChatOptions): OpenAIChatOptions;
+    static builder(): OpenAIOptionsBuilder;
+}
+export declare class OpenAIOptionsBuilder {
+    private options;
+    withModel(model: string): OpenAIOptionsBuilder;
+    withTemperature(temperature: number): OpenAIOptionsBuilder;
+    withMaxTokens(maxTokens: number): OpenAIOptionsBuilder;
+    withTopP(topP: number): OpenAIOptionsBuilder;
+    withPresencePenalty(presencePenalty: number): OpenAIOptionsBuilder;
+    withFrequencyPenalty(frequencyPenalty: number): OpenAIOptionsBuilder;
+    withN(n: number): OpenAIOptionsBuilder;
+    withOutputModalities(modalities: string[]): OpenAIOptionsBuilder;
+    withOutputAudio(audio: AudioParameters): OpenAIOptionsBuilder;
+    withLogitBias(bias: Record<string, number>): OpenAIOptionsBuilder;
+    withLogprobs(logprobs: boolean): OpenAIOptionsBuilder;
+    withTopLogprobs(topLogprobs: number): OpenAIOptionsBuilder;
+    withResponseFormat(format: ResponseFormat): OpenAIOptionsBuilder;
+    withStreamOptions(options: StreamOptions): OpenAIOptionsBuilder;
+    withSeed(seed: number): OpenAIOptionsBuilder;
+    withStop(stop: string[]): OpenAIOptionsBuilder;
+    withTools(tools: FunctionTool[]): OpenAIOptionsBuilder;
+    withToolChoice(choice: any): OpenAIOptionsBuilder;
+    withUser(user: string): OpenAIOptionsBuilder;
+    withParallelToolCalls(parallel: boolean): OpenAIOptionsBuilder;
+    withFunctionCallbacks(callbacks: FunctionCallback[]): OpenAIOptionsBuilder;
+    withFunctions(functions: Set<string>): OpenAIOptionsBuilder;
+    withProxyToolCalls(proxy: boolean): OpenAIOptionsBuilder;
+    withHttpHeaders(headers: Record<string, string>): OpenAIOptionsBuilder;
+    withToolContext(context: Record<string, any>): OpenAIOptionsBuilder;
+    build(): OpenAIChatOptions;
+}
+//# sourceMappingURL=openai-chat-options.d.ts.map

package/lib/common/model/openai-chat-options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-chat-options.d.ts","sourceRoot":"","sources":["../../../src/common/model/openai-chat-options.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAE3E,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAEtF;;GAEG;AACH,qBACa,iBAAkB,YAAW,sBAAsB;IAC5D,MAAM,CAAC,QAAQ,CAAC,aAAa,6BAA6B;IAE1D;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;OAGG;IAEH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;;OAIG;IAEH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;OAGG;IAEH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;;OAGG;IAEH,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAE1B;;;OAGG;IACH,CAAC,CAAC,EAAE,MAAM,CAAC;IAEX;;;OAGG;IAEH,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAE5B;;;OAGG;IAEH,WAAW,CAAC,EAAE,eAAe,CAAC;IAE9B;;;OAGG;IAEH,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAEnC;;;OAGG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IAEH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;OAGG;IAEH,cAAc,CAAC,EAAE,cAAc,CAAC;IAEhC;;OAEG;IAEH,aAAa,CAAC,EAAE,aAAa,CAAC;IAE9B;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAEhB;;;OAGG;IACH,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;IAEvB;;OAEG;IAEH,UAAU,CAAC,EAAE,GAAG,CAAC;IAEjB;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;OAEG;IAEH,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAE5B;;OAEG;IAEH,iBAAiB,EAAE,gBAAgB,EAAE,CAAM;IAE3C;;;OAGG;IAEH,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,CAAqB;IAE3C;;OAEG;IAEH,cAAc,CAAC,EAAE,OAAO,CAAC;IAEzB;;OAEG;IAEH,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAM;IAEzC;;OAEG;IAEH,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAElC,IAAI,IAAI,iBAAiB;IAIzB,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,iBAAiB,GAAG,iBAAiB;IASjE,MAAM,CAAC,OAAO,IAAI,oBAAoB;CAGzC;AAED,qBAAa,oBAAoB;IAC7B,OAAO,CAAC,OAAO,CAA2B;IAE1C,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,oBAAoB;IAK9C,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,oBAAoB;IAK1D,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,oBAAoB;IAKtD,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,oBAAoB;IAK5C,mBAAmB,CAAC,eAAe,EAAE,MAAM,GAAG,oBAAoB;IAKlE,oBAAoB,CAAC,gBAAgB,EAAE,MAAM,GAAG,oBAAoB;IAKpE,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,oBAAoB;IAKtC,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,oBAAoB;IAKhE,eAAe,CAAC,KAAK,EAAE,eAAe,GAAG,oBAAoB;IAK7D,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,oBAAoB;IAKjE,YAAY,CAAC,QAAQ,EAAE,OAAO,GAAG,oBAAoB;IAKrD,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,oBAAoB;IAK1D,kBAAkB,CAAC,MAAM,EAAE,cAAc,GAAG,oBAAoB;IAKhE,iBAAiB,CAAC,OAAO,EAAE,aAAa,GAAG,oBAAoB;IAK/D,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,oBAAoB;IAK5C,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,oBAAoB;IAK9C,SAAS,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,oBAAoB;IAKtD,cAAc,CAAC,MAAM,EAAE,GAAG,GAAG,oBAAoB;IAKjD,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,oBAAoB;IAK5C,qBAAqB,CAAC,QAAQ,EAAE,OAAO,GAAG,oBAAoB;IAK9D,qBAAqB,CAAC,SAAS,EAAE,gBAAgB,EAAE,GAAG,oBAAoB;IAK1E,aAAa,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG,oBAAoB;IAK3D,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,oBAAoB;IAKxD,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,oBAAoB;IAKtE,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,oBAAoB;IASnE,KAAK,IAAI,iBAAiB;CAG7B"}

package/lib/common/model/openai-chat-options.js

@@ -0,0 +1,231 @@
+"use strict";
+var OpenAIChatOptions_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIOptionsBuilder = exports.OpenAIChatOptions = void 0;
+const tslib_1 = require("tslib");
+const class_transformer_1 = require("class-transformer");
+const openai_model_1 = require("../api/openai-model");
+const core_1 = require("@celljs/core");
+const api_1 = require("../api");
+/**
+ * Options for the OpenAI Chat API.
+ */
+let OpenAIChatOptions = OpenAIChatOptions_1 = class OpenAIChatOptions {
+    constructor() {
+        /**
+         * Tool Function Callbacks to register with the ChatModel.
+         */
+        this.functionCallbacks = [];
+        /**
+         * List of functions, identified by their names, to configure for function calling in
+         * the chat completion requests.
+         */
+        this.functions = new Set();
+        /**
+         * Optional HTTP headers to be added to the chat completion request.
+         */
+        this.httpHeaders = {};
+    }
+    copy() {
+        return OpenAIChatOptions_1.options(this);
+    }
+    static options(fromOptions) {
+        const options = new OpenAIChatOptions_1();
+        Object.assign(options, fromOptions);
+        options.functionCallbacks = [...(fromOptions.functionCallbacks || [])];
+        options.functions = new Set(fromOptions.functions);
+        options.httpHeaders = { ...fromOptions.httpHeaders };
+        return options;
+    }
+    static builder() {
+        return new OpenAIOptionsBuilder();
+    }
+};
+OpenAIChatOptions.DEFAULT_MODEL = openai_model_1.OpenAIModel.GPT_3_5_TURBO;
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'max_tokens' }),
+    tslib_1.__metadata("design:type", Number)
+], OpenAIChatOptions.prototype, "maxTokens", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'top_p' }),
+    tslib_1.__metadata("design:type", Number)
+], OpenAIChatOptions.prototype, "topP", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'presence_penalty' }),
+    tslib_1.__metadata("design:type", Number)
+], OpenAIChatOptions.prototype, "presencePenalty", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'frequency_penalty' }),
+    tslib_1.__metadata("design:type", Number)
+], OpenAIChatOptions.prototype, "frequencyPenalty", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'modalities' }),
+    tslib_1.__metadata("design:type", Array)
+], OpenAIChatOptions.prototype, "outputModalities", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'audio' }),
+    tslib_1.__metadata("design:type", api_1.AudioParameters)
+], OpenAIChatOptions.prototype, "outputAudio", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'logit_bias' }),
+    tslib_1.__metadata("design:type", Object)
+], OpenAIChatOptions.prototype, "logitBias", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'top_logprobs' }),
+    tslib_1.__metadata("design:type", Number)
+], OpenAIChatOptions.prototype, "topLogprobs", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'response_format' }),
+    tslib_1.__metadata("design:type", api_1.ResponseFormat)
+], OpenAIChatOptions.prototype, "responseFormat", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'stream_options' }),
+    tslib_1.__metadata("design:type", api_1.StreamOptions)
+], OpenAIChatOptions.prototype, "streamOptions", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'tool_choice' }),
+    tslib_1.__metadata("design:type", Object)
+], OpenAIChatOptions.prototype, "toolChoice", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'parallel_tool_calls' }),
+    tslib_1.__metadata("design:type", Boolean)
+], OpenAIChatOptions.prototype, "parallelToolCalls", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Exclude)(),
+    tslib_1.__metadata("design:type", Array)
+], OpenAIChatOptions.prototype, "functionCallbacks", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Exclude)(),
+    tslib_1.__metadata("design:type", Set)
+], OpenAIChatOptions.prototype, "functions", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Exclude)(),
+    tslib_1.__metadata("design:type", Boolean)
+], OpenAIChatOptions.prototype, "proxyToolCalls", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Exclude)(),
+    tslib_1.__metadata("design:type", Object)
+], OpenAIChatOptions.prototype, "httpHeaders", void 0);
+tslib_1.__decorate([
+    (0, class_transformer_1.Exclude)(),
+    tslib_1.__metadata("design:type", Object)
+], OpenAIChatOptions.prototype, "toolContext", void 0);
+OpenAIChatOptions = OpenAIChatOptions_1 = tslib_1.__decorate([
+    (0, core_1.Constant)(OpenAIChatOptions_1, new OpenAIChatOptions_1())
+], OpenAIChatOptions);
+exports.OpenAIChatOptions = OpenAIChatOptions;
+class OpenAIOptionsBuilder {
+    constructor() {
+        this.options = new OpenAIChatOptions();
+    }
+    withModel(model) {
+        this.options.model = model;
+        return this;
+    }
+    withTemperature(temperature) {
+        this.options.temperature = temperature;
+        return this;
+    }
+    withMaxTokens(maxTokens) {
+        this.options.maxTokens = maxTokens;
+        return this;
+    }
+    withTopP(topP) {
+        this.options.topP = topP;
+        return this;
+    }
+    withPresencePenalty(presencePenalty) {
+        this.options.presencePenalty = presencePenalty;
+        return this;
+    }
+    withFrequencyPenalty(frequencyPenalty) {
+        this.options.frequencyPenalty = frequencyPenalty;
+        return this;
+    }
+    withN(n) {
+        this.options.n = n;
+        return this;
+    }
+    withOutputModalities(modalities) {
+        this.options.outputModalities = modalities;
+        return this;
+    }
+    withOutputAudio(audio) {
+        this.options.outputAudio = audio;
+        return this;
+    }
+    withLogitBias(bias) {
+        this.options.logitBias = bias;
+        return this;
+    }
+    withLogprobs(logprobs) {
+        this.options.logprobs = logprobs;
+        return this;
+    }
+    withTopLogprobs(topLogprobs) {
+        this.options.topLogprobs = topLogprobs;
+        return this;
+    }
+    withResponseFormat(format) {
+        this.options.responseFormat = format;
+        return this;
+    }
+    withStreamOptions(options) {
+        this.options.streamOptions = options;
+        return this;
+    }
+    withSeed(seed) {
+        this.options.seed = seed;
+        return this;
+    }
+    withStop(stop) {
+        this.options.stop = stop;
+        return this;
+    }
+    withTools(tools) {
+        this.options.tools = tools;
+        return this;
+    }
+    withToolChoice(choice) {
+        this.options.toolChoice = choice;
+        return this;
+    }
+    withUser(user) {
+        this.options.user = user;
+        return this;
+    }
+    withParallelToolCalls(parallel) {
+        this.options.parallelToolCalls = parallel;
+        return this;
+    }
+    withFunctionCallbacks(callbacks) {
+        this.options.functionCallbacks = callbacks;
+        return this;
+    }
+    withFunctions(functions) {
+        this.options.functions = functions;
+        return this;
+    }
+    withProxyToolCalls(proxy) {
+        this.options.proxyToolCalls = proxy;
+        return this;
+    }
+    withHttpHeaders(headers) {
+        this.options.httpHeaders = headers;
+        return this;
+    }
+    withToolContext(context) {
+        if (!this.options.toolContext) {
+            this.options.toolContext = context;
+        }
+        else {
+            Object.assign(this.options.toolContext, context);
+        }
+        return this;
+    }
+    build() {
+        return OpenAIChatOptions.options(this.options);
+    }
+}
+exports.OpenAIOptionsBuilder = OpenAIOptionsBuilder;
+//# sourceMappingURL=openai-chat-options.js.map

package/lib/common/model/openai-chat-options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-chat-options.js","sourceRoot":"","sources":["../../../src/common/model/openai-chat-options.ts"],"names":[],"mappings":";;;;;AACA,yDAAoD;AACpD,sDAAkD;AAClD,uCAAwC;AACxC,gCAAsF;AAEtF;;GAEG;AAEI,IAAM,iBAAiB,yBAAvB,MAAM,iBAAiB;IAAvB;QAmIH;;WAEG;QAEH,sBAAiB,GAAuB,EAAE,CAAC;QAE3C;;;WAGG;QAEH,cAAS,GAAgB,IAAI,GAAG,EAAU,CAAC;QAQ3C;;WAEG;QAEH,gBAAW,GAA2B,EAAE,CAAC;IAwB7C,CAAC;IAhBG,IAAI;QACA,OAAO,mBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,WAA8B;QACzC,MAAM,OAAO,GAAG,IAAI,mBAAiB,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACpC,OAAO,CAAC,iBAAiB,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,SAAS,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QACnD,OAAO,CAAC,WAAW,GAAG,EAAE,GAAG,WAAW,CAAC,WAAW,EAAE,CAAC;QACrD,OAAO,OAAO,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,OAAO;QACV,OAAO,IAAI,oBAAoB,EAAE,CAAC;IACtC,CAAC;;AAhLe,+BAAa,GAAG,0BAAW,CAAC,aAAa,CAAC;AAkB1D;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;;oDACZ;AAOnB;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;;+CACZ;AAMd;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;;0DACZ;AAMzB;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;;2DACZ;AAY1B;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;;2DACH;AAM5B;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;sCACZ,qBAAe;sDAAC;AAM9B;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;;oDACI;AAYnC;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;;sDACZ;AAMrB;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;sCACnB,oBAAc;yDAAC;AAKhC;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;sCACnB,mBAAa;wDAAC;AAsB9B;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;;qDACf;AAUjB;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC;;4DACZ;AAK5B;IAAC,IAAA,2BAAO,GAAE;;4DACiC;AAM3C;IAAC,IAAA,2BAAO,GAAE;sCACC,GAAG;oDAA6B;AAK3C;IAAC,IAAA,2BAAO,GAAE;;yDACe;AAKzB;IAAC,IAAA,2BAAO,GAAE;;sDAC+B;AAKzC;IAAC,IAAA,2BAAO,GAAE;;sDACwB;AAhKzB,iBAAiB;IAD7B,IAAA,eAAQ,EAAC,mBAAiB,EAAE,IAAI,mBAAiB,EAAE,CAAC;GACxC,iBAAiB,CAkL7B;AAlLY,8CAAiB;AAoL9B,MAAa,oBAAoB;IAAjC;QACY,YAAO,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAsI9C,CAAC;IApIG,SAAS,CAAC,KAAa;QACnB,IAAI,CAAC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;QAC3B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,WAAmB;QAC/B,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;QACvC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,aAAa,CAAC,SAAiB;QAC3B,IAAI,CAAC,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,IAAY;QACjB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,mBAAmB,CAAC,eAAuB;QACvC,IAAI,CAAC,OAAO,CAAC,eAAe,GAAG,eAAe,CAAC;QAC/C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,oBAAoB,CAAC,gBAAwB;QACzC,IAAI,CAAC,OAAO,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACjD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,CAAS;QACX,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC;QACnB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,oBAAoB,CAAC,UAAoB;QACrC,IAAI,CAAC,OAAO,CAAC,gBAAgB,GAAG,UAAU,CAAC;QAC3C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,KAAsB;QAClC,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,KAAK,CAAC;QACjC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,aAAa,CAAC,IAA4B;QACtC,IAAI,CAAC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;QAC9B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,YAAY,CAAC,QAAiB;QAC1B,IAAI,CAAC,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACjC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,WAAmB;QAC/B,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;QACvC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,kBAAkB,CAAC,MAAsB;QACrC,IAAI,CAAC,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC;QACrC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,iBAAiB,CAAC,OAAsB;QACpC,IAAI,CAAC,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC;QACrC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,IAAY;QACjB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,IAAc;QACnB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,SAAS,CAAC,KAAqB;QAC3B,IAAI,CAAC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;QAC3B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,MAAW;QACtB,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,MAAM,CAAC;QACjC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,IAAY;QACjB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,qBAAqB,CAAC,QAAiB;QACnC,IAAI,CAAC,OAAO,CAAC,iBAAiB,GAAG,QAAQ,CAAC;QAC1C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,qBAAqB,CAAC,SAA6B;QAC/C,IAAI,CAAC,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;QAC3C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,aAAa,CAAC,SAAsB;QAChC,IAAI,CAAC,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,kBAAkB,CAAC,KAAc;QAC7B,IAAI,CAAC,OAAO,CAAC,cAAc,GAAG,KAAK,CAAC;QACpC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,OAA+B;QAC3C,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC;QACnC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,eAAe,CAAC,OAA4B;QACxC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE;YAC3B,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,OAAO,CAAC;SACtC;aAAM;YACH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;SACpD;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK;QACD,OAAO,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACnD,CAAC;CACJ;AAvID,oDAuIC"}

package/lib/common/model/openai-embedding-model.d.ts

@@ -0,0 +1,15 @@
+import { EmbeddingModel, EmbeddingOptions, EmbeddingRequest, EmbeddingResponse } from '@celljs/ai-core';
+import { EmbeddingRequest as OpenAIEmbeddingsRequest, OpenAIAPI } from '../api';
+import { Logger } from '@celljs/core';
+import { OpenAIEmbeddingOptions } from './openai-embedding-options';
+export declare class OpenAIEmbeddingModel implements EmbeddingModel {
+    protected readonly defaultOptions: OpenAIEmbeddingOptions;
+    /**
+     * Low-level access to the OpenAI API.
+     */
+    protected readonly chatApi: OpenAIAPI;
+    protected readonly logger: Logger;
+    call(request: EmbeddingRequest): Promise<EmbeddingResponse>;
+    protected createOpenAIEmbeddingsRequest(inputContent: string[], options?: EmbeddingOptions): OpenAIEmbeddingsRequest<string[]>;
+}
+//# sourceMappingURL=openai-embedding-model.d.ts.map

package/lib/common/model/openai-embedding-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.d.ts","sourceRoot":"","sources":["../../../src/common/model/openai-embedding-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAEH,cAAc,EACd,gBAAgB,EAChB,gBAAgB,EAChB,iBAAiB,EAGpB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACH,gBAAgB,IAAI,uBAAuB,EAC3C,SAAS,EACZ,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAsD,MAAM,EAAE,MAAM,cAAc,CAAC;AAC1F,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAEpE,qBACa,oBAAqB,YAAW,cAAc;IAGvD,SAAS,CAAC,QAAQ,CAAC,cAAc,EAAE,sBAAsB,CAAC;IAE1D;;OAEG;IAEH,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC;IAGtC,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAErB,IAAI,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA4BxE,SAAS,CAAC,6BAA6B,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,gBAAgB,GAAG,uBAAuB,CAAC,MAAM,EAAE,CAAC;CAuBjI"}

package/lib/common/model/openai-embedding-model.js

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddingModel = void 0;
+const tslib_1 = require("tslib");
+const ai_core_1 = require("@celljs/ai-core");
+const api_1 = require("../api");
+const core_1 = require("@celljs/core");
+const openai_embedding_options_1 = require("./openai-embedding-options");
+let OpenAIEmbeddingModel = class OpenAIEmbeddingModel {
+    async call(request) {
+        core_1.Assert.notEmpty(request.instructions, 'At least one text is required!');
+        const openAIRequet = this.createOpenAIEmbeddingsRequest(request.instructions, request.options);
+        const openAIResponse = (await this.chatApi.embed(openAIRequet)).body;
+        core_1.Assert.notNull(openAIResponse, 'No embedding data available.');
+        const embeddings = openAIResponse.data.map(embedding => ({
+            embedding: embedding.embedding,
+            index: embedding.index,
+            metadata: ai_core_1.EmbeddingResultMetadata.EMPTY,
+            output: embedding.embedding
+        }));
+        const embeddingResponseMetadata = {
+            model: openAIResponse.model,
+            usage: openAIResponse.usage,
+            extra: {}
+        };
+        core_1.Assert.notEmpty(embeddings, 'No embedding data available.');
+        return {
+            embeddings,
+            result: embeddings[0],
+            results: embeddings,
+            metadata: embeddingResponseMetadata
+        };
+    }
+    createOpenAIEmbeddingsRequest(inputContent, options) {
+        let runtimeOptions = openai_embedding_options_1.OpenAIEmbeddingOptions.builder().build();
+        if (options) {
+            runtimeOptions = Object.assign(runtimeOptions, options);
+        }
+        runtimeOptions = Object.assign(this.defaultOptions, runtimeOptions);
+        // Override the model.
+        if (!runtimeOptions.model) {
+            throw new core_1.IllegalArgumentError('Model is not set!');
+        }
+        const model = runtimeOptions.model;
+        return new api_1.EmbeddingRequest(inputContent, model, runtimeOptions.encodingFormat, runtimeOptions.dimensions, runtimeOptions.user);
+    }
+};
+tslib_1.__decorate([
+    (0, core_1.Autowired)(openai_embedding_options_1.OpenAIEmbeddingOptions),
+    tslib_1.__metadata("design:type", openai_embedding_options_1.OpenAIEmbeddingOptions)
+], OpenAIEmbeddingModel.prototype, "defaultOptions", void 0);
+tslib_1.__decorate([
+    (0, core_1.Autowired)(api_1.OpenAIAPI),
+    tslib_1.__metadata("design:type", Object)
+], OpenAIEmbeddingModel.prototype, "chatApi", void 0);
+tslib_1.__decorate([
+    (0, core_1.Autowired)(core_1.Logger),
+    tslib_1.__metadata("design:type", Object)
+], OpenAIEmbeddingModel.prototype, "logger", void 0);
+OpenAIEmbeddingModel = tslib_1.__decorate([
+    (0, core_1.Component)(ai_core_1.EmbeddingModel)
+], OpenAIEmbeddingModel);
+exports.OpenAIEmbeddingModel = OpenAIEmbeddingModel;
+//# sourceMappingURL=openai-embedding-model.js.map

package/lib/common/model/openai-embedding-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.js","sourceRoot":"","sources":["../../../src/common/model/openai-embedding-model.ts"],"names":[],"mappings":";;;;AAAA,6CAQyB;AACzB,gCAGgB;AAChB,uCAA0F;AAC1F,yEAAoE;AAG7D,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IActB,KAAK,CAAC,IAAI,CAAC,OAAyB;QACvC,aAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,gCAAgC,CAAC,CAAC;QACxE,MAAM,YAAY,GAAG,IAAI,CAAC,6BAA6B,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/F,MAAM,cAAc,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;QACrE,aAAM,CAAC,OAAO,CAAC,cAAc,EAAE,8BAA8B,CAAC,CAAC;QAE/D,MAAM,UAAU,GAAgB,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClE,SAAS,EAAE,SAAS,CAAC,SAAS;YAC9B,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,QAAQ,EAAE,iCAAuB,CAAC,KAAK;YACvC,MAAM,EAAE,SAAS,CAAC,SAAS;SAC9B,CAAC,CAAC,CAAC;QAEJ,MAAM,yBAAyB,GAA8B;YACzD,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,KAAK,EAAE,EAAE;SACZ,CAAC;QACF,aAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,8BAA8B,CAAC,CAAC;QAC5D,OAAO;YACH,UAAU;YACV,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;YACrB,OAAO,EAAE,UAAU;YACnB,QAAQ,EAAE,yBAAyB;SACtC,CAAC;IAEN,CAAC;IAES,6BAA6B,CAAC,YAAsB,EAAE,OAA0B;QACtF,IAAI,cAAc,GAA2B,iDAAsB,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,CAAC;QACtF,IAAI,OAAO,EAAE;YACT,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;SAC3D;QAED,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;QAEpE,sBAAsB;QACtB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE;YACvB,MAAM,IAAI,2BAAoB,CAAC,mBAAmB,CAAC,CAAC;SACvD;QACD,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC;QAEnC,OAAO,IAAI,sBAAuB,CAC9B,YAAY,EACZ,KAAK,EACL,cAAc,CAAC,cAAc,EAC7B,cAAc,CAAC,UAAU,EACzB,cAAc,CAAC,IAAI,CACtB,CAAC;IACN,CAAC;CAEJ,CAAA;AA/DG;IAAC,IAAA,gBAAS,EAAC,iDAAsB,CAAC;sCACC,iDAAsB;4DAAC;AAK1D;IAAC,IAAA,gBAAS,EAAC,eAAS,CAAC;;qDACiB;AAEtC;IAAC,IAAA,gBAAS,EAAC,aAAM,CAAC;;oDACgB;AAZzB,oBAAoB;IADhC,IAAA,gBAAS,EAAC,wBAAc,CAAC;GACb,oBAAoB,CAiEhC;AAjEY,oDAAoB"}

package/lib/common/model/openai-embedding-model.spec.d.ts

@@ -0,0 +1,2 @@
+import '../index';
+//# sourceMappingURL=openai-embedding-model.spec.d.ts.map

package/lib/common/model/openai-embedding-model.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.spec.d.ts","sourceRoot":"","sources":["../../../src/common/model/openai-embedding-model.spec.ts"],"names":[],"mappings":"AACA,OAAO,UAAU,CAAC"}