@celljs/ai-openai 3.4.4

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

@@ -0,0 +1,100 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const chai_1 = require("chai");
+require("../index");
+const openai_embedding_model_1 = require("./openai-embedding-model");
+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 openai_embedding_options_1 = require("./openai-embedding-options");
+const container = (0, test_container_1.createContainer)();
+describe('OpenAIEmbeddingModel', () => {
+    let openAIEmbeddingModel;
+    let defaultOptions;
+    before(() => {
+        container.rebind(api_protocol_1.OpenAIAPI).to(mock_openai_api_1.MockOpenAIAPI).inSingletonScope();
+        openAIEmbeddingModel = container.get(openai_embedding_model_1.OpenAIEmbeddingModel);
+        defaultOptions = container.get(openai_embedding_options_1.OpenAIEmbeddingOptions);
+    });
+    describe('call', () => {
+        it('should return embedding response for given text', async () => {
+            const request = {
+                instructions: ['Hello world', 'Another text sample'],
+                inputs: ['Hello world', 'Another text sample'],
+                options: {
+                    model: 'text-embedding-ada-002'
+                }
+            };
+            const response = await openAIEmbeddingModel.call(request);
+            (0, chai_1.expect)(response).to.have.property('embeddings');
+            (0, chai_1.expect)(response.embeddings).to.be.an('array');
+            (0, chai_1.expect)(response.embeddings.length).to.equal(1);
+            (0, chai_1.expect)(response.embeddings[0].embedding).to.be.an('array');
+            (0, chai_1.expect)(response.metadata).to.have.property('model');
+            (0, chai_1.expect)(response.metadata).to.have.property('usage');
+        });
+        it('should throw error when no text is provided', async () => {
+            const request = {
+                instructions: [],
+                inputs: [],
+                options: {
+                    model: 'text-embedding-ada-002'
+                }
+            };
+            try {
+                await openAIEmbeddingModel.call(request);
+                chai_1.expect.fail('Should throw error for empty instructions');
+            }
+            catch (error) {
+                (0, chai_1.expect)(error.message).to.include('At least one text is required');
+            }
+        });
+        it('should throw error when model is not set', async () => {
+            const request = {
+                instructions: ['Test text'],
+                inputs: ['Test text'],
+                options: {
+                    model: ''
+                }
+            };
+            try {
+                await openAIEmbeddingModel.call(request);
+                chai_1.expect.fail('Should throw error for missing model');
+            }
+            catch (error) {
+                (0, chai_1.expect)(error.message).to.include('Model is not set');
+            }
+        });
+        it('should use default model when not specified in request', async () => {
+            // Temporarily set default model for this test
+            const originalModel = defaultOptions.model;
+            defaultOptions.model = 'text-embedding-ada-002';
+            const request = {
+                instructions: ['Test text'],
+                inputs: ['Test text'],
+                options: {}
+            };
+            const response = await openAIEmbeddingModel.call(request);
+            (0, chai_1.expect)(response.metadata.model).to.equal('text-embedding-ada-002');
+            // Restore original default model
+            defaultOptions.model = originalModel;
+        });
+        it('should pass embedding options correctly', async () => {
+            const request = {
+                instructions: ['Test text'],
+                inputs: ['Test text'],
+                options: {
+                    model: 'text-embedding-ada-002',
+                    dimensions: 512,
+                    encodingFormat: 'float',
+                    user: 'test-user'
+                }
+            };
+            const response = await openAIEmbeddingModel.call(request);
+            (0, chai_1.expect)(response).to.have.property('embeddings');
+            (0, chai_1.expect)(response.embeddings[0]).to.have.property('embedding');
+            (0, chai_1.expect)(response.result).to.equal(response.embeddings[0]);
+        });
+    });
+});
+//# sourceMappingURL=openai-embedding-model.spec.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.spec.js","sourceRoot":"","sources":["../../../src/common/model/openai-embedding-model.spec.ts"],"names":[],"mappings":";;AAAA,+BAA8B;AAC9B,oBAAkB;AAClB,qEAAgE;AAEhE,2DAAyD;AACzD,sDAAgD;AAChD,6DAAwD;AACxD,yEAAoE;AAEpE,MAAM,SAAS,GAAG,IAAA,gCAAe,GAAE,CAAC;AAEpC,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IAClC,IAAI,oBAA0C,CAAC;IAC/C,IAAI,cAAsC,CAAC;IAE3C,MAAM,CAAC,GAAG,EAAE;QACR,SAAS,CAAC,MAAM,CAAC,wBAAS,CAAC,CAAC,EAAE,CAAC,+BAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACjE,oBAAoB,GAAG,SAAS,CAAC,GAAG,CAAC,6CAAoB,CAAC,CAAC;QAC3D,cAAc,GAAG,SAAS,CAAC,GAAG,CAAC,iDAAsB,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QAClB,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,OAAO,GAAqB;gBAC9B,YAAY,EAAE,CAAC,aAAa,EAAE,qBAAqB,CAAC;gBACpD,MAAM,EAAE,CAAC,aAAa,EAAE,qBAAqB,CAAC;gBAC9C,OAAO,EAAE;oBACL,KAAK,EAAE,wBAAwB;iBAClC;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAE1D,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAChD,IAAA,aAAM,EAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;YAC9C,IAAA,aAAM,EAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC/C,IAAA,aAAM,EAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;YAC3D,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;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,OAAO,GAAqB;gBAC9B,YAAY,EAAE,EAAE;gBAChB,MAAM,EAAE,EAAE;gBACV,OAAO,EAAE;oBACL,KAAK,EAAE,wBAAwB;iBAClC;aACJ,CAAC;YAEF,IAAI;gBACA,MAAM,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzC,aAAM,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;aAC5D;YAAC,OAAO,KAAK,EAAE;gBACZ,IAAA,aAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;aACrE;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,OAAO,GAAqB;gBAC9B,YAAY,EAAE,CAAC,WAAW,CAAC;gBAC3B,MAAM,EAAE,CAAC,WAAW,CAAC;gBACrB,OAAO,EAAE;oBACL,KAAK,EAAE,EAAE;iBACZ;aACJ,CAAC;YAEF,IAAI;gBACA,MAAM,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzC,aAAM,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;aACvD;YAAC,OAAO,KAAK,EAAE;gBACZ,IAAA,aAAM,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;aACxD;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACpE,8CAA8C;YAC9C,MAAM,aAAa,GAAG,cAAc,CAAC,KAAK,CAAC;YAC3C,cAAc,CAAC,KAAK,GAAG,wBAAwB,CAAC;YAEhD,MAAM,OAAO,GAAqB;gBAC9B,YAAY,EAAE,CAAC,WAAW,CAAC;gBAC3B,MAAM,EAAE,CAAC,WAAW,CAAC;gBACrB,OAAO,EAAE,EAAE;aACd,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1D,IAAA,aAAM,EAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAEnE,iCAAiC;YACjC,cAAc,CAAC,KAAK,GAAG,aAAa,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,OAAO,GAAqB;gBAC9B,YAAY,EAAE,CAAC,WAAW,CAAC;gBAC3B,MAAM,EAAE,CAAC,WAAW,CAAC;gBACrB,OAAO,EAAE;oBACL,KAAK,EAAE,wBAAwB;oBAC/B,UAAU,EAAE,GAAG;oBACf,cAAc,EAAE,OAAO;oBACvB,IAAI,EAAE,WAAW;iBACA;aACxB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1D,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAChD,IAAA,aAAM,EAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAC7D,IAAA,aAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,35 @@
+import { EmbeddingOptions } from '@celljs/ai-core';
+/**
+ * OpenAI Embedding Options.
+ */
+export declare class OpenAIEmbeddingOptions implements EmbeddingOptions {
+    /**
+     * ID of the model to use.
+     */
+    model?: string;
+    /**
+     * The format to return the embeddings in. Can be either float or base64.
+     */
+    encodingFormat?: string;
+    /**
+     * The number of dimensions the resulting output embeddings should have.
+     * Only supported in text-embedding-3 and later models.
+     */
+    dimensions?: number;
+    /**
+     * A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.
+     */
+    user?: string;
+    copy(): OpenAIEmbeddingOptions;
+    static options(fromOptions: OpenAIEmbeddingOptions): OpenAIEmbeddingOptions;
+    static builder(): OpenAIEmbeddingOptionsBuilder;
+}
+export declare class OpenAIEmbeddingOptionsBuilder {
+    private options;
+    withModel(model: string): OpenAIEmbeddingOptionsBuilder;
+    withEncodingFormat(encodingFormat: string): OpenAIEmbeddingOptionsBuilder;
+    withDimensions(dimensions: number): OpenAIEmbeddingOptionsBuilder;
+    withUser(user: string): OpenAIEmbeddingOptionsBuilder;
+    build(): OpenAIEmbeddingOptions;
+}
+//# sourceMappingURL=openai-embedding-options.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-options.d.ts","sourceRoot":"","sources":["../../../src/common/model/openai-embedding-options.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAInD;;GAEG;AACH,qBACa,sBAAuB,YAAW,gBAAgB;IAC3D;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IAEH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,IAAI,IAAI,sBAAsB;IAI9B,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,sBAAsB,GAAG,sBAAsB;IAK3E,MAAM,CAAC,OAAO,IAAI,6BAA6B;CAGlD;AAED,qBAAa,6BAA6B;IACtC,OAAO,CAAC,OAAO,CAAgC;IAE/C,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,6BAA6B;IAKvD,kBAAkB,CAAC,cAAc,EAAE,MAAM,GAAG,6BAA6B;IAKzE,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,6BAA6B;IAKjE,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,6BAA6B;IAKrD,KAAK,IAAI,sBAAsB;CAGlC"}

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

