Merge pull request #47 from boldare/feat/multi-assitant

feat: multi assistant support - allowing to pass assistant ID

Author: sebastianmusial

apps/api/src/main.ts

@@ -9,7 +9,7 @@ async function bootstrap() {
   const globalPrefix = 'api';
   const config = new DocumentBuilder()
     .setTitle('@boldare/openai-assistant')
-    .setVersion('0.1.0')
+    .setVersion('1.0.0')
     .build();
   const document = SwaggerModule.createDocument(app, config);
 

libs/openai-assistant/src/lib/assistant/assistant.module.ts

@@ -1,4 +1,4 @@
-import { DynamicModule, Inject, Module, OnModuleInit } from '@nestjs/common';
+import { DynamicModule, Inject, Module, OnModuleInit, Optional } from '@nestjs/common';
 import { HttpModule } from '@nestjs/axios';
 import {
   AssistantService,
@@ -38,10 +38,14 @@ export class AssistantModule implements OnModuleInit {
   constructor(
     private readonly assistantService: AssistantService,
     private readonly configService: ConfigService,
-    @Inject('config') private config: AssistantConfigParams,
+    @Inject('config') @Optional() private config: AssistantConfigParams,
   ) {}
 
   async onModuleInit(): Promise<void> {
+    if (!this.config) {
+      return;
+    }
+
     this.configService.set(this.config);
     await this.assistantService.init();
   }

libs/openai-assistant/src/lib/chat/chat.gateway.ts

@@ -26,6 +26,7 @@ export class ChatGateway implements OnGatewayConnection {
     @ConnectedSocket() socket: Socket,
   ) {
     this.logger.log(`Socket "${ChatEvents.SendMessage}" (${socket.id}):
+    * assistant: ${request.assistantId}
     * thread: ${request.threadId}
     * files: ${request?.file_ids}
     * content: ${request.content}`);
@@ -34,6 +35,7 @@ export class ChatGateway implements OnGatewayConnection {
 
     this.server?.to(socket.id).emit(ChatEvents.MessageReceived, message);
     this.logger.log(`Socket "${ChatEvents.MessageReceived}" (${socket.id}):
+    * assistant: ${request.assistantId}
     * thread: ${message.threadId}
     * content: ${message.content}`);
   }

libs/openai-assistant/src/lib/chat/chat.model.ts

@@ -33,6 +33,9 @@ export class ChatCallDto {
   @ApiProperty()
   content!: string;
 
+  @ApiProperty({ required: false })
+  assistantId?: string;
+
   @ApiProperty({ required: false })
   file_ids?: string[];
 

libs/openai-assistant/src/lib/chat/chat.service.ts

@@ -27,9 +27,8 @@ export class ChatService {
 
     await this.threads.messages.create(threadId, message);
 
-    const run = await this.threads.runs.create(threadId, {
-      assistant_id: process.env['ASSISTANT_ID'] || '',
-    });
+    const assistant_id = payload?.assistantId || process.env['ASSISTANT_ID'] || '';
+    const run = await this.threads.runs.create(threadId, { assistant_id });
 
     await this.runService.resolve(run);