Merge pull request #74 from import-ai/fix/message_index

fix(search): Exclude some type of messages

Author: LucienShui

src/search/search.service.ts

@@ -8,7 +8,10 @@ import {
   SearchParams,
 } from 'meilisearch';
 import { Resource } from 'src/resources/resources.entity';
-import { Message } from 'src/messages/entities/message.entity';
+import {
+  Message,
+  OpenAIMessageRole,
+} from 'src/messages/entities/message.entity';
 import { DocType } from './doc-type.enum';
 import {
   IndexedDocDto,
@@ -114,7 +117,14 @@ export class SearchService implements OnModuleInit {
     conversationId: string,
     message: Message,
   ) {
-    if (!message.message.content) {
+    if (!message.message.content?.trim()) {
+      return;
+    }
+    if (
+      [OpenAIMessageRole.TOOL, OpenAIMessageRole.SYSTEM].includes(
+        message.message.role,
+      )
+    ) {
       return;
     }
     const content = message.message.content;

src/wizard/stream.service.ts

@@ -107,7 +107,7 @@ export class StreamService {
           throw new Error('Message ID is not set in context');
         }
         const message: Message = await this.messagesService.updateOpenAIMessage(
-          context.messageId!,
+          context.messageId,
           chunk.message,
           chunk.attrs,
         );
@@ -128,6 +128,7 @@ export class StreamService {
         context.parentId = message.id;
         context.messageId = undefined;
       } else if (chunk.response_type === 'done') {
+        // Do nothing, this is the end of the stream
       } else if (chunk.response_type === 'error') {
         const err = new Error(chunk.message || 'Unknown error');
         err.name = 'AgentError';