@celljs/ai-core 3.0.0 → 3.1.0

package/lib/common/chat/index.js

@@ -1,21 +1,8 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./message"), exports);
-__exportStar(require("./metadata"), exports);
-__exportStar(require("./prompt"), exports);
-__exportStar(require("./model"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./message"), exports);
+tslib_1.__exportStar(require("./metadata"), exports);
+tslib_1.__exportStar(require("./prompt"), exports);
+tslib_1.__exportStar(require("./model"), exports);
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/chat/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,4CAA0B;AAC1B,6CAA2B;AAC3B,2CAAyB;AACzB,0CAAwB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/chat/index.ts"],"names":[],"mappings":";;;AAAA,oDAA0B;AAC1B,qDAA2B;AAC3B,mDAAyB;AACzB,kDAAwB"}

package/lib/common/chat/message/index.js

@@ -1,23 +1,10 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./message-protocol"), exports);
-__exportStar(require("./abstract-message"), exports);
-__exportStar(require("./user-message"), exports);
-__exportStar(require("./system-message"), exports);
-__exportStar(require("./assistant-message"), exports);
-__exportStar(require("./function-message"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./message-protocol"), exports);
+tslib_1.__exportStar(require("./abstract-message"), exports);
+tslib_1.__exportStar(require("./user-message"), exports);
+tslib_1.__exportStar(require("./system-message"), exports);
+tslib_1.__exportStar(require("./assistant-message"), exports);
+tslib_1.__exportStar(require("./function-message"), exports);
 //# sourceMappingURL=index.js.map

package/lib/common/chat/message/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/message/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,qDAAmC;AACnC,iDAA+B;AAC/B,mDAAiC;AACjC,sDAAoC;AACpC,qDAAmC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/message/index.ts"],"names":[],"mappings":";;;AAAA,6DAAmC;AACnC,6DAAmC;AACnC,yDAA+B;AAC/B,2DAAiC;AACjC,8DAAoC;AACpC,6DAAmC"}

package/lib/common/chat/metadata/index.js

@@ -1,18 +1,5 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./metadata-protocol"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./metadata-protocol"), exports);
 //# sourceMappingURL=index.js.map

package/lib/common/chat/metadata/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/metadata/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAoC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/metadata/index.ts"],"names":[],"mappings":";;;AAAA,8DAAoC"}

package/lib/common/chat/model/chat-protocol.d.ts

@@ -1,6 +1,8 @@
 import { Model, ModelResponse, ModelResult, StreamingModel } from '../../model/model-protocol';
 import { AssistantMessage } from '../message';
 import { Prompt } from '../prompt/prompt-protocol';
+export declare const StreamingChatModel: unique symbol;
+export declare const ChatModel: unique symbol;
 /**
  * Represents a response returned by the AI.
  */

package/lib/common/chat/model/chat-protocol.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-protocol.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/model/chat-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC/F,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAC;AAEnD;;GAEG;AACH,MAAM,WAAW,UAAW,SAAQ,WAAW,CAAC,gBAAgB,CAAC;CAEhE;AAED,MAAM,WAAW,YAAa,SAAQ,aAAa,CAAC,UAAU,CAAC;CAE9D;AAED,MAAM,WAAW,kBAAmB,SAAQ,cAAc,CAAC,MAAM,EAAE,YAAY,CAAC;CAE/E;AAED,MAAM,WAAW,SAAU,SAAQ,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,kBAAkB;CAEjF"}
+{"version":3,"file":"chat-protocol.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/model/chat-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC/F,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAC;AAEnD,eAAO,MAAM,kBAAkB,eAA+B,CAAC;AAC/D,eAAO,MAAM,SAAS,eAAsB,CAAC;AAE7C;;GAEG;AACH,MAAM,WAAW,UAAW,SAAQ,WAAW,CAAC,gBAAgB,CAAC;CAEhE;AAED,MAAM,WAAW,YAAa,SAAQ,aAAa,CAAC,UAAU,CAAC;CAE9D;AAED,MAAM,WAAW,kBAAmB,SAAQ,cAAc,CAAC,MAAM,EAAE,YAAY,CAAC;CAE/E;AAED,MAAM,WAAW,SAAU,SAAQ,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,kBAAkB;CAEjF"}

package/lib/common/chat/model/chat-protocol.js

@@ -1,3 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatModel = exports.StreamingChatModel = void 0;
+exports.StreamingChatModel = Symbol('StreamingChatModel');
+exports.ChatModel = Symbol('ChatModel');
 //# sourceMappingURL=chat-protocol.js.map

package/lib/common/chat/model/chat-protocol.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/model/chat-protocol.ts"],"names":[],"mappings":""}
+{"version":3,"file":"chat-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/model/chat-protocol.ts"],"names":[],"mappings":";;;AAIa,QAAA,kBAAkB,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;AAClD,QAAA,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC"}

package/lib/common/chat/model/index.js

@@ -1,18 +1,5 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./chat-protocol"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./chat-protocol"), exports);
 //# sourceMappingURL=index.js.map

