@celljs/ai-core 3.2.0 → 3.4.0

package/README.md

@@ -1 +1,108 @@
 # Cell - AI Core Component
+
+## 概览
+
+AI Core 模块是一个用于与 AI 模型服务交互的库，提供了生成聊天响应和嵌入向量的功能。通过简单易用的 API 接口，支持消息的创建、请求的发送和响应的处理。是所有 AI 模块的基础，提供了 AI 模块通用的 API 接口。
+
+
+## 特性
+
+- 生成聊天响应
+- 生成嵌入向量
+- 支持流式响应
+- 支持多种模型参数配置
+
+## 安装
+
+使用 npm 安装 AI Core 模块：
+
+```bash
+npm install @celljs/ai-core
+```
+
+或者使用 yarn：
+
+```bash
+yarn add @celljs/ai-core
+```
+
+AI Core 模块是所有 AI 模块的基础，提供了 AI 模块通用的 API 接口，包括消息的创建、请求的发送和响应的处理。所以在使用 AI Core 模块之前，需要先安装厂商对应的模型服务适配模块。例如：`@celljs/ai-ollama`。
+
+```bash
+npm install @celljs/ai-ollama
+```
+
+或者使用 yarn：
+
+```bash
+yarn add @celljs/ai-ollama
+```
+
+## 快速开始
+
+以下是一个简单的示例，展示如何使用 AI Ollama 模块生成聊天响应和嵌入向量：
+
+```typescript
+import { AssistantMessage, PromptTemplate } from '@celljs/ai-core';
+import { Component Autowired } from '@celljs/core';
+
+@Component()
+export class OllamaDemo {
+    @Autowired(OllamChatModel)
+    private chatModel: ChatModel;
+
+    @Autowired(EmbeddingModel)
+    private embeddingModel: EmbeddingModel;
+
+    @Autowired(PromptTemplate)
+    private promptTemplate: PromptTemplate;
+
+    /**
+     * Chat with Ollama
+     */
+    async chat() {
+        const prompt = await this.promptTemplate.create(
+            'Hello {name}',
+            { 
+                chatOptions: { model: 'llama3.2' },
+                variables: { name: 'Ollama' }
+            }
+        );
+        const response = await this.chatModel.call(prompt);
+        console.log(response.result.output);
+    }
+
+    /**
+     * Stream chat response
+     */
+    async stream() {
+        const prompt = await this.promptTemplate.create(
+            'Hello {name}',
+            { 
+                chatOptions: { model: 'llama3.2' },
+                variables: { name: 'Ollama' }
+            }
+        );
+        const response$ = await this.chatModel.stream(prompt);
+        response$.subscribe({
+            next: response => console.log(response.result.output),
+            complete: () => console.log('Chat completed!')
+        });
+    }
+
+    /**
+     * Embed text to vector
+     */
+    async embed() {
+        const response = await this.embeddingModel.call({
+            inputs: ['text to embed'],
+            options: { model: 'llama3.2' }
+        });
+        console.log(response.result.embeddings);
+    }
+}
+```
+
+## 许可证
+
+本项目采用 MIT 许可证。

package/lib/common/chat/message/assistant-message.d.ts

@@ -1,9 +1,10 @@
 import { AbstractMessage } from './abstract-message';
-import { Media } from './message-protocol';
+import { Media, ToolCall } from './message-protocol';
 export declare class AssistantMessage extends AbstractMessage {
     readonly content: string;
     readonly media: Media[];
     readonly metadata: Record<string, any>;
-    constructor(content: string, media?: Media[], metadata?: Record<string, any>);
+    readonly toolCalls: ToolCall[];
+    constructor(content?: string, media?: Media[], metadata?: Record<string, any>, toolCalls?: ToolCall[]);
 }
 //# sourceMappingURL=assistant-message.d.ts.map

package/lib/common/chat/message/assistant-message.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"assistant-message.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/assistant-message.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,KAAK,EAAe,MAAM,oBAAoB,CAAC;AAExD,qBAAa,gBAAiB,SAAQ,eAAe;aAG3B,OAAO,EAAE,MAAM;aACf,KAAK,EAAE,KAAK,EAAE;aACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;gBAF7B,OAAO,EAAE,MAAM,EACf,KAAK,GAAE,KAAK,EAAO,EACnB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAG3D"}
+{"version":3,"file":"assistant-message.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/assistant-message.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,KAAK,EAAe,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAElE,qBAAa,gBAAiB,SAAQ,eAAe;aAG3B,OAAO,EAAE,MAAM;aACf,KAAK,EAAE,KAAK,EAAE;aACd,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAC/C,QAAQ,CAAC,SAAS,EAAE,QAAQ,EAAE;gBAHZ,OAAO,GAAE,MAAW,EACpB,KAAK,GAAE,KAAK,EAAO,EACnB,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,EAC3C,SAAS,GAAE,QAAQ,EAAO;CAI1C"}

package/lib/common/chat/message/assistant-message.js

