@fr-data-fabric/chatbot-api-nest 0.0.1

package/dist/chatbot-api.module.d.ts

@@ -0,0 +1,27 @@
+import { ConversationRepository } from '@domain/interfaces/ConversationRepository';
+import { SendChatUC } from '@domain/use-cases/chats/SendChatUC';
+import { ICreateConversationUC } from '@domain/use-cases/conversations/CreateConversationUC';
+import { IGetConversationsUC } from '@domain/use-cases/conversations/GetConversationsUC';
+import { IGetConversationUC } from '@domain/use-cases/conversations/GetConversationUC';
+import { DynamicModule, Type } from '@nestjs/common';
+import { GqlTypeReference } from '@nestjs/graphql';
+export type ChatbotApiModuleOptions = {
+    useCases: {
+        getConversation: Type<IGetConversationUC>;
+        getConversations: Type<IGetConversationsUC>;
+        createConversation: Type<ICreateConversationUC>;
+        sendChat: Type<SendChatUC<never>>;
+    };
+    conversationRepository: Type<ConversationRepository>;
+    gql: {
+        GqlConversation: GqlTypeReference;
+        GqlChat: GqlTypeReference;
+        GqlUser: GqlTypeReference;
+        GqlChatSource: GqlTypeReference;
+        GqlChatTextContent: GqlTypeReference;
+        GqlToolUse: GqlTypeReference;
+    };
+};
+export declare class ChatbotApiModule {
+    static register(options: ChatbotApiModuleOptions): DynamicModule;
+}

package/dist/chatbot-api.module.js

@@ -0,0 +1,63 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var ChatbotApiModule_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatbotApiModule = void 0;
+const ConversationRepository_1 = require("@domain/interfaces/ConversationRepository");
+const SendChatUC_1 = require("@domain/use-cases/chats/SendChatUC");
+const CreateConversationUC_1 = require("@domain/use-cases/conversations/CreateConversationUC");
+const DeleteConversationUC_1 = require("@domain/use-cases/conversations/DeleteConversationUC");
+const GetConversationsUC_1 = require("@domain/use-cases/conversations/GetConversationsUC");
+const GetConversationUC_1 = require("@domain/use-cases/conversations/GetConversationUC");
+const UpdateConversationUC_1 = require("@domain/use-cases/conversations/UpdateConversationUC");
+const common_1 = require("@nestjs/common");
+const ChatsController_1 = require("src/inputs/controllers/ChatsController");
+const ChatsResolver_1 = require("src/inputs/resolvers/ChatsResolver");
+const ConversationsResolver_1 = require("src/inputs/resolvers/ConversationsResolver");
+let ChatbotApiModule = ChatbotApiModule_1 = class ChatbotApiModule {
+    static register(options) {
+        return {
+            module: ChatbotApiModule_1,
+            providers: [
+                // Repositories
+                {
+                    provide: ConversationRepository_1.CONVERSATION_REPOSITORY,
+                    useClass: options.conversationRepository,
+                },
+                // Use cases
+                {
+                    provide: GetConversationUC_1.GET_CONVERSATION_UC,
+                    useClass: options.useCases.getConversation,
+                },
+                {
+                    provide: GetConversationsUC_1.GET_CONVERSATIONS_UC,
+                    useClass: options.useCases.getConversations,
+                },
+                {
+                    provide: CreateConversationUC_1.CREATE_CONVERSATION_UC,
+                    useClass: options.useCases.createConversation,
+                },
+                DeleteConversationUC_1.DeleteConversationUC,
+                UpdateConversationUC_1.UpdateConversationUC,
+                {
+                    provide: SendChatUC_1.SEND_CHAT_UC,
+                    useClass: options.useCases.sendChat,
+                },
+                // Resolvers
+                (0, ConversationsResolver_1.GetConversationsResolver)(options.gql.GqlConversation, options.gql.GqlChat, options.gql.GqlUser),
+                (0, ChatsResolver_1.GetChatsResolver)(options.gql.GqlChat, options.gql.GqlChatSource, options.gql.GqlChatTextContent, options.gql.GqlToolUse),
+            ],
+            controllers: [ChatsController_1.ChatsController],
+        };
+    }
+};
+exports.ChatbotApiModule = ChatbotApiModule;
+exports.ChatbotApiModule = ChatbotApiModule = ChatbotApiModule_1 = __decorate([
+    (0, common_1.Module)({})
+], ChatbotApiModule);
+//# sourceMappingURL=chatbot-api.module.js.map

