Fixed prompt bug

antoninoLorenzo · Jul 1, 2024 · 12f0b26 · 12f0b26
1 parent 257d3d0
commit 12f0b26
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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(