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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fr-data-fabric/chatbot-api-nest",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "main": "dist/index.js",
   "scripts": {
     "build": "tsc -p tsconfig.json"

package/src/domain/use-cases/chats/GetChatCompletionUC.ts

@@ -39,9 +39,19 @@ export abstract class GetChatCompletionUC<
   abstract getLLM(options: { model?: string }): BaseChatModel;
   abstract getPrompt(
     question: string,
+    contextDocuments: { content: string; id: string }[],
     documents: DocumentEntity[],
   ): Promise<BaseMessage[]>;
 
+  // Override to retrieve RAG context based on the user's question.
+  // Called before getPrompt(); returned documents are merged with any
+  // explicitly provided documents and passed together to getPrompt().
+  abstract getContextDocuments(
+    question: string,
+    conversation: ConversationEntity,
+    currentUser: UserEntity,
+  ): Promise<{ content: string; id: string }[]>;
+
   execute(input: GetChatCompletionInput<AvailableTools>) {
     return new Observable<StreamEvents<AvailableTools>>((observer) => {
       const model = this.getLLM({
@@ -65,7 +75,16 @@ export abstract class GetChatCompletionUC<
       });
 
       void (async () => {
-        const prompt = await this.getPrompt(input.message, input.documents);
+        const contextDocuments = await this.getContextDocuments(
+          input.message,
+          input.conversation,
+          input.currentUser,
+        );
+        const prompt = await this.getPrompt(
+          input.message,
+          contextDocuments,
+          input.documents,
+        );
 
         const eventStream = agent.streamEvents(
           { messages: prompt },

package/src/index.ts

@@ -1 +1,37 @@
 export * from './chatbot-api.module';
+
+// Entities
+export * from './domain/entities/ChatEntity';
+export * from './domain/entities/ChatSourceEntity';
+export * from './domain/entities/ChatTextContentEntity';
+export * from './domain/entities/ConversationEntity';
+export * from './domain/entities/DocumentEntity';
+export * from './domain/entities/ToolUseEntity';
+export * from './domain/entities/UserEntity';
+
+// Interfaces & DI tokens
+export * from './domain/interfaces/ChatRepository';
+export * from './domain/interfaces/ConversationRepository';
+export * from './domain/interfaces/LLMService';
+export * from './domain/interfaces/PromptService';
+
+// Use-case common types
+export * from './domain/use-cases/_common';
+
+// Chat use cases & types
+export * from './domain/use-cases/chats/chat.types';
+export * from './domain/use-cases/chats/GetChatCompletionUC';
+export * from './domain/use-cases/chats/SendChatUC';
+
+// Conversation use cases
+export * from './domain/use-cases/conversations/CreateConversationUC';
+export * from './domain/use-cases/conversations/DeleteConversationUC';
+export * from './domain/use-cases/conversations/GenerateConversationTitleUC';
+export * from './domain/use-cases/conversations/GetConversationUC';
+export * from './domain/use-cases/conversations/GetConversationsUC';
+export * from './domain/use-cases/conversations/UpdateConversationUC';
+
+// Input helpers (decorators & GQL utilities)
+export * from './inputs/controllers/_common/CurrentUserRest';
+export * from './inputs/resolvers/_common/CurrentUserGql';
+export * from './inputs/resolvers/_common/GqlPaginatedOutput';