@@ -0,0 +1,56 @@
+"use strict";
+var OpenAIEmbeddingOptions_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddingOptionsBuilder = exports.OpenAIEmbeddingOptions = void 0;
+const tslib_1 = require("tslib");
+const class_transformer_1 = require("class-transformer");
+const core_1 = require("@celljs/core");
+/**
+ * OpenAI Embedding Options.
+ */
+let OpenAIEmbeddingOptions = OpenAIEmbeddingOptions_1 = class OpenAIEmbeddingOptions {
+    copy() {
+        return OpenAIEmbeddingOptions_1.options(this);
+    }
+    static options(fromOptions) {
+        const options = new OpenAIEmbeddingOptions_1();
+        return Object.assign(options, fromOptions);
+    }
+    static builder() {
+        return new OpenAIEmbeddingOptionsBuilder();
+    }
+};
+tslib_1.__decorate([
+    (0, class_transformer_1.Expose)({ name: 'encoding_format' }),
+    tslib_1.__metadata("design:type", String)
+], OpenAIEmbeddingOptions.prototype, "encodingFormat", void 0);
+OpenAIEmbeddingOptions = OpenAIEmbeddingOptions_1 = tslib_1.__decorate([
+    (0, core_1.Constant)(OpenAIEmbeddingOptions_1, new OpenAIEmbeddingOptions_1())
+], OpenAIEmbeddingOptions);
+exports.OpenAIEmbeddingOptions = OpenAIEmbeddingOptions;
+class OpenAIEmbeddingOptionsBuilder {
+    constructor() {
+        this.options = new OpenAIEmbeddingOptions();
+    }
+    withModel(model) {
+        this.options.model = model;
+        return this;
+    }
+    withEncodingFormat(encodingFormat) {
+        this.options.encodingFormat = encodingFormat;
+        return this;
+    }
+    withDimensions(dimensions) {
+        this.options.dimensions = dimensions;
+        return this;
+    }
+    withUser(user) {
+        this.options.user = user;
+        return this;
+    }
+    build() {
+        return OpenAIEmbeddingOptions.options(this.options);
+    }
+}
+exports.OpenAIEmbeddingOptionsBuilder = OpenAIEmbeddingOptionsBuilder;
+//# sourceMappingURL=openai-embedding-options.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-options.js","sourceRoot":"","sources":["../../../src/common/model/openai-embedding-options.ts"],"names":[],"mappings":";;;;;AACA,yDAA2C;AAC3C,uCAAwC;AAExC;;GAEG;AAEI,IAAM,sBAAsB,8BAA5B,MAAM,sBAAsB;IAuB/B,IAAI;QACA,OAAO,wBAAsB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,WAAmC;QAC9C,MAAM,OAAO,GAAG,IAAI,wBAAsB,EAAE,CAAC;QAC7C,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,OAAO;QACV,OAAO,IAAI,6BAA6B,EAAE,CAAC;IAC/C,CAAC;CACJ,CAAA;AA1BG;IAAC,IAAA,0BAAM,EAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;;8DACZ;AAVf,sBAAsB;IADlC,IAAA,eAAQ,EAAC,wBAAsB,EAAE,IAAI,wBAAsB,EAAE,CAAC;GAClD,sBAAsB,CAmClC;AAnCY,wDAAsB;AAqCnC,MAAa,6BAA6B;IAA1C;QACY,YAAO,GAAG,IAAI,sBAAsB,EAAE,CAAC;IAyBnD,CAAC;IAvBG,SAAS,CAAC,KAAa;QACnB,IAAI,CAAC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;QAC3B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,kBAAkB,CAAC,cAAsB;QACrC,IAAI,CAAC,OAAO,CAAC,cAAc,GAAG,cAAc,CAAC;QAC7C,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,UAAkB;QAC7B,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,UAAU,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,KAAK;QACD,OAAO,sBAAsB,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxD,CAAC;CACJ;AA1BD,sEA0BC"}

