citation parsing working much better

1 files changed, 3 insertions, 5 deletions

diff --git a/src/client/views/nodes/ChatBox/Agent.ts b/src/client/views/nodes/ChatBox/Agent.ts
index 6757b2ce8..fd3c6e5e8 100644
--- a/src/client/views/nodes/ChatBox/Agent.ts
+++ b/src/client/views/nodes/ChatBox/Agent.ts
@@ -95,6 +95,8 @@ export class Agent {
                         break;
                     } else {
                         console.log('Error: No valid action');
+                        this.interMessages.push({ role: 'user', content: 'No valid action, try again.' });
+                        break;
                     }
                 } else if (key === 'action_input') {
                     const actionInput = builder.build({ action_input: step[key] });
@@ -116,11 +118,7 @@ export class Agent {
                     }
                 } else if (key === 'answer') {
                     console.log('Answer found. Ending query.');
-                    const answerContent = builder.build({ answer: step[key] });
-                    this.messages.push({ role: 'assistant', content: answerContent });
-                    this.interMessages = [];
-                    console.log(this.messages);
-                    return answerContent;
+                    return result;
                 }
             }
         }