From 12f0b269ad8c0497db87481f58a37d2dba020856 Mon Sep 17 00:00:00 2001 From: antoninoLorenzo Date: Mon, 1 Jul 2024 17:48:23 +0200 Subject: [PATCH] Fixed prompt bug --- src/agent/agent.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/agent/agent.py b/src/agent/agent.py index 7fd36da..daa42e5 100644 --- a/src/agent/agent.py +++ b/src/agent/agent.py @@ -42,9 +42,7 @@ def query(self, sid: int, user_in: str, rag=True, stream=True): context=context ) else: - print('Cutted: \n', self.user_plan_gen.split('\n')[:-3]) prompt = '\n'.join(self.user_plan_gen.split('\n')[:-3]) - print(f'Prompt ({len(prompt)}):\n{prompt}') prompt = prompt.format(user_input=user_in) self.mem.store_message(