@@ -4,11 +4,12 @@ exports.AssistantMessage = void 0;
 const abstract_message_1 = require("./abstract-message");
 const message_protocol_1 = require("./message-protocol");
 class AssistantMessage extends abstract_message_1.AbstractMessage {
-    constructor(content, media = [], metadata = {}) {
+    constructor(content = '', media = [], metadata = {}, toolCalls = []) {
         super(message_protocol_1.MessageType.ASSISTANT, content, media, metadata);
         this.content = content;
         this.media = media;
         this.metadata = metadata;
+        this.toolCalls = toolCalls;
     }
 }
 exports.AssistantMessage = AssistantMessage;

package/lib/common/chat/message/assistant-message.js.map

@@ -1 +1 @@
-{"version":3,"file":"assistant-message.js","sourceRoot":"","sources":["../../../../src/common/chat/message/assistant-message.ts"],"names":[],"mappings":";;;AAAA,yDAAqD;AACrD,yDAAwD;AAExD,MAAa,gBAAiB,SAAQ,kCAAe;IAEjD,YACsB,OAAe,EACf,QAAiB,EAAE,EACnB,WAAgC,EAAE;QACpD,KAAK,CAAC,8BAAW,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QAHrC,YAAO,GAAP,OAAO,CAAQ;QACf,UAAK,GAAL,KAAK,CAAc;QACnB,aAAQ,GAAR,QAAQ,CAA0B;IAExD,CAAC;CACJ;AARD,4CAQC"}
+{"version":3,"file":"assistant-message.js","sourceRoot":"","sources":["../../../../src/common/chat/message/assistant-message.ts"],"names":[],"mappings":";;;AAAA,yDAAqD;AACrD,yDAAkE;AAElE,MAAa,gBAAiB,SAAQ,kCAAe;IAEjD,YACsB,UAAkB,EAAE,EACpB,QAAiB,EAAE,EACnB,WAAgC,EAAE,EAC3C,YAAwB,EAAE;QAEnC,KAAK,CAAC,8BAAW,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QALrC,YAAO,GAAP,OAAO,CAAa;QACpB,UAAK,GAAL,KAAK,CAAc;QACnB,aAAQ,GAAR,QAAQ,CAA0B;QAC3C,cAAS,GAAT,SAAS,CAAiB;IAGvC,CAAC;CACJ;AAVD,4CAUC"}

package/lib/common/chat/message/index.d.ts

@@ -4,4 +4,5 @@ export * from './user-message';
 export * from './system-message';
 export * from './assistant-message';
 export * from './function-message';
+export * from './tool-response-message';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,cAAc,oBAAoB,CAAC;AACnC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AACnC,cAAc,oBAAoB,CAAC;AACnC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,yBAAyB,CAAC"}

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

@@ -7,4 +7,5 @@ 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);
+tslib_1.__exportStar(require("./tool-response-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,6DAAmC;AACnC,6DAAmC;AACnC,yDAA+B;AAC/B,2DAAiC;AACjC,8DAAoC;AACpC,6DAAmC"}
+{"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;AACnC,kEAAwC"}

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

@@ -22,7 +22,8 @@ export declare enum MessageType {
     USER = "user",
     ASSISTANT = "assistant",
     SYSTEM = "system",
-    FUNCTION = "function"
+    FUNCTION = "function",
+    TOOL = "tool"
 }
 /**
  * The Message interface represents a message that can be sent or received in a chat
@@ -38,4 +39,25 @@ export interface Message extends Content {
 export declare namespace Message {
     function isMessage(obj: any): obj is Message;
 }
+/**
+ * Represents a tool call.
+ */
+export interface ToolCall {
+    /**
+     * The unique identifier for the tool call.
+     */
+    id: string;
+    /**
+     * The type of the tool.
+     */
+    type: string;
+    /**
+     * The name of the tool.
+     */
+    name: string;
+    /**
+     * The arguments passed to the tool.
+     */
+    arguments: string;
+}
 //# sourceMappingURL=message-protocol.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"message-protocol.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/message-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAErD;;;;;;GAMG;AACH,MAAM,WAAW,KAAK;IAElB,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAE9B;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC;CAEtB;AAED;;;GAGG;AACH,oBAAY,WAAW;IACnB,IAAI,SAAS;IACb,SAAS,cAAc;IACvB,MAAM,WAAW;IACjB,QAAQ,aAAa;CACxB;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,OAAQ,SAAQ,OAAO;IACpC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;CACrC;AAED,yBAAiB,OAAO,CAAC;IACrB,SAAgB,SAAS,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,OAAO,CAElD;CACJ"}
+{"version":3,"file":"message-protocol.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/message-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAErD;;;;;;GAMG;AACH,MAAM,WAAW,KAAK;IAElB,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAE9B;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC;CAEtB;AAED;;;GAGG;AACH,oBAAY,WAAW;IACnB,IAAI,SAAS;IACb,SAAS,cAAc;IACvB,MAAM,WAAW;IACjB,QAAQ,aAAa;IACrB,IAAI,SAAS;CAChB;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,OAAQ,SAAQ,OAAO;IACpC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;CACrC;AAED,yBAAiB,OAAO,CAAC;IACrB,SAAgB,SAAS,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,OAAO,CAElD;CACJ;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACrB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IAEX;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;CACrB"}

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

@@ -11,6 +11,7 @@ var MessageType;
     MessageType["ASSISTANT"] = "assistant";
     MessageType["SYSTEM"] = "system";
     MessageType["FUNCTION"] = "function";
+    MessageType["TOOL"] = "tool";
 })(MessageType = exports.MessageType || (exports.MessageType = {}));
 var Message;
 (function (Message) {

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

@@ -1 +1 @@
-{"version":3,"file":"message-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/message/message-protocol.ts"],"names":[],"mappings":";;;AAqBA;;;GAGG;AACH,IAAY,WAKX;AALD,WAAY,WAAW;IACnB,4BAAa,CAAA;IACb,sCAAuB,CAAA;IACvB,gCAAiB,CAAA;IACjB,oCAAqB,CAAA;AACzB,CAAC,EALW,WAAW,GAAX,mBAAW,KAAX,mBAAW,QAKtB;AAcD,IAAiB,OAAO,CAIvB;AAJD,WAAiB,OAAO;IACpB,SAAgB,SAAS,CAAC,GAAQ;QAC9B,OAAO,GAAG,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,CAAC;IAChD,CAAC;IAFe,iBAAS,YAExB,CAAA;AACL,CAAC,EAJgB,OAAO,GAAP,eAAO,KAAP,eAAO,QAIvB"}
+{"version":3,"file":"message-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/message/message-protocol.ts"],"names":[],"mappings":";;;AAqBA;;;GAGG;AACH,IAAY,WAMX;AAND,WAAY,WAAW;IACnB,4BAAa,CAAA;IACb,sCAAuB,CAAA;IACvB,gCAAiB,CAAA;IACjB,oCAAqB,CAAA;IACrB,4BAAa,CAAA;AACjB,CAAC,EANW,WAAW,GAAX,mBAAW,KAAX,mBAAW,QAMtB;AAcD,IAAiB,OAAO,CAIvB;AAJD,WAAiB,OAAO;IACpB,SAAgB,SAAS,CAAC,GAAQ;QAC9B,OAAO,GAAG,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,CAAC;IAChD,CAAC;IAFe,iBAAS,YAExB,CAAA;AACL,CAAC,EAJgB,OAAO,GAAP,eAAO,KAAP,eAAO,QAIvB"}