package/dist/chatbot-api.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatbot-api.module.js","sourceRoot":"/","sources":["chatbot-api.module.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,sFAGmD;AACnD,mEAA8E;AAC9E,+FAG8D;AAC9D,+FAA4F;AAC5F,2FAG4D;AAC5D,yFAG2D;AAC3D,+FAA4F;AAC5F,2CAA6D;AAE7D,4EAAyE;AACzE,sEAAsE;AACtE,sFAAsF;AAqB/E,IAAM,gBAAgB,wBAAtB,MAAM,gBAAgB;IAC3B,MAAM,CAAC,QAAQ,CAAC,OAAgC;QAC9C,OAAO;YACL,MAAM,EAAE,kBAAgB;YACxB,SAAS,EAAE;gBACT,eAAe;gBACf;oBACE,OAAO,EAAE,gDAAuB;oBAChC,QAAQ,EAAE,OAAO,CAAC,sBAAsB;iBACzC;gBACD,YAAY;gBACZ;oBACE,OAAO,EAAE,uCAAmB;oBAC5B,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,eAAe;iBAC3C;gBACD;oBACE,OAAO,EAAE,yCAAoB;oBAC7B,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,gBAAgB;iBAC5C;gBACD;oBACE,OAAO,EAAE,6CAAsB;oBAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,kBAAkB;iBAC9C;gBACD,2CAAoB;gBACpB,2CAAoB;gBACpB;oBACE,OAAO,EAAE,yBAAY;oBACrB,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,QAAQ;iBACpC;gBACD,YAAY;gBACZ,IAAA,gDAAwB,EACtB,OAAO,CAAC,GAAG,CAAC,eAAe,EAC3B,OAAO,CAAC,GAAG,CAAC,OAAO,EACnB,OAAO,CAAC,GAAG,CAAC,OAAO,CACpB;gBACD,IAAA,gCAAgB,EACd,OAAO,CAAC,GAAG,CAAC,OAAO,EACnB,OAAO,CAAC,GAAG,CAAC,aAAa,EACzB,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAC9B,OAAO,CAAC,GAAG,CAAC,UAAU,CACvB;aACF;YACD,WAAW,EAAE,CAAC,iCAAe,CAAC;SAC/B,CAAC;IACJ,CAAC;CACF,CAAA;AA7CY,4CAAgB;2BAAhB,gBAAgB;IAD5B,IAAA,eAAM,EAAC,EAAE,CAAC;GACE,gBAAgB,CA6C5B"}

package/dist/domain/entities/ChatEntity.d.ts

@@ -0,0 +1,11 @@
+import { ChatSourceEntity } from '@domain/entities/ChatSourceEntity';
+import { ChatTextContentEntity } from '@domain/entities/ChatTextContentEntity';
+import { ToolUseEntity } from '@domain/entities/ToolUseEntity';
+export interface ChatEntity {
+    id: string;
+    createdAt: Date;
+    updatedAt: Date;
+    _sources: Promise<ChatSourceEntity[]>;
+    _textContents: Promise<ChatTextContentEntity[]>;
+    _toolUses: Promise<ToolUseEntity[]>;
+}

package/dist/domain/entities/ChatEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=ChatEntity.js.map

package/dist/domain/entities/ChatEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatEntity.js","sourceRoot":"/","sources":["domain/entities/ChatEntity.ts"],"names":[],"mappings":""}

package/dist/domain/entities/ChatSourceEntity.d.ts

@@ -0,0 +1,6 @@
+export interface ChatSourceEntity {
+    content: string;
+    link: string;
+    title: string;
+    index: number;
+}

package/dist/domain/entities/ChatSourceEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=ChatSourceEntity.js.map

