diff options
| -rw-r--r-- | src/client/views/nodes/ChatBox/Agent.ts | 9 | ||||
| -rw-r--r-- | src/client/views/nodes/ChatBox/prompts.ts | 4 |
2 files changed, 9 insertions, 4 deletions
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> |