package/lib/common/chat/message/tool-response-message.d.ts

@@ -0,0 +1,32 @@
+import { AbstractMessage } from './abstract-message';
+/**
+ * Represents a tool response.
+ */
+export interface ToolResponse {
+    /**
+     * The unique identifier for the tool.
+     */
+    id: string;
+    /**
+     * The name of the tool.
+     */
+    name: string;
+    /**
+     * The response data from the tool.
+     */
+    responseData: string;
+}
+/**
+ * The ToolResponseMessage class represents a message with a function content in a chat
+ * application.
+ */
+export declare class ToolResponseMessage extends AbstractMessage {
+    readonly responses: ToolResponse[];
+    /**
+     * Creates an instance of ToolResponseMessage.
+     * @param responses The list of tool responses.
+     * @param metadata Additional metadata.
+     */
+    constructor(responses: ToolResponse[], metadata?: Record<string, any>);
+}
+//# sourceMappingURL=tool-response-message.d.ts.map

package/lib/common/chat/message/tool-response-message.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-response-message.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/message/tool-response-message.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD;;GAEG;AACH,MAAM,WAAW,YAAY;IACzB;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IAEX;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,YAAY,EAAE,MAAM,CAAC;CACxB;AAED;;;GAGG;AACH,qBAAa,mBAAoB,SAAQ,eAAe;IAQhD,QAAQ,CAAC,SAAS,EAAE,YAAY,EAAE;IANtC;;;;OAIG;gBAEU,SAAS,EAAE,YAAY,EAAE,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;CAI7E"}

package/lib/common/chat/message/tool-response-message.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolResponseMessage = void 0;
+const abstract_message_1 = require("./abstract-message");
+const message_protocol_1 = require("./message-protocol");
+/**
+ * The ToolResponseMessage class represents a message with a function content in a chat
+ * application.
+ */
+class ToolResponseMessage extends abstract_message_1.AbstractMessage {
+    /**
+     * Creates an instance of ToolResponseMessage.
+     * @param responses The list of tool responses.
+     * @param metadata Additional metadata.
+     */
+    constructor(responses, metadata = {}) {
+        super(message_protocol_1.MessageType.TOOL, '', [], metadata);
+        this.responses = responses;
+        this.responses = responses;
+    }
+}
+exports.ToolResponseMessage = ToolResponseMessage;
+//# sourceMappingURL=tool-response-message.js.map