package/lib/common/module.d.ts

@@ -0,0 +1,4 @@
+import './index';
+declare const _default: import("inversify/lib/interfaces/interfaces").interfaces.ContainerModule;
+export default _default;
+//# sourceMappingURL=module.d.ts.map

package/lib/common/module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"module.d.ts","sourceRoot":"","sources":["../../src/common/module.ts"],"names":[],"mappings":"AACA,OAAO,SAAS,CAAC;;AAEjB,wBAA0B"}

package/lib/common/module.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@celljs/core");
+require("./index");
+exports.default = (0, core_1.autoBind)();
+//# sourceMappingURL=module.js.map

package/lib/common/module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"module.js","sourceRoot":"","sources":["../../src/common/module.ts"],"names":[],"mappings":";;AAAA,uCAAwC;AACxC,mBAAiB;AAEjB,kBAAe,IAAA,eAAQ,GAAE,CAAC"}

package/lib/common/support/index.d.ts

@@ -0,0 +1,3 @@
+export * from './openai-api-response-headers';
+export * from './openai-response-header-extractor';
+//# sourceMappingURL=index.d.ts.map

package/lib/common/support/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/common/support/index.ts"],"names":[],"mappings":"AAAA,cAAc,+BAA+B,CAAC;AAC9C,cAAc,oCAAoC,CAAC"}

package/lib/common/support/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./openai-api-response-headers"), exports);
+tslib_1.__exportStar(require("./openai-response-header-extractor"), exports);
+//# sourceMappingURL=index.js.map

package/lib/common/support/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/support/index.ts"],"names":[],"mappings":";;;AAAA,wEAA8C;AAC9C,6EAAmD"}

package/lib/common/support/openai-api-response-headers.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Enumeration of {@literal OpenAI} API response headers.
+ */
+export declare enum OpenAIApiResponseHeaders {
+    /**
+     * Total number of requests allowed within timeframe.
+     */
+    REQUESTS_LIMIT_HEADER = "x-ratelimit-limit-requests",
+    /**
+     * Remaining number of requests available in timeframe.
+     */
+    REQUESTS_REMAINING_HEADER = "x-ratelimit-remaining-requests",
+    /**
+     * Duration of time until the number of requests reset.
+     */
+    REQUESTS_RESET_HEADER = "x-ratelimit-reset-requests",
+    /**
+     * Total number of tokens allowed within timeframe.
+     */
+    TOKENS_RESET_HEADER = "x-ratelimit-reset-tokens",
+    /**
+     * Remaining number of tokens available in timeframe.
+     */
+    TOKENS_LIMIT_HEADER = "x-ratelimit-limit-tokens",
+    /**
+     * Duration of time until the number of tokens reset.
+     */
+    TOKENS_REMAINING_HEADER = "x-ratelimit-remaining-tokens"
+}
+/**
+ * Metadata description for OpenAI API response headers.
+ */
+export declare const OpenAIApiResponseHeadersDescription: Record<OpenAIApiResponseHeaders, string>;
+//# sourceMappingURL=openai-api-response-headers.d.ts.map

package/lib/common/support/openai-api-response-headers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-api-response-headers.d.ts","sourceRoot":"","sources":["../../../src/common/support/openai-api-response-headers.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,oBAAY,wBAAwB;IAChC;;OAEG;IACH,qBAAqB,+BAA+B;IAEpD;;OAEG;IACH,yBAAyB,mCAAmC;IAE5D;;OAEG;IACH,qBAAqB,+BAA+B;IAEpD;;OAEG;IACH,mBAAmB,6BAA6B;IAEhD;;OAEG;IACH,mBAAmB,6BAA6B;IAEhD;;OAEG;IACH,uBAAuB,iCAAiC;CAC3D;AAED;;GAEG;AACH,eAAO,MAAM,mCAAmC,EAAE,MAAM,CAAC,wBAAwB,EAAE,MAAM,CAOxF,CAAC"}