package/dist/domain/entities/ChatSourceEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatSourceEntity.js","sourceRoot":"/","sources":["domain/entities/ChatSourceEntity.ts"],"names":[],"mappings":""}

package/dist/domain/entities/ChatTextContentEntity.d.ts

@@ -0,0 +1,4 @@
+export interface ChatTextContentEntity {
+    content: string;
+    order: number;
+}

package/dist/domain/entities/ChatTextContentEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=ChatTextContentEntity.js.map

package/dist/domain/entities/ChatTextContentEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatTextContentEntity.js","sourceRoot":"/","sources":["domain/entities/ChatTextContentEntity.ts"],"names":[],"mappings":""}

package/dist/domain/entities/ConversationEntity.d.ts

@@ -0,0 +1,9 @@
+import { ChatEntity } from '@domain/entities/ChatEntity';
+import { UserEntity } from '@domain/entities/UserEntity';
+export interface ConversationEntity {
+    id: string;
+    title: string;
+    shareId: string | null;
+    _chats: Promise<ChatEntity[]>;
+    _user: Promise<UserEntity>;
+}

package/dist/domain/entities/ConversationEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=ConversationEntity.js.map

package/dist/domain/entities/ConversationEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConversationEntity.js","sourceRoot":"/","sources":["domain/entities/ConversationEntity.ts"],"names":[],"mappings":""}

package/dist/domain/entities/DocumentEntity.d.ts

@@ -0,0 +1,3 @@
+export interface DocumentEntity {
+    id: string;
+}

package/dist/domain/entities/DocumentEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=DocumentEntity.js.map

package/dist/domain/entities/DocumentEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DocumentEntity.js","sourceRoot":"/","sources":["domain/entities/DocumentEntity.ts"],"names":[],"mappings":""}

package/dist/domain/entities/ToolUseEntity.d.ts

@@ -0,0 +1,6 @@
+export interface ToolUseEntity {
+    name: string;
+    input: string;
+    output: string | null;
+    order: number;
+}

package/dist/domain/entities/ToolUseEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=ToolUseEntity.js.map

package/dist/domain/entities/ToolUseEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolUseEntity.js","sourceRoot":"/","sources":["domain/entities/ToolUseEntity.ts"],"names":[],"mappings":""}

package/dist/domain/entities/UserEntity.d.ts

@@ -0,0 +1,3 @@
+export interface UserEntity {
+    id: string;
+}

package/dist/domain/entities/UserEntity.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=UserEntity.js.map

package/dist/domain/entities/UserEntity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UserEntity.js","sourceRoot":"/","sources":["domain/entities/UserEntity.ts"],"names":[],"mappings":""}

package/dist/domain/interfaces/ChatRepository.d.ts

@@ -0,0 +1,5 @@
+import { ChatEntity } from '@domain/entities/ChatEntity';
+export declare const CHAT_REPOSITORY = "ChatRepository";
+export interface ChatRepository {
+    save: (chat: ChatEntity) => Promise<ChatEntity>;
+}

package/dist/domain/interfaces/ChatRepository.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CHAT_REPOSITORY = void 0;
+exports.CHAT_REPOSITORY = 'ChatRepository';
+//# sourceMappingURL=ChatRepository.js.map

package/dist/domain/interfaces/ChatRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChatRepository.js","sourceRoot":"/","sources":["domain/interfaces/ChatRepository.ts"],"names":[],"mappings":";;;AAEa,QAAA,eAAe,GAAG,gBAAgB,CAAC"}

package/dist/domain/interfaces/ConversationRepository.d.ts

@@ -0,0 +1,6 @@
+import { ConversationEntity } from '@domain/entities/ConversationEntity';
+export declare const CONVERSATION_REPOSITORY = "ConversationRepository";
+export interface ConversationRepository {
+    save: (conversation: ConversationEntity) => Promise<ConversationEntity>;
+    remove: (conversation: ConversationEntity) => Promise<ConversationEntity>;
+}

package/dist/domain/interfaces/ConversationRepository.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CONVERSATION_REPOSITORY = void 0;
+exports.CONVERSATION_REPOSITORY = 'ConversationRepository';
+//# sourceMappingURL=ConversationRepository.js.map

