From ff5fa6654a74096bbd9f14c5d236b8d08861fe2f Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Wed, 10 Jul 2024 18:02:24 -0400
Subject: Works much better now.

---
 src/client/views/nodes/ChatBox/Agent.ts   | 9 +++++----
 src/client/views/nodes/ChatBox/prompts.ts | 4 ++++
 2 files changed, 9 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/client/views/nodes/ChatBox/Agent.ts b/src/client/views/nodes/ChatBox/Agent.ts
index 355acb19f..cec5e1770 100644
--- a/src/client/views/nodes/ChatBox/Agent.ts
+++ b/src/client/views/nodes/ChatBox/Agent.ts
@@ -49,7 +49,7 @@ export class Agent {
         console.log(`System prompt: ${systemPrompt}`);
         this.interMessages = [{ role: 'system', content: systemPrompt }];
 
-        this.interMessages.push({ role: 'assistant', content: `<query>${question}</query>` });
+        this.interMessages.push({ role: 'user', content: `<query>${question}</query>` });
 
         const parser = new XMLParser();
         const builder = new XMLBuilder();
@@ -85,7 +85,7 @@ export class Agent {
                                 text: builder.build({ action_rules: this.tools[currentAction].getActionRule() }),
                             },
                         ];
-                        this.interMessages.push({ role: 'assistant', content: nextPrompt });
+                        this.interMessages.push({ role: 'user', content: nextPrompt });
                         break;
                     } else {
                         console.log('Error: No valid action');
@@ -97,7 +97,8 @@ export class Agent {
                         try {
                             const observation = await this.processAction(currentAction, step[key]);
                             const nextPrompt = [{ type: 'text', text: '<observation>' }, ...observation, { type: 'text', text: '</observation>' }];
-                            this.interMessages.push({ role: 'assistant', content: nextPrompt });
+                            console.log('Action result: ' + observation);
+                            this.interMessages.push({ role: 'user', content: nextPrompt });
                             break;
                         } catch (error) {
                             console.log(`Error processing action: ${error}`);
@@ -122,7 +123,7 @@ export class Agent {
     }
 
     private async execute(): Promise<string> {
-        console.log('Messages: ' + this.interMessages);
+        console.log(this.interMessages);
         const completion = await this.client.chat.completions.create({
             model: 'gpt-4o',
             messages: this.interMessages as ChatCompletionMessageParam[],
diff --git a/src/client/views/nodes/ChatBox/prompts.ts b/src/client/views/nodes/ChatBox/prompts.ts
index ffea13788..c011d65b7 100644
--- a/src/client/views/nodes/ChatBox/prompts.ts
+++ b/src/client/views/nodes/ChatBox/prompts.ts
@@ -30,6 +30,8 @@ export function getReactPrompt(tools: Tool[], chatHistory: string): string {
         <action>wikipedia</action>
     </step1>
 
+    THEN PAUSE AND DO NOT OUTPUT ANYTHING.
+
     You will be called again with this:
     <action_rules>
         {
@@ -56,6 +58,8 @@ export function getReactPrompt(tools: Tool[], chatHistory: string): string {
         </action_input>
     </step2>
 
+    THEN PAUSE AND DO NOT OUTPUT ANYTHING.
+
     You will then be called again with this:
     <observation>France is a country. The capital is Paris.</observation>
 
-- 
cgit v1.2.3-70-g09d2