package/lib/common/support/openai-api-response-headers.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIApiResponseHeadersDescription = exports.OpenAIApiResponseHeaders = void 0;
+/**
+ * Enumeration of {@literal OpenAI} API response headers.
+ */
+var OpenAIApiResponseHeaders;
+(function (OpenAIApiResponseHeaders) {
+    /**
+     * Total number of requests allowed within timeframe.
+     */
+    OpenAIApiResponseHeaders["REQUESTS_LIMIT_HEADER"] = "x-ratelimit-limit-requests";
+    /**
+     * Remaining number of requests available in timeframe.
+     */
+    OpenAIApiResponseHeaders["REQUESTS_REMAINING_HEADER"] = "x-ratelimit-remaining-requests";
+    /**
+     * Duration of time until the number of requests reset.
+     */
+    OpenAIApiResponseHeaders["REQUESTS_RESET_HEADER"] = "x-ratelimit-reset-requests";
+    /**
+     * Total number of tokens allowed within timeframe.
+     */
+    OpenAIApiResponseHeaders["TOKENS_RESET_HEADER"] = "x-ratelimit-reset-tokens";
+    /**
+     * Remaining number of tokens available in timeframe.
+     */
+    OpenAIApiResponseHeaders["TOKENS_LIMIT_HEADER"] = "x-ratelimit-limit-tokens";
+    /**
+     * Duration of time until the number of tokens reset.
+     */
+    OpenAIApiResponseHeaders["TOKENS_REMAINING_HEADER"] = "x-ratelimit-remaining-tokens";
+})(OpenAIApiResponseHeaders = exports.OpenAIApiResponseHeaders || (exports.OpenAIApiResponseHeaders = {}));
+/**
+ * Metadata description for OpenAI API response headers.
+ */
+exports.OpenAIApiResponseHeadersDescription = {
+    [OpenAIApiResponseHeaders.REQUESTS_LIMIT_HEADER]: 'Total number of requests allowed within timeframe.',
+    [OpenAIApiResponseHeaders.REQUESTS_REMAINING_HEADER]: 'Remaining number of requests available in timeframe.',
+    [OpenAIApiResponseHeaders.REQUESTS_RESET_HEADER]: 'Duration of time until the number of requests reset.',
+    [OpenAIApiResponseHeaders.TOKENS_RESET_HEADER]: 'Total number of tokens allowed within timeframe.',
+    [OpenAIApiResponseHeaders.TOKENS_LIMIT_HEADER]: 'Remaining number of tokens available in timeframe.',
+    [OpenAIApiResponseHeaders.TOKENS_REMAINING_HEADER]: 'Duration of time until the number of tokens reset.'
+};
+//# sourceMappingURL=openai-api-response-headers.js.map

package/lib/common/support/openai-api-response-headers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-api-response-headers.js","sourceRoot":"","sources":["../../../src/common/support/openai-api-response-headers.ts"],"names":[],"mappings":";;;AAAA;;GAEG;AACH,IAAY,wBA8BX;AA9BD,WAAY,wBAAwB;IAChC;;OAEG;IACH,gFAAoD,CAAA;IAEpD;;OAEG;IACH,wFAA4D,CAAA;IAE5D;;OAEG;IACH,gFAAoD,CAAA;IAEpD;;OAEG;IACH,4EAAgD,CAAA;IAEhD;;OAEG;IACH,4EAAgD,CAAA;IAEhD;;OAEG;IACH,oFAAwD,CAAA;AAC5D,CAAC,EA9BW,wBAAwB,GAAxB,gCAAwB,KAAxB,gCAAwB,QA8BnC;AAED;;GAEG;AACU,QAAA,mCAAmC,GAA6C;IACzF,CAAC,wBAAwB,CAAC,qBAAqB,CAAC,EAAE,oDAAoD;IACtG,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,EAAE,sDAAsD;IAC5G,CAAC,wBAAwB,CAAC,qBAAqB,CAAC,EAAE,sDAAsD;IACxG,CAAC,wBAAwB,CAAC,mBAAmB,CAAC,EAAE,kDAAkD;IAClG,CAAC,wBAAwB,CAAC,mBAAmB,CAAC,EAAE,oDAAoD;IACpG,CAAC,wBAAwB,CAAC,uBAAuB,CAAC,EAAE,oDAAoD;CAC3G,CAAC"}

package/lib/common/support/openai-response-header-extractor.d.ts

@@ -0,0 +1,12 @@
+import { ResponseEntity } from '@celljs/http';
+import { RateLimit } from '@celljs/ai-core';
+/**
+ * Utility used to extract known HTTP response headers for the {@literal OpenAI} API.
+ */
+export declare abstract class OpenAIResponseHeaderExtractor {
+    static extractAIResponseHeaders(response: ResponseEntity<unknown>): RateLimit;
+    private static getHeaderAsDuration;
+    private static getHeaderAsLong;
+    private static parseLong;
+}
+//# sourceMappingURL=openai-response-header-extractor.d.ts.map

package/lib/common/support/openai-response-header-extractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-response-header-extractor.d.ts","sourceRoot":"","sources":["../../../src/common/support/openai-response-header-extractor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAG5C;;GAEG;AACH,8BAAsB,6BAA6B;WAEjC,wBAAwB,CAAC,QAAQ,EAAE,cAAc,CAAC,OAAO,CAAC,GAAG,SAAS;IAmBpF,OAAO,CAAC,MAAM,CAAC,mBAAmB;IAWlC,OAAO,CAAC,MAAM,CAAC,eAAe;IAU9B,OAAO,CAAC,MAAM,CAAC,SAAS;CAS3B"}

package/lib/common/support/openai-response-header-extractor.js