package/lib/common/chat/message/tool-response-message.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-response-message.js","sourceRoot":"","sources":["../../../../src/common/chat/message/tool-response-message.ts"],"names":[],"mappings":";;;AAAA,yDAAqD;AACrD,yDAAiD;AAsBjD;;;GAGG;AACH,MAAa,mBAAoB,SAAQ,kCAAe;IAEpD;;;;OAIG;IACH,YACa,SAAyB,EAAE,WAAgC,EAAE;QACtE,KAAK,CAAC,8BAAW,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;QADjC,cAAS,GAAT,SAAS,CAAgB;QAElC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC/B,CAAC;CACJ;AAZD,kDAYC"}

package/lib/common/chat/metadata/chat-generation-metadata.d.ts

@@ -0,0 +1,28 @@
+import { ResultMetadata } from '../../model/model-protocol';
+/**
+ * Abstract Data Type (ADT) encapsulating information on the completion choices in the AI
+ * response.
+ */
+export interface ChatGenerationMetadata extends ResultMetadata {
+    /**
+     * The reason this choice completed for the generation.
+     */
+    readonly contentFilterMetadata?: any;
+    /**
+     * The reason this choice completed for the generation.
+     */
+    readonly finishReason?: string;
+}
+export declare class ChatGenerationMetadata {
+    static EMPTY: ChatGenerationMetadata;
+    /**
+     * Factory method used to construct a new {@link ChatGenerationMetadata} from the
+     * given finish reason and content filter metadata.
+     *
+     * @param finishReason The reason for completion.
+     * @param contentFilterMetadata The content filter metadata.
+     * @returns A new instance of {@link ChatGenerationMetadata}.
+     */
+    static from(finishReason?: string, contentFilterMetadata?: any): ChatGenerationMetadata;
+}
+//# sourceMappingURL=chat-generation-metadata.d.ts.map

package/lib/common/chat/metadata/chat-generation-metadata.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-generation-metadata.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/metadata/chat-generation-metadata.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAE5D;;;GAGG;AACH,MAAM,WAAW,sBAAuB,SAAQ,cAAc;IAE1D;;OAEG;IACH,QAAQ,CAAC,qBAAqB,CAAC,EAAE,GAAG,CAAC;IAErC;;OAEG;IACH,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,qBAAa,sBAAsB;IAE/B,MAAM,CAAC,KAAK,EAAE,sBAAsB,CAAiC;IAErE;;;;;;;OAOG;IACH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,qBAAqB,CAAC,EAAE,GAAG,GAAG,sBAAsB;CAK1F"}

package/lib/common/chat/metadata/chat-generation-metadata.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatGenerationMetadata = void 0;
+class ChatGenerationMetadata {
+    /**
+     * Factory method used to construct a new {@link ChatGenerationMetadata} from the
+     * given finish reason and content filter metadata.
+     *
+     * @param finishReason The reason for completion.
+     * @param contentFilterMetadata The content filter metadata.
+     * @returns A new instance of {@link ChatGenerationMetadata}.
+     */
+    static from(finishReason, contentFilterMetadata) {
+        return {
+            finishReason, contentFilterMetadata
+        };
+    }
+}
+exports.ChatGenerationMetadata = ChatGenerationMetadata;
+ChatGenerationMetadata.EMPTY = ChatGenerationMetadata.from();
+//# sourceMappingURL=chat-generation-metadata.js.map

package/lib/common/chat/metadata/chat-generation-metadata.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-generation-metadata.js","sourceRoot":"","sources":["../../../../src/common/chat/metadata/chat-generation-metadata.ts"],"names":[],"mappings":";;;AAmBA,MAAa,sBAAsB;IAI/B;;;;;;;OAOG;IACH,MAAM,CAAC,IAAI,CAAC,YAAqB,EAAE,qBAA2B;QAC1D,OAAO;YACH,YAAY,EAAE,qBAAqB;SACtC,CAAC;IACN,CAAC;;AAhBL,wDAiBC;AAfU,4BAAK,GAA2B,sBAAsB,CAAC,IAAI,EAAE,CAAC"}

package/lib/common/chat/metadata/index.d.ts

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

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/metadata/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/metadata/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,4BAA4B,CAAC"}

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

@@ -2,4 +2,5 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const tslib_1 = require("tslib");
 tslib_1.__exportStar(require("./metadata-protocol"), exports);