package/lib/common/chat/model/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/model/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/model/index.ts"],"names":[],"mappings":";;;AAAA,0DAAgC"}

package/lib/common/chat/prompt/index.js

@@ -1,19 +1,6 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./prompt-protocol"), exports);
-__exportStar(require("./prompt"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./prompt-protocol"), exports);
+tslib_1.__exportStar(require("./prompt"), exports);
 //# sourceMappingURL=index.js.map

package/lib/common/chat/prompt/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/prompt/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,oDAAkC;AAClC,2CAAyB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/prompt/index.ts"],"names":[],"mappings":";;;AAAA,4DAAkC;AAClC,mDAAyB"}

package/lib/common/embedding/embedding-protocol.d.ts

@@ -0,0 +1,118 @@
+import { MimeType } from '@celljs/core';
+import { Usage } from '../chat/metadata/metadata-protocol';
+import { Model, ModelOptions, ModelRequest, ModelResponse, ModelResult, ResponseMetadata, ResultMetadata } from '../model/model-protocol';
+export declare const EmbeddingModel: unique symbol;
+/**
+ * Represents the metadata for an embedding result.
+ */
+export interface EmbeddingResultMetadata extends ResultMetadata {
+    /**
+     * The modality type of the source data used to generate the embedding.
+     */
+    modalityType: ModalityType;
+    /**
+     * The document ID associated with the embedding.
+     */
+    documentId: string;
+    /**
+     * The MIME type of the source data used to generate the embedding.
+     */
+    mimeType: MimeType;
+    /**
+     * The document data associated with the embedding.
+     */
+    documentData: any;
+}
+/**
+ * Enum representing the modality type of the source data.
+ */
+export declare enum ModalityType {
+    TEXT = "TEXT",
+    IMAGE = "IMAGE",
+    AUDIO = "AUDIO",
+    VIDEO = "VIDEO"
+}
+/**
+ * Utility class for modality-related operations.
+ */
+export declare class ModalityUtils {
+    private static TEXT_MIME_TYPE;
+    private static IMAGE_MIME_TYPE;
+    private static VIDEO_MIME_TYPE;
+    private static AUDIO_MIME_TYPE;
+    /**
+     * Infers the {@link ModalityType} of the source data used to generate the
+     * embedding using the source data {@link MimeType}.
+     * @param mimeType the {@link MimeType} of the source data.
+     * @return Returns the {@link ModalityType} of the source data used to generate
+     * the embedding.
+     */
+    static getModalityType(mimeType: MimeType): ModalityType;
+}
+/**
+ * Metadata for the embedding response.
+ */
+export interface EmbeddingResponseMetadata extends ResponseMetadata {
+    /**
+     * The model that handled the request.
+     */
+    model: string;
+    /**
+     * The AI provider specific metadata on API usage.
+     * @see Usage
+     */
+    usage: Usage;
+}
+/**
+ * Represents a single embedding vector.
+ */
+export interface Embedding extends ModelResult<number[]> {
+    /**
+     * The embedding vector values.
+     */
+    embedding: number[];
+    /**
+     * The embedding index in a list of embeddings.
+     */
+    index: number;
+    /**
+     * The metadata associated with the embedding.
+     */
+    metadata: EmbeddingResultMetadata;
+}
+export interface EmbeddingOptions extends ModelOptions {
+    /**
+     * The name of the embedding model to use.
+     */
+    model?: string;
+    /**
+     * The dimensionality of the embedding vectors.
+     */
+    dimensions?: number;
+}
+export interface EmbeddingRequest extends ModelRequest<string[]> {
+    /**
+     * The list of input strings for the embedding request.
+     */
+    inputs: string[];
+    /**
+     * The options for the embedding request.
+     */
+    options: EmbeddingOptions;
+}
+/**
+ * Embedding response object.
+ */
+export interface EmbeddingResponse extends ModelResponse<Embedding> {
+    /**
+     * Embedding data.
+     */
+    embeddings: Embedding[];
+    /**
+     * Embedding metadata.
+     */
+    metadata: EmbeddingResponseMetadata;
+}
+export interface EmbeddingModel extends Model<EmbeddingRequest, EmbeddingResponse> {
+}
+//# sourceMappingURL=embedding-protocol.d.ts.map

package/lib/common/embedding/embedding-protocol.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding-protocol.d.ts","sourceRoot":"","sources":["../../../src/common/embedding/embedding-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAiB,MAAM,cAAc,CAAC;AACvD,OAAO,EAAE,KAAK,EAAE,MAAM,oCAAoC,CAAC;AAC3D,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE1I,eAAO,MAAM,cAAc,eAA2B,CAAC;AAEvD;;GAEG;AACH,MAAM,WAAW,uBAAwB,SAAQ,cAAc;IAC3D;;OAEG;IACH,YAAY,EAAE,YAAY,CAAC;IAE3B;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,QAAQ,EAAE,QAAQ,CAAC;IAEnB;;OAEG;IACH,YAAY,EAAE,GAAG,CAAC;CACrB;AAED;;GAEG;AACH,oBAAY,YAAY;IACpB,IAAI,SAAS;IACb,KAAK,UAAU;IACf,KAAK,UAAU;IACf,KAAK,UAAU;CAClB;AAED;;GAEG;AACH,qBAAa,aAAa;IACtB,OAAO,CAAC,MAAM,CAAC,cAAc,CAAyC;IACtE,OAAO,CAAC,MAAM,CAAC,eAAe,CAA0C;IACxE,OAAO,CAAC,MAAM,CAAC,eAAe,CAA0C;IACxE,OAAO,CAAC,MAAM,CAAC,eAAe,CAA0C;IAExE;;;;;;OAMG;WACW,eAAe,CAAC,QAAQ,EAAE,QAAQ,GAAG,YAAY;CAiBlE;AAED;;GAEG;AACH,MAAM,WAAW,yBAA0B,SAAQ,gBAAgB;IAC/D;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;;OAGG;IACH,KAAK,EAAE,KAAK,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,SAAU,SAAQ,WAAW,CAAC,MAAM,EAAE,CAAC;IACpD;;OAEG;IACH,SAAS,EAAE,MAAM,EAAE,CAAC;IAEpB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,QAAQ,EAAE,uBAAuB,CAAC;CACrC;AAED,MAAM,WAAW,gBAAiB,SAAQ,YAAY;IAElD;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,gBAAiB,SAAQ,YAAY,CAAC,MAAM,EAAE,CAAC;IAC5D;;OAEG;IACH,MAAM,EAAE,MAAM,EAAE,CAAC;IAEjB;;OAEG;IACH,OAAO,EAAE,gBAAgB,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,aAAa,CAAC,SAAS,CAAC;IAC/D;;OAEG;IACH,UAAU,EAAE,SAAS,EAAE,CAAC;IAExB;;OAEG;IACH,QAAQ,EAAE,yBAAyB,CAAC;CACvC;AAED,MAAM,WAAW,cAAe,SAAQ,KAAK,CAAC,gBAAgB,EAAE,iBAAiB,CAAC;CAAG"}

package/lib/common/embedding/embedding-protocol.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModalityUtils = exports.ModalityType = exports.EmbeddingModel = void 0;
+const core_1 = require("@celljs/core");
+exports.EmbeddingModel = Symbol('EmbeddingModel');
+/**
+ * Enum representing the modality type of the source data.
+ */
+var ModalityType;
+(function (ModalityType) {
+    ModalityType["TEXT"] = "TEXT";
+    ModalityType["IMAGE"] = "IMAGE";
+    ModalityType["AUDIO"] = "AUDIO";
+    ModalityType["VIDEO"] = "VIDEO";
+})(ModalityType = exports.ModalityType || (exports.ModalityType = {}));
+/**
+ * Utility class for modality-related operations.
+ */
+class ModalityUtils {
+    /**
+     * Infers the {@link ModalityType} of the source data used to generate the
+     * embedding using the source data {@link MimeType}.
+     * @param mimeType the {@link MimeType} of the source data.
+     * @return Returns the {@link ModalityType} of the source data used to generate
+     * the embedding.
+     */
+    static getModalityType(mimeType) {
+        if (!mimeType) {
+            return ModalityType.TEXT;
+        }
+        if (mimeType.isCompatibleWith(ModalityUtils.IMAGE_MIME_TYPE)) {
+            return ModalityType.IMAGE;
+        }
+        else if (mimeType.isCompatibleWith(ModalityUtils.AUDIO_MIME_TYPE)) {
+            return ModalityType.AUDIO;
+        }
+        else if (mimeType.isCompatibleWith(ModalityUtils.VIDEO_MIME_TYPE)) {
+            return ModalityType.VIDEO;
+        }
+        else if (mimeType.isCompatibleWith(ModalityUtils.TEXT_MIME_TYPE)) {
+            return ModalityType.TEXT;
+        }
+        throw new Error('Unsupported MimeType: ' + mimeType);
+    }
+}
+exports.ModalityUtils = ModalityUtils;
+ModalityUtils.TEXT_MIME_TYPE = core_1.MimeTypeUtils.parseMimeType('text/*');
+ModalityUtils.IMAGE_MIME_TYPE = core_1.MimeTypeUtils.parseMimeType('image/*');
+ModalityUtils.VIDEO_MIME_TYPE = core_1.MimeTypeUtils.parseMimeType('video/*');
+ModalityUtils.AUDIO_MIME_TYPE = core_1.MimeTypeUtils.parseMimeType('audio/*');
+//# sourceMappingURL=embedding-protocol.js.map

package/lib/common/embedding/embedding-protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding-protocol.js","sourceRoot":"","sources":["../../../src/common/embedding/embedding-protocol.ts"],"names":[],"mappings":";;;AAAA,uCAAuD;AAI1C,QAAA,cAAc,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;AA2BvD;;GAEG;AACH,IAAY,YAKX;AALD,WAAY,YAAY;IACpB,6BAAa,CAAA;IACb,+BAAe,CAAA;IACf,+BAAe,CAAA;IACf,+BAAe,CAAA;AACnB,CAAC,EALW,YAAY,GAAZ,oBAAY,KAAZ,oBAAY,QAKvB;AAED;;GAEG;AACH,MAAa,aAAa;IAMtB;;;;;;OAMG;IACI,MAAM,CAAC,eAAe,CAAC,QAAkB;QAC5C,IAAI,CAAC,QAAQ,EAAE;YACX,OAAO,YAAY,CAAC,IAAI,CAAC;SAC5B;QAED,IAAI,QAAQ,CAAC,gBAAgB,CAAC,aAAa,CAAC,eAAe,CAAC,EAAE;YAC1D,OAAO,YAAY,CAAC,KAAK,CAAC;SAC7B;aAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,aAAa,CAAC,eAAe,CAAC,EAAE;YACjE,OAAO,YAAY,CAAC,KAAK,CAAC;SAC7B;aAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,aAAa,CAAC,eAAe,CAAC,EAAE;YACjE,OAAO,YAAY,CAAC,KAAK,CAAC;SAC7B;aAAM,IAAI,QAAQ,CAAC,gBAAgB,CAAC,aAAa,CAAC,cAAc,CAAC,EAAE;YAChE,OAAO,YAAY,CAAC,IAAI,CAAC;SAC5B;QAED,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,QAAQ,CAAC,CAAC;IACzD,CAAC;;AA7BL,sCA8BC;AA7BkB,4BAAc,GAAG,oBAAa,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;AACvD,6BAAe,GAAG,oBAAa,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;AACzD,6BAAe,GAAG,oBAAa,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;AACzD,6BAAe,GAAG,oBAAa,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC"}

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

@@ -0,0 +1,2 @@
+export * from './embedding-protocol';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/common/embedding/index.ts"],"names":[],"mappings":"AAAA,cAAc,sBAAsB,CAAC"}

package/lib/common/embedding/index.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./embedding-protocol"), exports);
+//# sourceMappingURL=index.js.map

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

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

package/lib/common/index.d.ts

@@ -1,4 +1,5 @@
 export * from './chat';
 export * from './model';
+export * from './embedding';
 export * from './utils';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/common/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/common/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,aAAa,CAAC;AAC5B,cAAc,SAAS,CAAC"}

package/lib/common/index.js

@@ -1,20 +1,8 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./chat"), exports);
-__exportStar(require("./model"), exports);
-__exportStar(require("./utils"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./chat"), exports);
+tslib_1.__exportStar(require("./model"), exports);
+tslib_1.__exportStar(require("./embedding"), exports);
+tslib_1.__exportStar(require("./utils"), exports);
 //# sourceMappingURL=index.js.map

package/lib/common/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/common/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,0CAAwB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/common/index.ts"],"names":[],"mappings":";;;AAAA,iDAAuB;AACvB,kDAAwB;AACxB,sDAA4B;AAC5B,kDAAwB"}

package/lib/common/model/function/index.js

@@ -1,18 +1,5 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./function-protocol"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./function-protocol"), exports);
 //# sourceMappingURL=index.js.map

package/lib/common/model/function/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/model/function/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAoC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/model/function/index.ts"],"names":[],"mappings":";;;AAAA,8DAAoC"}

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

@@ -1,2 +1,3 @@
 export * from './model-protocol';
+export * from './function';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/common/model/index.ts"],"names":[],"mappings":"AACA,cAAc,kBAAkB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/common/model/index.ts"],"names":[],"mappings":"AACA,cAAc,kBAAkB,CAAC;AACjC,cAAc,YAAY,CAAC"}

package/lib/common/model/index.js

@@ -1,18 +1,6 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./model-protocol"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./model-protocol"), exports);
+tslib_1.__exportStar(require("./function"), exports);
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/model/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AACA,mDAAiC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/model/index.ts"],"names":[],"mappings":";;;AACA,2DAAiC;AACjC,qDAA2B"}