@@ -0,0 +1,131 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIResponseHeaderExtractor = void 0;
+const core_1 = require("@celljs/core");
+const openai_api_response_headers_1 = require("./openai-api-response-headers");
+/**
+ * Utility used to extract known HTTP response headers for the {@literal OpenAI} API.
+ */
+class OpenAIResponseHeaderExtractor {
+    static extractAIResponseHeaders(response) {
+        const requestsLimit = this.getHeaderAsLong(response, openai_api_response_headers_1.OpenAIApiResponseHeaders.REQUESTS_LIMIT_HEADER);
+        const requestsRemaining = this.getHeaderAsLong(response, openai_api_response_headers_1.OpenAIApiResponseHeaders.REQUESTS_REMAINING_HEADER);
+        const tokensLimit = this.getHeaderAsLong(response, openai_api_response_headers_1.OpenAIApiResponseHeaders.TOKENS_LIMIT_HEADER);
+        const tokensRemaining = this.getHeaderAsLong(response, openai_api_response_headers_1.OpenAIApiResponseHeaders.TOKENS_REMAINING_HEADER);
+        const requestsReset = this.getHeaderAsDuration(response, openai_api_response_headers_1.OpenAIApiResponseHeaders.REQUESTS_RESET_HEADER);
+        const tokensReset = this.getHeaderAsDuration(response, openai_api_response_headers_1.OpenAIApiResponseHeaders.TOKENS_RESET_HEADER);
+        return {
+            requestsLimit,
+            requestsRemaining,
+            requestsReset,
+            tokensLimit,
+            tokensRemaining,
+            tokensReset
+        };
+    }
+    static getHeaderAsDuration(response, headerName) {
+        const headers = response.headers;
+        if (headerName in headers) {
+            const values = headers[headerName];
+            if (values && values.length > 0) {
+                return DurationFormatter.TIME_UNIT.parse(values[0]);
+            }
+        }
+        return undefined;
+    }
+    static getHeaderAsLong(response, headerName) {
+        const headers = response.headers;
+        if (headerName in headers) {
+            const values = headers[headerName];
+            if (values && values.length > 0) {
+                return this.parseLong(headerName, values[0]);
+            }
+        }
+    }
+    static parseLong(headerName, headerValue) {
+        if (headerValue) {
+            try {
+                return parseInt(headerValue.trim(), 10);
+            }
+            catch (e) {
+                core_1.ContainerUtil.get(core_1.Logger).warn(`Value [${headerValue}] for HTTP header [${headerName}] is not valid: ${e.message}`);
+            }
+        }
+    }
+}
+exports.OpenAIResponseHeaderExtractor = OpenAIResponseHeaderExtractor;
+/**
+ * Duration formatter for parsing time units.
+ */
+class DurationFormatter {
+    constructor(pattern) {
+        this.pattern = pattern;
+    }
+    parse(text) {
+        core_1.Assert.hasText(text, `Text [${text}] to parse as a Duration must not be null or empty`);
+        const matches = text.match(this.pattern);
+        let total = 0;
+        if (matches) {
+            for (const value of matches) {
+                total += Unit.toDuration(Unit.parseUnit(value), value);
+            }
+        }
+        return total;
+    }
+}
+DurationFormatter.TIME_UNIT = new DurationFormatter(/\d+[a-zA-Z]{1,2}/);
+/**
+ * Time unit enum for duration parsing.
+ */
+var Unit;
+(function (Unit) {
+    Unit["NANOSECONDS"] = "ns";
+    Unit["MICROSECONDS"] = "us";
+    Unit["MILLISECONDS"] = "ms";
+    Unit["SECONDS"] = "s";
+    Unit["MINUTES"] = "m";
+    Unit["HOURS"] = "h";
+    Unit["DAYS"] = "d";
+})(Unit || (Unit = {}));
+(function (Unit) {
+    const unitMap = new Map([
+        [Unit.NANOSECONDS, { name: 'nanoseconds', multiplier: 1 }],
+        [Unit.MICROSECONDS, { name: 'microseconds', multiplier: 1000 }],
+        [Unit.MILLISECONDS, { name: 'milliseconds', multiplier: 1000 * 1000 }],
+        [Unit.SECONDS, { name: 'seconds', multiplier: 1000 * 1000 * 1000 }],
+        [Unit.MINUTES, { name: 'minutes', multiplier: 1000 * 1000 * 1000 * 60 }],
+        [Unit.HOURS, { name: 'hours', multiplier: 1000 * 1000 * 1000 * 60 * 60 }],
+        [Unit.DAYS, { name: 'days', multiplier: 1000 * 1000 * 1000 * 60 * 60 * 24 }]
+    ]);
+    function parseUnit(value) {
+        const symbol = parseSymbol(value);
+        const unit = [...unitMap.keys()].find(u => u === symbol);
+        if (!unit) {
+            throw new Error(`Value [${value}] does not contain a valid time unit`);
+        }
+        return unit;
+    }
+    Unit.parseUnit = parseUnit;
+    function parse(value, predicate) {
+        core_1.Assert.hasText(value, `Value [${value}] must not be null or empty`);
+        return Array.from(value).filter(predicate).join('');
+    }
+    function parseSymbol(value) {
+        return parse(value, char => /[a-zA-Z]/.test(char));
+    }
+    function parseTime(value) {
+        return parseInt(parse(value, char => /\d/.test(char)), 10);
+    }
+    function getName(unit) {
+        var _a, _b;
+        return (_b = (_a = unitMap.get(unit)) === null || _a === void 0 ? void 0 : _a.name) !== null && _b !== void 0 ? _b : unit;
+    }
+    Unit.getName = getName;
+    function toDuration(unit, value) {
+        var _a, _b;
+        const multiplier = (_b = (_a = unitMap.get(unit)) === null || _a === void 0 ? void 0 : _a.multiplier) !== null && _b !== void 0 ? _b : 1;
+        return parseTime(value) * multiplier;
+    }
+    Unit.toDuration = toDuration;
+})(Unit || (Unit = {}));
+//# sourceMappingURL=openai-response-header-extractor.js.map