+tslib_1.__exportStar(require("./chat-generation-metadata"), 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,8DAAoC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/metadata/index.ts"],"names":[],"mappings":";;;AAAA,8DAAoC;AACpC,qEAA2C"}

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

@@ -19,6 +19,9 @@ export interface Usage {
      */
     readonly totalTokens: number;
 }
+export declare namespace Usage {
+    function from(promptTokens: number, generationTokens: number): Usage;
+}
 /**
  * Abstract Data Type (ADT) encapsulating metadata from an AI provider's API rate limits
  * granted to the API key in use and the API key's current balance.
@@ -90,19 +93,32 @@ export declare namespace PromptMetadata {
     function findByPromptIndex(promptMetadata: PromptMetadata, promptIndex: number): PromptFilterMetadata | undefined;
 }
 export interface ChatResponseMetadata extends ResponseMetadata {
+    id?: string;
+    model?: string;
     /**
      * AI provider specific metadata on rate limits.
      * @see RateLimit
      */
-    readonly rateLimit: RateLimit;
+    rateLimit: RateLimit;
     /**
      * AI provider specific metadata on API usage.
      * @see Usage
      */
-    readonly usage: Usage;
-    readonly promptMetadata: PromptMetadata;
+    usage: Usage;
+    promptMetadata: PromptMetadata;
+}
+export declare class ChatResponseMetadataBuilder {
+    private readonly chatResponseMetadata;
+    id(id: string): ChatResponseMetadataBuilder;
+    model(model: string): ChatResponseMetadataBuilder;
+    rateLimit(rateLimit: RateLimit): ChatResponseMetadataBuilder;
+    usage(usage: Usage): ChatResponseMetadataBuilder;
+    promptMetadata(promptMetadata: PromptMetadata): ChatResponseMetadataBuilder;
+    keyValue(key: string, value: any): ChatResponseMetadataBuilder;
+    build(): ChatResponseMetadata;
 }
 export declare namespace ChatResponseMetadata {
     const EMPTY: ChatResponseMetadata;
+    function builder(): ChatResponseMetadataBuilder;
 }
 //# sourceMappingURL=metadata-protocol.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"metadata-protocol.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/metadata/metadata-protocol.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAE9D;;;GAGG;AACH,MAAM,WAAW,KAAK;IAElB;;OAEG;IACH,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAE9B;;;OAGG;IACH,QAAQ,CAAC,gBAAgB,EAAE,MAAM,CAAC;IAElC;;;OAGG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;CAEhC;AAED;;;GAGG;AACH,MAAM,WAAW,SAAS;IAEtB;;;OAGG;IACH,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAE/B;;;OAGG;IACH,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC;IAEnC;;;OAGG;IACH,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAE/B;;;OAGG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAE7B;;;OAGG;IACH,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;IAEjC;;;OAGG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;CAEhC;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB,CAAC,CAAC,GAAG,GAAG;IAEzC;;OAEG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAE7B;;;;;;;OAOG;IACH,QAAQ,CAAC,qBAAqB,EAAE,CAAC,CAAC;CAErC;AAED;;;GAGG;AACH,oBAAY,cAAc,GAAG,oBAAoB,EAAE,CAAC;AAEpD,yBAAiB,cAAc,CAAC;IAC5B;;;;;;OAMG;IACH,SAAgB,iBAAiB,CAAC,cAAc,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,GAAG,oBAAoB,GAAG,SAAS,CAOvH;CACJ;AAED,MAAM,WAAW,oBAAqB,SAAQ,gBAAgB;IAC1D;;;OAGG;IACH,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAE9B;;;OAGG;IACH,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IAEtB,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;CAE3C;AAED,yBAAiB,oBAAoB,CAAC;IAC3B,MAAM,KAAK,sBAeO,CAAC;CAC7B"}
+{"version":3,"file":"metadata-protocol.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/metadata/metadata-protocol.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAE9D;;;GAGG;AACH,MAAM,WAAW,KAAK;IAElB;;OAEG;IACH,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAE9B;;;OAGG;IACH,QAAQ,CAAC,gBAAgB,EAAE,MAAM,CAAC;IAElC;;;OAGG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;CAEhC;AAED,yBAAiB,KAAK,CAAC;IACnB,SAAgB,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,GAAG,KAAK,CAE1E;CACJ;AAED;;;GAGG;AACH,MAAM,WAAW,SAAS;IAEtB;;;OAGG;IACH,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAE/B;;;OAGG;IACH,QAAQ,CAAC,iBAAiB,EAAE,MAAM,CAAC;IAEnC;;;OAGG;IACH,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAE/B;;;OAGG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAE7B;;;OAGG;IACH,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;IAEjC;;;OAGG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;CAEhC;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB,CAAC,CAAC,GAAG,GAAG;IAEzC;;OAEG;IACH,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAE7B;;;;;;;OAOG;IACH,QAAQ,CAAC,qBAAqB,EAAE,CAAC,CAAC;CAErC;AAED;;;GAGG;AACH,oBAAY,cAAc,GAAG,oBAAoB,EAAE,CAAC;AAEpD,yBAAiB,cAAc,CAAC;IAC5B;;;;;;OAMG;IACH,SAAgB,iBAAiB,CAAC,cAAc,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,GAAG,oBAAoB,GAAG,SAAS,CAOvH;CACJ;AAED,MAAM,WAAW,oBAAqB,SAAQ,gBAAgB;IAC1D,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,SAAS,EAAE,SAAS,CAAC;IAErB;;;OAGG;IACH,KAAK,EAAE,KAAK,CAAC;IAEb,cAAc,EAAE,cAAc,CAAC;CAElC;AAED,qBAAa,2BAA2B;IAEpC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAkBnC;IAEF,EAAE,CAAC,EAAE,EAAE,MAAM,GAAG,2BAA2B;IAK3C,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,2BAA2B;IAKjD,SAAS,CAAC,SAAS,EAAE,SAAS,GAAG,2BAA2B;IAK5D,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,2BAA2B;IAKhD,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,2BAA2B;IAK3E,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,2BAA2B;IAW9D,KAAK,IAAI,oBAAoB;CAIhC;AAED,yBAAiB,oBAAoB,CAAC;IAC3B,MAAM,KAAK,EAAE,oBAkBnB,CAAC;IAEF,SAAgB,OAAO,IAAI,2BAA2B,CAErD;CACJ"}

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

