Iterate

indexnetwork · Jun 25, 2024 · b953903 · b953903
1 parent abffd09
commit b953903
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/indexer/src/app/modules/agent.module.ts b/indexer/src/app/modules/agent.module.ts
@@ -152,7 +152,7 @@ export class Agent {
               return question.replace(/^\d+\.\s/g, '');
             })
             .filter((question: string) => {
-              return question.length > 5;
+              return question && question.length > 5;
             });
         },
       },

diff --git a/indexer/src/chat/service/chat.service.ts b/indexer/src/chat/service/chat.service.ts
@@ -125,10 +125,14 @@ export class ChatService {
     if (response.ids.length === 0)
       throw new HttpException('No documents have found', HttpStatus.NOT_FOUND);
 
-    const questions = await chain.invoke({
+    const resp = await chain.invoke({
       documents: response.documents,
     });
 
-    return questions;
+    if (resp.questions && resp.questions.length > 4) {
+      resp.questions = resp.questions.slice(0, 4);
+    }
+
+    return resp;
   }
 }