package/lib/common/support/openai-response-header-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-response-header-extractor.js","sourceRoot":"","sources":["../../../src/common/support/openai-response-header-extractor.ts"],"names":[],"mappings":";;;AAEA,uCAA6D;AAE7D,+EAAyE;AAEzE;;GAEG;AACH,MAAsB,6BAA6B;IAExC,MAAM,CAAC,wBAAwB,CAAC,QAAiC;QACpE,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,sDAAwB,CAAC,qBAAqB,CAAC,CAAC;QACrG,MAAM,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,sDAAwB,CAAC,yBAAyB,CAAC,CAAC;QAC7G,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,sDAAwB,CAAC,mBAAmB,CAAC,CAAC;QACjG,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,sDAAwB,CAAC,uBAAuB,CAAC,CAAC;QAEzG,MAAM,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,sDAAwB,CAAC,qBAAqB,CAAC,CAAC;QACzG,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,sDAAwB,CAAC,mBAAmB,CAAC,CAAC;QAErG,OAAO;YACH,aAAa;YACb,iBAAiB;YACjB,aAAa;YACb,WAAW;YACX,eAAe;YACf,WAAW;SACd,CAAC;IACN,CAAC;IAEO,MAAM,CAAC,mBAAmB,CAAC,QAAiC,EAAE,UAAkB;QACpF,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QACjC,IAAI,UAAU,IAAI,OAAO,EAAE;YACvB,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;YACnC,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,OAAO,iBAAiB,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;aACvD;SACJ;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,MAAM,CAAC,eAAe,CAAC,QAAiC,EAAE,UAAkB;QAChF,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QACjC,IAAI,UAAU,IAAI,OAAO,EAAE;YACvB,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;YACnC,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,OAAO,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;aAChD;SACJ;IACL,CAAC;IAEO,MAAM,CAAC,SAAS,CAAC,UAAkB,EAAE,WAAmB;QAC5D,IAAI,WAAW,EAAE;YACb,IAAI;gBACA,OAAO,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;aAC3C;YAAC,OAAO,CAAC,EAAE;gBACR,oBAAa,CAAC,GAAG,CAAS,aAAM,CAAC,CAAC,IAAI,CAAC,UAAU,WAAW,sBAAsB,UAAU,mBAAmB,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;aAC/H;SACJ;IACL,CAAC;CACJ;AAnDD,sEAmDC;AAED;;GAEG;AACH,MAAM,iBAAiB;IAGnB,YAA6B,OAAe;QAAf,YAAO,GAAP,OAAO,CAAQ;IAAG,CAAC;IAEhD,KAAK,CAAC,IAAY;QACd,aAAM,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,IAAI,oDAAoD,CAAC,CAAC;QAExF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,IAAI,OAAO,EAAE;YACT,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE;gBACzB,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;aAC1D;SACJ;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;;AAjBe,2BAAS,GAAG,IAAI,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;AAoB1E;;GAEG;AACH,IAAK,IAQJ;AARD,WAAK,IAAI;IACL,0BAAkB,CAAA;IAClB,2BAAmB,CAAA;IACnB,2BAAmB,CAAA;IACnB,qBAAa,CAAA;IACb,qBAAa,CAAA;IACb,mBAAW,CAAA;IACX,kBAAU,CAAA;AACd,CAAC,EARI,IAAI,KAAJ,IAAI,QAQR;AAED,WAAU,IAAI;IACV,MAAM,OAAO,GAAG,IAAI,GAAG,CAA+C;QAClE,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;QAC1D,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAC/D,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI,EAAE,CAAC;QACtE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;QACnE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,EAAE,EAAE,CAAC;QACxE,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC;QACzE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC;KAC/E,CAAC,CAAC;IAEH,SAAgB,SAAS,CAAC,KAAa;QACnC,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,EAAE;YACP,MAAM,IAAI,KAAK,CAAC,UAAU,KAAK,sCAAsC,CAAC,CAAC;SAC1E;QACD,OAAO,IAAY,CAAC;IACxB,CAAC;IAPe,cAAS,YAOxB,CAAA;IAED,SAAS,KAAK,CAAC,KAAa,EAAE,SAAoC;QAC9D,aAAM,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,KAAK,6BAA6B,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,SAAS,WAAW,CAAC,KAAa;QAC9B,OAAO,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,SAAS,SAAS,CAAC,KAAa;QAC5B,OAAO,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,SAAgB,OAAO,CAAC,IAAU;;QAC9B,OAAO,MAAA,MAAA,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,0CAAE,IAAI,mCAAI,IAAI,CAAC;IAC3C,CAAC;IAFe,YAAO,UAEtB,CAAA;IAED,SAAgB,UAAU,CAAC,IAAU,EAAE,KAAa;;QAChD,MAAM,UAAU,GAAG,MAAA,MAAA,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,0CAAE,UAAU,mCAAI,CAAC,CAAC;QACtD,OAAO,SAAS,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC;IACzC,CAAC;IAHe,eAAU,aAGzB,CAAA;AACL,CAAC,EAzCS,IAAI,KAAJ,IAAI,QAyCb"}

package/lib/common/test/mock-openai-api.d.ts

