Update agent.module.ts

indexnetwork · Jul 23, 2024 · 2ef4c58 · 2ef4c58
1 parent a71b7ac
commit 2ef4c58
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/indexer/src/app/modules/agent.module.ts b/indexer/src/app/modules/agent.module.ts
@@ -106,7 +106,7 @@ export class Agent {
         });
 
         const queryVector = await embeddingModel.embedQuery(
-          formatChatHistory(requestMessages.slice(-1)),
+          formatChatHistory(requestMessages),
         );
         payload.vector = queryVector;
       }
@@ -120,6 +120,12 @@ export class Agent {
       retrievedDocs = response.data
         .map((doc: any) => JSON.stringify(doc))
         .join('\n');
+
+      console.log({
+        payload,
+        requestMessages,
+        retrievedDocs,
+      });
     }
 
     const parser = new StringOutputParser();