package/dist/domain/interfaces/ConversationRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConversationRepository.js","sourceRoot":"/","sources":["domain/interfaces/ConversationRepository.ts"],"names":[],"mappings":";;;AAEa,QAAA,uBAAuB,GAAG,wBAAwB,CAAC"}

package/dist/domain/interfaces/LLMService.d.ts

@@ -0,0 +1,13 @@
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+export declare const LLM_SERVICE = "LLMService";
+type ProvidersOptions = {
+    temperature: number;
+    streaming: boolean;
+    model: string;
+    provider: string;
+};
+export type LLMOptions = ProvidersOptions;
+export interface LLMService {
+    getLLM(options: LLMOptions): BaseChatModel;
+}
+export {};

package/dist/domain/interfaces/LLMService.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LLM_SERVICE = void 0;
+exports.LLM_SERVICE = 'LLMService';
+//# sourceMappingURL=LLMService.js.map

package/dist/domain/interfaces/LLMService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LLMService.js","sourceRoot":"/","sources":["domain/interfaces/LLMService.ts"],"names":[],"mappings":";;;AAEa,QAAA,WAAW,GAAG,YAAY,CAAC"}

package/dist/domain/interfaces/PromptService.d.ts

@@ -0,0 +1,8 @@
+import { BaseMessage } from 'langchain';
+export declare const PROMPT_SERVICE = "PromptService";
+export interface PromptService {
+    getPrompt(prompt: {
+        name: string;
+        version: string;
+    }, values: Record<string, unknown>): Promise<BaseMessage[]>;
+}

package/dist/domain/interfaces/PromptService.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PROMPT_SERVICE = void 0;
+exports.PROMPT_SERVICE = 'PromptService';
+//# sourceMappingURL=PromptService.js.map

package/dist/domain/interfaces/PromptService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PromptService.js","sourceRoot":"/","sources":["domain/interfaces/PromptService.ts"],"names":[],"mappings":";;;AAEa,QAAA,cAAc,GAAG,eAAe,CAAC"}

package/dist/domain/use-cases/_common.d.ts

@@ -0,0 +1,17 @@
+import { UserEntity } from '@domain/entities/UserEntity';
+export type UCInput<Input> = {
+    input: Input;
+    currentUser: UserEntity;
+};
+export declare class DeleteByIdInput {
+    id: string;
+}
+export type PaginatedOutput<Item> = {
+    items: Item[];
+    totalCount: number;
+    hasNextPage: boolean;
+};
+export declare class PaginatedInput {
+    limit: number;
+    offset: number;
+}

package/dist/domain/use-cases/_common.js

@@ -0,0 +1,41 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PaginatedInput = exports.DeleteByIdInput = void 0;
+const graphql_1 = require("@nestjs/graphql");
+let DeleteByIdInput = class DeleteByIdInput {
+    id;
+};
+exports.DeleteByIdInput = DeleteByIdInput;
+__decorate([
+    (0, graphql_1.Field)(() => graphql_1.ID),
+    __metadata("design:type", String)
+], DeleteByIdInput.prototype, "id", void 0);
+exports.DeleteByIdInput = DeleteByIdInput = __decorate([
+    (0, graphql_1.InputType)()
+], DeleteByIdInput);
+let PaginatedInput = class PaginatedInput {
+    limit;
+    offset;
+};
+exports.PaginatedInput = PaginatedInput;
+__decorate([
+    (0, graphql_1.Field)(() => graphql_1.Int),
+    __metadata("design:type", Number)
+], PaginatedInput.prototype, "limit", void 0);
+__decorate([
+    (0, graphql_1.Field)(() => graphql_1.Int),
+    __metadata("design:type", Number)
+], PaginatedInput.prototype, "offset", void 0);
+exports.PaginatedInput = PaginatedInput = __decorate([
+    (0, graphql_1.InputType)()
+], PaginatedInput);
+//# sourceMappingURL=_common.js.map