@@ -0,0 +1,13 @@
+import { OpenAIAPI } from '../api/api-protocol';
+import { ChatCompletionRequest } from '../api/chat-request';
+import { ChatCompletion, ChatCompletionChunk } from '../api/chat-response';
+import { EmbeddingRequest } from '../api/embeddings-request';
+import { EmbeddingResponse } from '../api/embeddings-response';
+import { Observable } from 'rxjs';
+import { ResponseEntity } from '@celljs/http';
+export declare class MockOpenAIAPI implements OpenAIAPI {
+    chat(chatRequest: ChatCompletionRequest, additionalHttpHeader?: Record<string, string>): Promise<ResponseEntity<ChatCompletion>>;
+    streamingChat(chatRequest: ChatCompletionRequest, additionalHttpHeader?: Record<string, string>): Promise<Observable<ResponseEntity<ChatCompletionChunk>>>;
+    embed<T>(embeddingRequest: EmbeddingRequest<T>): Promise<ResponseEntity<EmbeddingResponse>>;
+}
+//# sourceMappingURL=mock-openai-api.d.ts.map

package/lib/common/test/mock-openai-api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-openai-api.d.ts","sourceRoot":"","sources":["../../../src/common/test/mock-openai-api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAiC,MAAM,sBAAsB,CAAC;AAC1G,OAAO,EAAE,gBAAgB,EAAa,MAAM,2BAA2B,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAM,UAAU,EAAE,MAAM,MAAM,CAAC;AAEtC,OAAO,EAAE,cAAc,EAAsB,MAAM,cAAc,CAAC;AAGlE,qBAAa,aAAc,YAAW,SAAS;IACrC,IAAI,CAAC,WAAW,EAAE,qBAAqB,EAAE,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;IAyBhI,aAAa,CAAC,WAAW,EAAE,qBAAqB,EAAE,oBAAoB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAsB1J,KAAK,CAAC,CAAC,EAAE,gBAAgB,EAAE,gBAAgB,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;CAWpG"}

package/lib/common/test/mock-openai-api.js

@@ -0,0 +1,57 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MockOpenAIAPI = void 0;
+const chat_response_1 = require("../api/chat-response");
+const embeddings_request_1 = require("../api/embeddings-request");
+const embeddings_response_1 = require("../api/embeddings-response");
+const rxjs_1 = require("rxjs");
+const message_1 = require("../api/message");
+const http_1 = require("@celljs/http");
+const usage_1 = require("../api/usage");
+class MockOpenAIAPI {
+    async chat(chatRequest, additionalHttpHeader) {
+        // 使用静态创建方法创建消息
+        const message = message_1.ChatCompletionMessage.create('Hello, how can I help you?', message_1.Role.ASSISTANT);
+        message.toolCalls = []; // 注意这里应该使用 toolCalls 而不是 tool_calls
+        const choice = new chat_response_1.Choice();
+        choice.index = 0;
+        choice.message = message;
+        choice.finishReason = message_1.ChatCompletionFinishReason.STOP; // 使用 finishReason 而不是 finish_reason
+        choice.logprobs = new chat_response_1.LogProbs();
+        const usage = new usage_1.Usage(10, 20, 30);
+        const completion = new chat_response_1.ChatCompletion();
+        completion.id = 'chatcmpl-mock-id';
+        completion.object = 'chat.completion';
+        completion.created = Math.floor(Date.now() / 1000);
+        completion.model = 'gpt-4';
+        completion.choices = [choice];
+        completion.usage = usage;
+        completion.systemFingerprint = 'fp_mock_123'; // 使用 systemFingerprint 而不是 system_fingerprint
+        return http_1.ResponseEntityUtil.ok(completion);
+    }
+    async streamingChat(chatRequest, additionalHttpHeader) {
+        // 使用静态创建方法创建消息
+        const delta = message_1.ChatCompletionMessage.create('Hello, how can I help you?', message_1.Role.ASSISTANT);
+        delta.toolCalls = []; // 注意这里应该使用 toolCalls 而不是 tool_calls
+        const choice = new chat_response_1.ChunkChoice();
+        choice.index = 0;
+        choice.delta = delta;
+        choice.finishReason = message_1.ChatCompletionFinishReason.STOP; // 使用 finishReason 而不是 finish_reason
+        choice.logprobs = new chat_response_1.LogProbs();
+        const chunk = new chat_response_1.ChatCompletionChunk();
+        chunk.id = 'chatcmpl-mock-id';
+        chunk.object = 'chat.completion.chunk';
+        chunk.created = Math.floor(Date.now() / 1000);
+        chunk.model = 'gpt-4';
+        chunk.choices = [choice];
+        chunk.systemFingerprint = 'fp_mock_123'; // 使用 systemFingerprint 而不是 system_fingerprint
+        return (0, rxjs_1.of)(http_1.ResponseEntityUtil.ok(chunk));
+    }
+    async embed(embeddingRequest) {
+        const embedding = new embeddings_request_1.Embedding(0, [0.1, 0.2, 0.3, 0.4, 0.5]);
+        const usage = new usage_1.Usage(8, 0, 8);
+        return http_1.ResponseEntityUtil.ok(new embeddings_response_1.EmbeddingResponse('list', [embedding], 'text-embedding-ada-002', usage));
+    }
+}
+exports.MockOpenAIAPI = MockOpenAIAPI;
+//# sourceMappingURL=mock-openai-api.js.map

