From 741292e1d26ef3ede5316e947bb4ebdf4af510ea Mon Sep 17 00:00:00 2001 From: alexisxy Date: Sat, 23 Sep 2023 00:16:18 -0400 Subject: [PATCH] fix force_prefix missing bug --- agent/agent.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/agent/agent.py b/agent/agent.py index 90e3692..923ebce 100644 --- a/agent/agent.py +++ b/agent/agent.py @@ -126,6 +126,10 @@ class PromptAgent(Agent): n = 0 while True: response = call_llm(lm_config, prompt) + force_prefix = self.prompt_constructor.instruction[ + "meta_data" + ].get("force_prefix", "") + response = f"{force_prefix}{response}" n += 1 try: parsed_response = self.prompt_constructor.extract_action(