package/lib/common/model/model-protocol.d.ts

@@ -9,6 +9,8 @@
  */
 import { Media } from '../chat';
 import { Observable } from 'rxjs';
+export declare const Model: unique symbol;
+export declare const StreamingModel: unique symbol;
 /**
  * Interface representing the customizable options for AI model interactions. This marker
  * interface allows for the specification of various settings and parameters that can

package/lib/common/model/model-protocol.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model-protocol.d.ts","sourceRoot":"","sources":["../../../src/common/model/model-protocol.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAElC;;;;;;GAMG;AACH,MAAM,WAAW,YAAY;CAAG;AAEhC;;;;;;GAMG;AACH,MAAM,WAAW,cAAc;CAAG;AAElC;;;;;;GAMG;AACH,MAAM,WAAW,gBAAgB;CAAG;AAEpC;;;;;;;GAOG;AACH,MAAM,WAAW,WAAW,CAAC,CAAC;IAE1B;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAEnB;;;OAGG;IACH,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC;CAErC;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,YAAY,CAAC,CAAC;IAE3B;;;OAGG;IACH,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC;IAEzB;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAC;CAElC;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,aAAa,CAAC,CAAC,SAAS,WAAW,CAAC,GAAG,CAAC;IAErD;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAEnB;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;IAEtB;;;OAGG;IACH,QAAQ,CAAC,QAAQ,EAAE,gBAAgB,CAAC;CAEvC;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,KAAK,CAAC,IAAI,SAAS,YAAY,CAAC,GAAG,CAAC,EAAE,IAAI,SAAS,aAAa,CAAC,GAAG,CAAC;IAElF;;;;OAIG;IACH,IAAI,CAAC,OAAO,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAEtC;AAED;;GAEG;AACH,MAAM,WAAW,OAAO;IAEpB;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEzB;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC;IAExB;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAE1C;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,cAAc,CAAC,IAAI,SAAS,YAAY,CAAC,GAAG,CAAC,EAAE,SAAS,SAAS,aAAa,CAAC,GAAG,CAAC;IAEhG;;;;OAIG;IACH,MAAM,CAAC,OAAO,EAAE,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;CAEhD"}
+{"version":3,"file":"model-protocol.d.ts","sourceRoot":"","sources":["../../../src/common/model/model-protocol.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAElC,eAAO,MAAM,KAAK,eAAkB,CAAC;AACrC,eAAO,MAAM,cAAc,eAA2B,CAAC;AAEvD;;;;;;GAMG;AACH,MAAM,WAAW,YAAY;CAAG;AAEhC;;;;;;GAMG;AACH,MAAM,WAAW,cAAc;CAAG;AAElC;;;;;;GAMG;AACH,MAAM,WAAW,gBAAgB;CAAG;AAEpC;;;;;;;GAOG;AACH,MAAM,WAAW,WAAW,CAAC,CAAC;IAE1B;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAEnB;;;OAGG;IACH,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC;CAErC;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,YAAY,CAAC,CAAC;IAE3B;;;OAGG;IACH,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC;IAEzB;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAC;CAElC;AAED;;;;;;;;GAQG;AACH,MAAM,WAAW,aAAa,CAAC,CAAC,SAAS,WAAW,CAAC,GAAG,CAAC;IAErD;;;OAGG;IACH,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAEnB;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;IAEtB;;;OAGG;IACH,QAAQ,CAAC,QAAQ,EAAE,gBAAgB,CAAC;CAEvC;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,KAAK,CAAC,IAAI,SAAS,YAAY,CAAC,GAAG,CAAC,EAAE,IAAI,SAAS,aAAa,CAAC,GAAG,CAAC;IAElF;;;;OAIG;IACH,IAAI,CAAC,OAAO,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAEtC;AAED;;GAEG;AACH,MAAM,WAAW,OAAO;IAEpB;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEzB;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC;IAExB;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAE1C;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,cAAc,CAAC,IAAI,SAAS,YAAY,CAAC,GAAG,CAAC,EAAE,SAAS,SAAS,aAAa,CAAC,GAAG,CAAC;IAEhG;;;;OAIG;IACH,MAAM,CAAC,OAAO,EAAE,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;CAEhD"}

package/lib/common/model/model-protocol.js

@@ -9,4 +9,7 @@
  * ensuring a broad applicability across diverse AI scenarios.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.StreamingModel = exports.Model = void 0;
+exports.Model = Symbol('Model');
+exports.StreamingModel = Symbol('StreamingModel');
 //# sourceMappingURL=model-protocol.js.map

package/lib/common/model/model-protocol.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-protocol.js","sourceRoot":"","sources":["../../../src/common/model/model-protocol.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG"}
+{"version":3,"file":"model-protocol.js","sourceRoot":"","sources":["../../../src/common/model/model-protocol.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AAKU,QAAA,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;AACxB,QAAA,cAAc,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC"}

package/lib/common/utils/index.js

@@ -1,18 +1,5 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./prompt-util"), exports);
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./prompt-util"), exports);
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/utils/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA8B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/common/utils/index.ts"],"names":[],"mappings":";;;AAAA,wDAA8B"}

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "@celljs/ai-core",
-  "version": "3.0.0",
+  "version": "3.1.0",
   "description": "Core domain for AI programming",
   "main": "lib/common/index.js",
   "typings": "lib/common/index.d.ts",
   "dependencies": {
-    "@celljs/core": "3.0.0",
-    "@celljs/http": "3.0.0",
+    "@celljs/core": "3.1.0",
+    "@celljs/http": "3.1.0",
     "rxjs": "^6.6.0"
   },
   "publishConfig": {
@@ -37,7 +37,7 @@
     "test": "cell-component test:js"
   },
   "devDependencies": {
-    "@celljs/component": "3.0.0"
+    "@celljs/component": "3.1.0"
   },
-  "gitHead": "c0ef8d269e33a795406bcbb240cac4b845c6c24c"
+  "gitHead": "6bb7168beaf93b3c54ef92ce51775e9603163444"
 }

package/src/common/chat/model/chat-protocol.ts

@@ -2,6 +2,9 @@ import { Model, ModelResponse, ModelResult, StreamingModel } from '../../model/m
 import { AssistantMessage } from '../message';
 import { Prompt } from '../prompt/prompt-protocol';
 
+export const StreamingChatModel = Symbol('StreamingChatModel');
+export const ChatModel = Symbol('ChatModel');
+
 /**
  * Represents a response returned by the AI.
  */