package/lib/common/test/mock-openai-api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-openai-api.js","sourceRoot":"","sources":["../../../src/common/test/mock-openai-api.ts"],"names":[],"mappings":";;;AAEA,wDAA0G;AAC1G,kEAAwE;AACxE,oEAA+D;AAC/D,+BAAsC;AACtC,4CAAyF;AACzF,uCAAkE;AAClE,wCAAqC;AAErC,MAAa,aAAa;IACtB,KAAK,CAAC,IAAI,CAAC,WAAkC,EAAE,oBAA6C;QACxF,eAAe;QACf,MAAM,OAAO,GAAG,+BAAqB,CAAC,MAAM,CAAC,4BAA4B,EAAE,cAAI,CAAC,SAAS,CAAC,CAAC;QAC3F,OAAO,CAAC,SAAS,GAAG,EAAE,CAAC,CAAC,oCAAoC;QAE5D,MAAM,MAAM,GAAG,IAAI,sBAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;QACjB,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,MAAM,CAAC,YAAY,GAAG,oCAA0B,CAAC,IAAI,CAAC,CAAC,oCAAoC;QAC3F,MAAM,CAAC,QAAQ,GAAG,IAAI,wBAAQ,EAAE,CAAC;QAEjC,MAAM,KAAK,GAAG,IAAI,aAAK,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAEpC,MAAM,UAAU,GAAG,IAAI,8BAAc,EAAE,CAAC;QACxC,UAAU,CAAC,EAAE,GAAG,kBAAkB,CAAC;QACnC,UAAU,CAAC,MAAM,GAAG,iBAAiB,CAAC;QACtC,UAAU,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QACnD,UAAU,CAAC,KAAK,GAAG,OAAO,CAAC;QAC3B,UAAU,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9B,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;QACzB,UAAU,CAAC,iBAAiB,GAAG,aAAa,CAAC,CAAC,8CAA8C;QAE5F,OAAO,yBAAkB,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,WAAkC,EAAE,oBAA6C;QACjG,eAAe;QACf,MAAM,KAAK,GAAG,+BAAqB,CAAC,MAAM,CAAC,4BAA4B,EAAE,cAAI,CAAC,SAAS,CAAC,CAAC;QACzF,KAAK,CAAC,SAAS,GAAG,EAAE,CAAC,CAAC,oCAAoC;QAE1D,MAAM,MAAM,GAAG,IAAI,2BAAW,EAAE,CAAC;QACjC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,MAAM,CAAC,YAAY,GAAG,oCAA0B,CAAC,IAAI,CAAC,CAAC,oCAAoC;QAC3F,MAAM,CAAC,QAAQ,GAAG,IAAI,wBAAQ,EAAE,CAAC;QAEjC,MAAM,KAAK,GAAG,IAAI,mCAAmB,EAAE,CAAC;QACxC,KAAK,CAAC,EAAE,GAAG,kBAAkB,CAAC;QAC9B,KAAK,CAAC,MAAM,GAAG,uBAAuB,CAAC;QACvC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC9C,KAAK,CAAC,KAAK,GAAG,OAAO,CAAC;QACtB,KAAK,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC;QACzB,KAAK,CAAC,iBAAiB,GAAG,aAAa,CAAC,CAAC,8CAA8C;QAEvF,OAAO,IAAA,SAAE,EAAC,yBAAkB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,KAAK,CAAI,gBAAqC;QAChD,MAAM,SAAS,GAAG,IAAI,8BAAS,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,IAAI,aAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAEjC,OAAO,yBAAkB,CAAC,EAAE,CAAC,IAAI,uCAAiB,CAC9C,MAAM,EACN,CAAC,SAAS,CAAC,EACX,wBAAwB,EACxB,KAAK,CACR,CAAC,CAAC;IACP,CAAC;CACJ;AA3DD,sCA2DC"}

package/lib/common/test/test-container.d.ts

@@ -0,0 +1,3 @@
+import '../index';
+export declare const createContainer: () => import("inversify/lib/interfaces/interfaces").interfaces.Container;
+//# sourceMappingURL=test-container.d.ts.map

package/lib/common/test/test-container.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-container.d.ts","sourceRoot":"","sources":["../../../src/common/test/test-container.ts"],"names":[],"mappings":"AAIA,OAAO,UAAU,CAAC;AAIlB,eAAO,MAAM,eAAe,0EAI3B,CAAC"}

package/lib/common/test/test-container.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createContainer = void 0;
+const test_module_1 = require("@celljs/core/lib/common/test/test-module");
+const module_1 = require("@celljs/http/lib/common/module");
+const module_2 = require("@celljs/ai-core/lib/common/module");
+const core_1 = require("@celljs/core");
+require("../index");
+const openAITestModule = (0, core_1.autoBind)();
+const createContainer = () => {
+    const container = core_1.ContainerFactory.create(test_module_1.coreTestModule, module_1.default, module_2.default, openAITestModule);
+    core_1.ContainerProvider.set(container);
+    return container;
+};
+exports.createContainer = createContainer;
+//# sourceMappingURL=test-container.js.map

package/lib/common/test/test-container.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-container.js","sourceRoot":"","sources":["../../../src/common/test/test-container.ts"],"names":[],"mappings":";;;AAAA,0EAA0E;AAC1E,2DAAwD;AACxD,8DAA6D;AAC7D,uCAA6E;AAC7E,oBAAkB;AAElB,MAAM,gBAAgB,GAAG,IAAA,eAAQ,GAAE,CAAC;AAE7B,MAAM,eAAe,GAAG,GAAG,EAAE;IAChC,MAAM,SAAS,GAAG,uBAAgB,CAAC,MAAM,CAAC,4BAAc,EAAE,gBAAU,EAAE,gBAAY,EAAE,gBAAgB,CAAC,CAAC;IACtG,wBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACjC,OAAO,SAAS,CAAC;AACrB,CAAC,CAAC;AAJW,QAAA,eAAe,mBAI1B"}

package/lib/package.spec.d.ts

@@ -0,0 +1 @@
+//# sourceMappingURL=package.spec.d.ts.map

package/lib/package.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"package.spec.d.ts","sourceRoot":"","sources":["../src/package.spec.ts"],"names":[],"mappings":""}

package/lib/package.spec.js

@@ -0,0 +1,4 @@
+describe('ai-openai package', () => {
+    it('support ai-openai coverage statistics', () => { });
+});
+//# sourceMappingURL=package.spec.js.map

package/lib/package.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"package.spec.js","sourceRoot":"","sources":["../src/package.spec.ts"],"names":[],"mappings":"AAAA,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAE/B,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC"}