package/dist/domain/use-cases/_common.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_common.js","sourceRoot":"/","sources":["domain/use-cases/_common.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,6CAA4D;AAQrD,IAAM,eAAe,GAArB,MAAM,eAAe;IAE1B,EAAE,CAAU;CACb,CAAA;AAHY,0CAAe;AAE1B;IADC,IAAA,eAAK,EAAC,GAAG,EAAE,CAAC,YAAE,CAAC;;2CACJ;0BAFD,eAAe;IAD3B,IAAA,mBAAS,GAAE;GACC,eAAe,CAG3B;AASM,IAAM,cAAc,GAApB,MAAM,cAAc;IAEzB,KAAK,CAAU;IAGf,MAAM,CAAU;CACjB,CAAA;AANY,wCAAc;AAEzB;IADC,IAAA,eAAK,EAAC,GAAG,EAAE,CAAC,aAAG,CAAC;;6CACF;AAGf;IADC,IAAA,eAAK,EAAC,GAAG,EAAE,CAAC,aAAG,CAAC;;8CACD;yBALL,cAAc;IAD1B,IAAA,mBAAS,GAAE;GACC,cAAc,CAM1B"}

package/dist/domain/use-cases/chats/GetChatCompletionUC.d.ts

@@ -0,0 +1,28 @@
+import { ConversationEntity } from '@domain/entities/ConversationEntity';
+import { DocumentEntity } from '@domain/entities/DocumentEntity';
+import { UserEntity } from '@domain/entities/UserEntity';
+import { AvailableTool, StreamEvents, ToolOptions } from '@domain/use-cases/chats/chat.types';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { BaseMessage, DynamicStructuredTool } from 'langchain';
+import { Observable } from 'rxjs';
+export type GetChatCompletionInput<AvailableTools extends AvailableTool[]> = {
+    message: string;
+    previousMessages: {
+        role: 'user' | 'assistant';
+        content: string;
+    }[];
+    conversation: ConversationEntity;
+    runId: string;
+    tools?: AvailableTools;
+    documents: DocumentEntity[];
+    currentUser: UserEntity;
+    model?: string;
+};
+export declare abstract class GetChatCompletionUC<AvailableTools extends AvailableTool[]> {
+    abstract getTools(toolOptions: ToolOptions, selectedTools: AvailableTools): DynamicStructuredTool[];
+    abstract getLLM(options: {
+        model?: string;
+    }): BaseChatModel;
+    abstract getPrompt(question: string, documents: DocumentEntity[]): Promise<BaseMessage[]>;
+    execute(input: GetChatCompletionInput<AvailableTools>): Observable<StreamEvents<AvailableTools>>;
+}

package/dist/domain/use-cases/chats/GetChatCompletionUC.js

@@ -0,0 +1,83 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GetChatCompletionUC = void 0;
+const langchain_1 = require("langchain");
+const rxjs_1 = require("rxjs");
+class GetChatCompletionUC {
+    execute(input) {
+        return new rxjs_1.Observable((observer) => {
+            const model = this.getLLM({
+                model: input.model,
+            });
+            let sourceIndex = 0;
+            const toolOptions = {
+                getSourceIndex() {
+                    return ++sourceIndex;
+                },
+                addSources(sources) {
+                    observer.next({ event: 'sources_add', sources });
+                },
+                currentUser: input.currentUser,
+            };
+            const agent = (0, langchain_1.createAgent)({
+                model,
+                tools: input.tools ? this.getTools(toolOptions, input.tools) : [],
+            });
+            void (async () => {
+                const prompt = await this.getPrompt(input.message, input.documents);
+                const eventStream = agent.streamEvents({ messages: prompt }, { streamMode: 'messages', configurable: { runId: input.runId } });
+                let lastToolId = undefined;
+                for await (const event of eventStream) {
+                    if (event.event === 'on_tool_end') {
+                        const output = event.data.output;
+                        observer.next({
+                            event: 'tool_end',
+                            name: event.name,
+                            id: output.tool_call_id,
+                            input: event.data.input.input,
+                            output: output.content,
+                        });
+                    }
+                    if (event.event === 'on_tool_start' && lastToolId) {
+                        observer.next({
+                            event: 'tool_start',
+                            name: event.name,
+                            id: lastToolId,
+                            input: event.data.input.input,
+                        });
+                    }
+                    if (event.event === 'on_chat_model_stream') {
+                        const chunk = event.data.chunk;
+                        if (chunk.tool_call_chunks?.length) {
+                            const toolCallChunk = chunk.tool_call_chunks[0];
+                            if (toolCallChunk.name && toolCallChunk.id) {
+                                lastToolId = toolCallChunk.id;
+                                observer.next({
+                                    event: 'tool_stream_start',
+                                    name: toolCallChunk.name,
+                                    id: toolCallChunk.id,
+                                });
+                            }
+                            else if (lastToolId && toolCallChunk.args) {
+                                observer.next({
+                                    event: 'tool_stream',
+                                    id: lastToolId,
+                                    input: toolCallChunk.args,
+                                });
+                            }
+                        }
+                        else if (chunk.content) {
+                            observer.next({
+                                event: 'chat_stream',
+                                content: chunk.content,
+                            });
+                        }
+                    }
+                }
+                observer.complete();
+            })();
+        });
+    }
+}
+exports.GetChatCompletionUC = GetChatCompletionUC;
+//# sourceMappingURL=GetChatCompletionUC.js.map