@@ -1,7 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChatResponseMetadata = exports.PromptMetadata = void 0;
+exports.ChatResponseMetadata = exports.ChatResponseMetadataBuilder = exports.PromptMetadata = exports.Usage = void 0;
 const core_1 = require("@celljs/core");
+var Usage;
+(function (Usage) {
+    function from(promptTokens, generationTokens) {
+        return { promptTokens, generationTokens, totalTokens: promptTokens + generationTokens };
+    }
+    Usage.from = from;
+})(Usage = exports.Usage || (exports.Usage = {}));
 var PromptMetadata;
 (function (PromptMetadata) {
     /**
@@ -19,9 +26,67 @@ var PromptMetadata;
     }
     PromptMetadata.findByPromptIndex = findByPromptIndex;
 })(PromptMetadata = exports.PromptMetadata || (exports.PromptMetadata = {}));
+class ChatResponseMetadataBuilder {
+    constructor() {
+        this.chatResponseMetadata = {
+            id: '',
+            model: '',
+            rateLimit: {
+                requestsLimit: 0,
+                requestsRemaining: 0,
+                requestsReset: 0,
+                tokensLimit: 0,
+                tokensRemaining: 0,
+                tokensReset: 0
+            },
+            usage: {
+                promptTokens: 0,
+                generationTokens: 0,
+                totalTokens: 0
+            },
+            promptMetadata: [],
+            extra: {}
+        };
+    }
+    id(id) {
+        this.chatResponseMetadata.id = id;
+        return this;
+    }
+    model(model) {
+        this.chatResponseMetadata.model = model;
+        return this;
+    }
+    rateLimit(rateLimit) {
+        this.chatResponseMetadata.rateLimit = rateLimit;
+        return this;
+    }
+    usage(usage) {
+        this.chatResponseMetadata.usage = usage;
+        return this;
+    }
+    promptMetadata(promptMetadata) {
+        this.chatResponseMetadata.promptMetadata = promptMetadata;
+        return this;
+    }
+    keyValue(key, value) {
+        if (!key) {
+            throw new core_1.IllegalArgumentError('Key must not be empty');
+        }
+        if (value) {
+            this.chatResponseMetadata.extra[key] = value;
+        }
+        return this;
+    }
+    build() {
+        return this.chatResponseMetadata;
+    }
+}
+exports.ChatResponseMetadataBuilder = ChatResponseMetadataBuilder;
 var ChatResponseMetadata;
 (function (ChatResponseMetadata) {
     ChatResponseMetadata.EMPTY = {
+        id: '',
+        model: '',
         rateLimit: {
             requestsLimit: 0,
             requestsRemaining: 0,
@@ -35,7 +100,12 @@ var ChatResponseMetadata;
             generationTokens: 0,
             totalTokens: 0
         },
-        promptMetadata: []
+        promptMetadata: [],
+        extra: {}
     };
+    function builder() {
+        return new ChatResponseMetadataBuilder();
+    }
+    ChatResponseMetadata.builder = builder;
 })(ChatResponseMetadata = exports.ChatResponseMetadata || (exports.ChatResponseMetadata = {}));
 //# sourceMappingURL=metadata-protocol.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"metadata-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/metadata/metadata-protocol.ts"],"names":[],"mappings":";;;AAAA,uCAAoD;AAqGpD,IAAiB,cAAc,CAgB9B;AAhBD,WAAiB,cAAc;IAC3B;;;;;;OAMG;IACH,SAAgB,iBAAiB,CAAC,cAA8B,EAAE,WAAmB;QAEjF,IAAI,WAAW,GAAG,CAAC,EAAE;YACjB,MAAM,IAAI,2BAAoB,CAAC,iBAAiB,WAAW,mCAAmC,CAAC,CAAC;SACnG;QAED,OAAO,cAAc,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,CAAC,oBAAoB,CAAC,WAAW,KAAK,WAAW,CAAC,CAAC;IACzG,CAAC;IAPe,gCAAiB,oBAOhC,CAAA;AACL,CAAC,EAhBgB,cAAc,GAAd,sBAAc,KAAd,sBAAc,QAgB9B;AAmBD,IAAiB,oBAAoB,CAiBpC;AAjBD,WAAiB,oBAAoB;IACpB,0BAAK,GAAG;QACjB,SAAS,EAAE;YACP,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,CAAC;YACpB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,eAAe,EAAE,CAAC;YAClB,WAAW,EAAE,CAAC;SACjB;QACD,KAAK,EAAE;YACH,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;SACjB;QACD,cAAc,EAAE,EAAoB;KACf,CAAC;AAC9B,CAAC,EAjBgB,oBAAoB,GAApB,4BAAoB,KAApB,4BAAoB,QAiBpC"}
+{"version":3,"file":"metadata-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/metadata/metadata-protocol.ts"],"names":[],"mappings":";;;AAAA,uCAAoD;AA4BpD,IAAiB,KAAK,CAIrB;AAJD,WAAiB,KAAK;IAClB,SAAgB,IAAI,CAAC,YAAoB,EAAE,gBAAwB;QAC/D,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,WAAW,EAAE,YAAY,GAAG,gBAAgB,EAAE,CAAC;IAC5F,CAAC;IAFe,UAAI,OAEnB,CAAA;AACL,CAAC,EAJgB,KAAK,GAAL,aAAK,KAAL,aAAK,QAIrB;AA2ED,IAAiB,cAAc,CAgB9B;AAhBD,WAAiB,cAAc;IAC3B;;;;;;OAMG;IACH,SAAgB,iBAAiB,CAAC,cAA8B,EAAE,WAAmB;QAEjF,IAAI,WAAW,GAAG,CAAC,EAAE;YACjB,MAAM,IAAI,2BAAoB,CAAC,iBAAiB,WAAW,mCAAmC,CAAC,CAAC;SACnG;QAED,OAAO,cAAc,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,CAAC,oBAAoB,CAAC,WAAW,KAAK,WAAW,CAAC,CAAC;IACzG,CAAC;IAPe,gCAAiB,oBAOhC,CAAA;AACL,CAAC,EAhBgB,cAAc,GAAd,sBAAc,KAAd,sBAAc,QAgB9B;AAqBD,MAAa,2BAA2B;IAAxC;QAEqB,yBAAoB,GAAyB;YAC1D,EAAE,EAAE,EAAE;YACN,KAAK,EAAE,EAAE;YACT,SAAS,EAAE;gBACP,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,CAAC;gBACpB,aAAa,EAAE,CAAC;gBAChB,WAAW,EAAE,CAAC;gBACd,eAAe,EAAE,CAAC;gBAClB,WAAW,EAAE,CAAC;aACjB;YACD,KAAK,EAAE;gBACH,YAAY,EAAE,CAAC;gBACf,gBAAgB,EAAE,CAAC;gBACnB,WAAW,EAAE,CAAC;aACjB;YACD,cAAc,EAAE,EAAoB;YACpC,KAAK,EAAE,EAAE;SACZ,CAAC;IA0CN,CAAC;IAxCG,EAAE,CAAC,EAAU;QACT,IAAI,CAAC,oBAAoB,CAAC,EAAE,GAAG,EAAE,CAAC;QAClC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,KAAa;QACf,IAAI,CAAC,oBAAoB,CAAC,KAAK,GAAG,KAAK,CAAC;QACxC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,SAAS,CAAC,SAAoB;QAC1B,IAAI,CAAC,oBAAoB,CAAC,SAAS,GAAG,SAAS,CAAC;QAChD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,KAAY;QACd,IAAI,CAAC,oBAAoB,CAAC,KAAK,GAAG,KAAK,CAAC;QACxC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,IAAI,CAAC,oBAAoB,CAAC,cAAc,GAAG,cAAc,CAAC;QAC1D,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,GAAW,EAAE,KAAU;QAC5B,IAAI,CAAC,GAAG,EAAE;YACN,MAAM,IAAI,2BAAoB,CAAC,uBAAuB,CAAC,CAAC;SAC3D;QACD,IAAI,KAAK,EAAE;YACP,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;SAChD;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK;QACD,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACrC,CAAC;CAEJ;AA9DD,kEA8DC;AAED,IAAiB,oBAAoB,CAwBpC;AAxBD,WAAiB,oBAAoB;IACpB,0BAAK,GAAyB;QACvC,EAAE,EAAE,EAAE;QACN,KAAK,EAAE,EAAE;QACT,SAAS,EAAE;YACP,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,CAAC;YACpB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,eAAe,EAAE,CAAC;YAClB,WAAW,EAAE,CAAC;SACjB;QACD,KAAK,EAAE;YACH,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;SACjB;QACD,cAAc,EAAE,EAAoB;QACpC,KAAK,EAAE,EAAE;KACZ,CAAC;IAEF,SAAgB,OAAO;QACnB,OAAO,IAAI,2BAA2B,EAAE,CAAC;IAC7C,CAAC;IAFe,4BAAO,UAEtB,CAAA;AACL,CAAC,EAxBgB,oBAAoB,GAApB,4BAAoB,KAApB,4BAAoB,QAwBpC"}

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

@@ -1,5 +1,6 @@
 import { Model, ModelResponse, ModelResult, StreamingModel } from '../../model/model-protocol';
 import { AssistantMessage } from '../message';
+import { ChatGenerationMetadata, ChatResponseMetadata } from '../metadata';
 import { Prompt } from '../prompt/prompt-protocol';
 export declare const StreamingChatModel: unique symbol;
 export declare const ChatModel: unique symbol;
@@ -10,6 +11,12 @@ export interface Generation extends ModelResult<AssistantMessage> {
 }
 export interface ChatResponse extends ModelResponse<Generation> {
 }
+export declare namespace Generation {
+    function from(output: AssistantMessage, metadata?: ChatGenerationMetadata): Generation;
+}
+export declare namespace ChatResponse {
+    function from(results: Generation[], metadata?: ChatResponseMetadata): ChatResponse;
+}
 export interface StreamingChatModel extends StreamingModel<Prompt, ChatResponse> {
 }
 export interface ChatModel extends Model<Prompt, ChatResponse>, StreamingChatModel {

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,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"}
+{"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,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAC3E,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,yBAAiB,UAAU,CAAC;IACxB,SAAgB,IAAI,CAAC,MAAM,EAAE,gBAAgB,EAAE,QAAQ,yBAA+B,GAAG,UAAU,CAElG;CACJ;AAED,yBAAiB,YAAY,CAAC;IAC1B,SAAgB,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,QAAQ,uBAA6B,GAAG,YAAY,CAE/F;CACJ;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,6 +1,21 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChatModel = exports.StreamingChatModel = void 0;
+exports.ChatResponse = exports.Generation = exports.ChatModel = exports.StreamingChatModel = void 0;
+const metadata_1 = require("../metadata");
 exports.StreamingChatModel = Symbol('StreamingChatModel');
 exports.ChatModel = Symbol('ChatModel');
+var Generation;
+(function (Generation) {
+    function from(output, metadata = metadata_1.ChatGenerationMetadata.EMPTY) {
+        return { output, metadata };
+    }
+    Generation.from = from;
+})(Generation = exports.Generation || (exports.Generation = {}));
+var ChatResponse;
+(function (ChatResponse) {
+    function from(results, metadata = metadata_1.ChatResponseMetadata.EMPTY) {
+        return { result: results[0], results, metadata };
+    }
+    ChatResponse.from = from;
+})(ChatResponse = exports.ChatResponse || (exports.ChatResponse = {}));
 //# 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":";;;AAIa,QAAA,kBAAkB,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;AAClD,QAAA,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC"}
+{"version":3,"file":"chat-protocol.js","sourceRoot":"","sources":["../../../../src/common/chat/model/chat-protocol.ts"],"names":[],"mappings":";;;AAEA,0CAA2E;AAG9D,QAAA,kBAAkB,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;AAClD,QAAA,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;AAa7C,IAAiB,UAAU,CAI1B;AAJD,WAAiB,UAAU;IACvB,SAAgB,IAAI,CAAC,MAAwB,EAAE,QAAQ,GAAG,iCAAsB,CAAC,KAAK;QAClF,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IAChC,CAAC;IAFe,eAAI,OAEnB,CAAA;AACL,CAAC,EAJgB,UAAU,GAAV,kBAAU,KAAV,kBAAU,QAI1B;AAED,IAAiB,YAAY,CAI5B;AAJD,WAAiB,YAAY;IACzB,SAAgB,IAAI,CAAC,OAAqB,EAAE,QAAQ,GAAG,+BAAoB,CAAC,KAAK;QAC7E,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IACrD,CAAC;IAFe,iBAAI,OAEnB,CAAA;AACL,CAAC,EAJgB,YAAY,GAAZ,oBAAY,KAAZ,oBAAY,QAI5B"}

package/lib/common/chat/prompt/index.d.ts

@@ -1,3 +1,4 @@
 export * from './prompt-protocol';
 export * from './prompt';
+export * from './prompt-template';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/prompt/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,UAAU,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/common/chat/prompt/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,UAAU,CAAC;AACzB,cAAc,mBAAmB,CAAC"}

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

@@ -3,4 +3,5 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const tslib_1 = require("tslib");
 tslib_1.__exportStar(require("./prompt-protocol"), exports);
 tslib_1.__exportStar(require("./prompt"), exports);
+tslib_1.__exportStar(require("./prompt-template"), 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,4DAAkC;AAClC,mDAAyB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/common/chat/prompt/index.ts"],"names":[],"mappings":";;;AAAA,4DAAkC;AAClC,mDAAyB;AACzB,4DAAkC"}