package/src/common/embedding/embedding-protocol.ts

@@ -0,0 +1,153 @@
+import { MimeType, MimeTypeUtils } from '@celljs/core';
+import { Usage } from '../chat/metadata/metadata-protocol';
+import { Model, ModelOptions, ModelRequest, ModelResponse, ModelResult, ResponseMetadata, ResultMetadata } from '../model/model-protocol';
+
+export const EmbeddingModel = Symbol('EmbeddingModel');
+
+/**
+ * Represents the metadata for an embedding result.
+ */
+export interface EmbeddingResultMetadata extends ResultMetadata {
+    /**
+     * The modality type of the source data used to generate the embedding.
+     */
+    modalityType: ModalityType;
+
+    /**
+     * The document ID associated with the embedding.
+     */
+    documentId: string;
+
+    /**
+     * The MIME type of the source data used to generate the embedding.
+     */
+    mimeType: MimeType;
+
+    /**
+     * The document data associated with the embedding.
+     */
+    documentData: any;
+}
+
+/**
+ * Enum representing the modality type of the source data.
+ */
+export enum ModalityType {
+    TEXT = 'TEXT',
+    IMAGE = 'IMAGE',
+    AUDIO = 'AUDIO',
+    VIDEO = 'VIDEO'
+}
+
+/**
+ * Utility class for modality-related operations.
+ */
+export class ModalityUtils {
+    private static TEXT_MIME_TYPE = MimeTypeUtils.parseMimeType('text/*');
+    private static IMAGE_MIME_TYPE = MimeTypeUtils.parseMimeType('image/*');
+    private static VIDEO_MIME_TYPE = MimeTypeUtils.parseMimeType('video/*');
+    private static AUDIO_MIME_TYPE = MimeTypeUtils.parseMimeType('audio/*');
+
+    /**
+     * Infers the {@link ModalityType} of the source data used to generate the
+     * embedding using the source data {@link MimeType}.
+     * @param mimeType the {@link MimeType} of the source data.
+     * @return Returns the {@link ModalityType} of the source data used to generate
+     * the embedding.
+     */
+    public static getModalityType(mimeType: MimeType): ModalityType {
+        if (!mimeType) {
+            return ModalityType.TEXT;
+        }
+
+        if (mimeType.isCompatibleWith(ModalityUtils.IMAGE_MIME_TYPE)) {
+            return ModalityType.IMAGE;
+        } else if (mimeType.isCompatibleWith(ModalityUtils.AUDIO_MIME_TYPE)) {
+            return ModalityType.AUDIO;
+        } else if (mimeType.isCompatibleWith(ModalityUtils.VIDEO_MIME_TYPE)) {
+            return ModalityType.VIDEO;
+        } else if (mimeType.isCompatibleWith(ModalityUtils.TEXT_MIME_TYPE)) {
+            return ModalityType.TEXT;
+        }
+
+        throw new Error('Unsupported MimeType: ' + mimeType);
+    }
+}
+
+/**
+ * Metadata for the embedding response.
+ */
+export interface EmbeddingResponseMetadata extends ResponseMetadata {
+    /**
+     * The model that handled the request.
+     */
+    model: string;
+
+    /**
+     * The AI provider specific metadata on API usage.
+     * @see Usage
+     */
+    usage: Usage;
+}
+
+/**
+ * Represents a single embedding vector.
+ */
+export interface Embedding extends ModelResult<number[]> {
+    /**
+     * The embedding vector values.
+     */
+    embedding: number[];
+
+    /**
+     * The embedding index in a list of embeddings.
+     */
+    index: number;
+
+    /**
+     * The metadata associated with the embedding.
+     */
+    metadata: EmbeddingResultMetadata;
+}
+
+export interface EmbeddingOptions extends ModelOptions {
+
+    /**
+     * The name of the embedding model to use.
+     */
+    model?: string;
+
+    /**
+     * The dimensionality of the embedding vectors.
+     */
+    dimensions?: number;
+}
+
+export interface EmbeddingRequest extends ModelRequest<string[]> {
+    /**
+     * The list of input strings for the embedding request.
+     */
+    inputs: string[];
+
+    /**
+     * The options for the embedding request.
+     */
+    options: EmbeddingOptions;
+}
+
+/**
+ * Embedding response object.
+ */
+export interface EmbeddingResponse extends ModelResponse<Embedding> {
+    /**
+     * Embedding data.
+     */
+    embeddings: Embedding[];
+
+    /**
+     * Embedding metadata.
+     */
+    metadata: EmbeddingResponseMetadata;
+}
+
+export interface EmbeddingModel extends Model<EmbeddingRequest, EmbeddingResponse> {}

package/src/common/embedding/index.ts

@@ -0,0 +1 @@
+export * from './embedding-protocol';

package/src/common/index.ts

@@ -1,3 +1,4 @@
 export * from './chat';
 export * from './model';
+export * from './embedding';
 export * from './utils';

package/src/common/model/index.ts

@@ -1,2 +1,3 @@
 
 export * from './model-protocol';
+export * from './function';

package/src/common/model/model-protocol.ts

@@ -11,6 +11,9 @@
 import { Media } from '../chat';
 import { Observable } from 'rxjs';
 
+export const Model = Symbol('Model');
+export const StreamingModel = Symbol('StreamingModel');
+
 /**
  * Interface representing the customizable options for AI model interactions. This marker
  * interface allows for the specification of various settings and parameters that can