package/dist/domain/use-cases/chats/GetChatCompletionUC.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GetChatCompletionUC.js","sourceRoot":"/","sources":["domain/use-cases/chats/GetChatCompletionUC.ts"],"names":[],"mappings":";;;AAUA,yCAMmB;AACnB,+BAAkC;AAalC,MAAsB,mBAAmB;IAcvC,OAAO,CAAC,KAA6C;QACnD,OAAO,IAAI,iBAAU,CAA+B,CAAC,QAAQ,EAAE,EAAE;YAC/D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;gBACxB,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB,CAAC,CAAC;YAEH,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,MAAM,WAAW,GAAgB;gBAC/B,cAAc;oBACZ,OAAO,EAAE,WAAW,CAAC;gBACvB,CAAC;gBACD,UAAU,CAAC,OAAO;oBAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnD,CAAC;gBACD,WAAW,EAAE,KAAK,CAAC,WAAW;aAC/B,CAAC;YAEF,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC;gBACxB,KAAK;gBACL,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;aAClE,CAAC,CAAC;YAEH,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;gBAEpE,MAAM,WAAW,GAAG,KAAK,CAAC,YAAY,CACpC,EAAE,QAAQ,EAAE,MAAM,EAAE,EACpB,EAAE,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,EAAE,CACjE,CAAC;gBAEF,IAAI,UAAU,GAAuB,SAAS,CAAC;gBAC/C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;oBACtC,IAAI,KAAK,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;wBAClC,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,MAAqB,CAAC;wBAChD,QAAQ,CAAC,IAAI,CAAC;4BACZ,KAAK,EAAE,UAAU;4BACjB,IAAI,EAAE,KAAK,CAAC,IAAiC;4BAC7C,EAAE,EAAE,MAAM,CAAC,YAAY;4BACvB,KAAK,EAAG,KAAK,CAAC,IAAI,CAAC,KAA2B,CAAC,KAAK;4BACpD,MAAM,EAAE,MAAM,CAAC,OAAO;yBACvB,CAAC,CAAC;oBACL,CAAC;oBACD,IAAI,KAAK,CAAC,KAAK,KAAK,eAAe,IAAI,UAAU,EAAE,CAAC;wBAClD,QAAQ,CAAC,IAAI,CAAC;4BACZ,KAAK,EAAE,YAAY;4BACnB,IAAI,EAAE,KAAK,CAAC,IAAiC;4BAC7C,EAAE,EAAE,UAAU;4BACd,KAAK,EAAG,KAAK,CAAC,IAAI,CAAC,KAA2B,CAAC,KAAK;yBACrD,CAAC,CAAC;oBACL,CAAC;oBACD,IAAI,KAAK,CAAC,KAAK,KAAK,sBAAsB,EAAE,CAAC;wBAC3C,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAuB,CAAC;wBAEjD,IAAI,KAAK,CAAC,gBAAgB,EAAE,MAAM,EAAE,CAAC;4BACnC,MAAM,aAAa,GAAG,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;4BAChD,IAAI,aAAa,CAAC,IAAI,IAAI,aAAa,CAAC,EAAE,EAAE,CAAC;gCAC3C,UAAU,GAAG,aAAa,CAAC,EAAE,CAAC;gCAC9B,QAAQ,CAAC,IAAI,CAAC;oCACZ,KAAK,EAAE,mBAAmB;oCAC1B,IAAI,EAAE,aAAa,CAAC,IAAiC;oCACrD,EAAE,EAAE,aAAa,CAAC,EAAE;iCACrB,CAAC,CAAC;4BACL,CAAC;iCAAM,IAAI,UAAU,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC;gCAC5C,QAAQ,CAAC,IAAI,CAAC;oCACZ,KAAK,EAAE,aAAa;oCACpB,EAAE,EAAE,UAAU;oCACd,KAAK,EAAE,aAAa,CAAC,IAAI;iCAC1B,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;6BAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;4BACzB,QAAQ,CAAC,IAAI,CAAC;gCACZ,KAAK,EAAE,aAAa;gCACpB,OAAO,EAAE,KAAK,CAAC,OAAiB;6BACjC,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtB,CAAC,CAAC,EAAE,CAAC;QACP,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AA/FD,kDA+FC"}

