Update main.py

main.py

@@ -109,7 +109,7 @@ def state_modifier(state) -> list[BaseMessage]:
             "collection_files": collection_files,
             "messages": state["messages"]
         })
-        
+        print(state["messages"])
         return trim_messages(
             formatted_prompt,
             token_counter=len,
@@ -119,7 +119,7 @@ def state_modifier(state) -> list[BaseMessage]:
             include_system=True,
             allow_partial=False,
         )
-    print(state["messages"])
+        
     except Exception as e:
         print(f"Error in state modifier: {str(e)}")
         return state["messages"]