package/dist/domain/use-cases/chats/SendChatUC.d.ts

@@ -0,0 +1,35 @@
+import { ChatEntity } from '@domain/entities/ChatEntity';
+import { ChatSourceEntity } from '@domain/entities/ChatSourceEntity';
+import { ChatTextContentEntity } from '@domain/entities/ChatTextContentEntity';
+import { ConversationEntity } from '@domain/entities/ConversationEntity';
+import { DocumentEntity } from '@domain/entities/DocumentEntity';
+import { ToolUseEntity } from '@domain/entities/ToolUseEntity';
+import { UserEntity } from '@domain/entities/UserEntity';
+import type { ChatRepository } from '@domain/interfaces/ChatRepository';
+import { UCInput } from '@domain/use-cases/_common';
+import { AvailableTool, ChatStreamEvents } from '@domain/use-cases/chats/chat.types';
+import { GetChatCompletionUC } from '@domain/use-cases/chats/GetChatCompletionUC';
+import type { IGetConversationUC } from '@domain/use-cases/conversations/GetConversationUC';
+import { EventEmitter2 } from '@nestjs/event-emitter';
+import { Observable } from 'rxjs';
+export declare class SendMessageInput<AvailableTools extends AvailableTool[]> {
+    conversationId: string;
+    content: string;
+    selectedTools?: AvailableTools;
+    documentIds?: string[];
+    model?: string;
+}
+export declare const SEND_CHAT_UC = "SendChatUC";
+export declare abstract class SendChatUC<AvailableTools extends AvailableTool[]> {
+    private readonly getConversationUC;
+    private readonly chatRepository;
+    private readonly eventEmitter;
+    private readonly getChatCompletionUC;
+    constructor(getConversationUC: IGetConversationUC, chatRepository: ChatRepository, eventEmitter: EventEmitter2, getChatCompletionUC: GetChatCompletionUC<AvailableTools>);
+    abstract getDocuments(documentIds: string[], currentUser: UserEntity): Promise<DocumentEntity[]>;
+    abstract createChatEntity(role: 'user' | 'assistant', content: string, conversation: ConversationEntity, documents?: DocumentEntity[]): ChatEntity;
+    execute({ input, currentUser, }: UCInput<SendMessageInput<AvailableTools>>): Promise<Observable<ChatStreamEvents<AvailableTools>>>;
+    abstract saveChatSources(sources: ChatSourceEntity[]): Promise<void>;
+    abstract saveChatTextContent(chatCompletion: ChatTextContentEntity[]): Promise<void>;
+    abstract saveToolUse(tools: ToolUseEntity[]): Promise<void>;
+}