Update main.py

main.py

@@ -104,22 +104,25 @@ prompt = ChatPromptTemplate.from_messages([
 
 def state_modifier(state) -> list[BaseMessage]:
     collection_files = "None"
-    # Format the prompt with the current state
-    formatted_prompt = prompt.invoke({
-        "collection_files": collection_files,
-        "messages": state["messages"]
-    })
-    
-    # Trim the messages
-    return trim_messages(
-        formatted_prompt,
-        token_counter=len,
-        max_tokens=16000,
-        strategy="last",
-        start_on="human",
-        include_system=True,
-        allow_partial=False,
-    )
+    try:
+        formatted_prompt = prompt.invoke({
+            "collection_files": collection_files,
+            "messages": state["messages"]
+        })
+        
+        return trim_messages(
+            formatted_prompt,
+            token_counter=len,
+            max_tokens=16000,
+            strategy="last",
+            start_on="human",
+            include_system=True,
+            allow_partial=False,
+        )
+    print(state["messages"])
+    except Exception as e:
+        print(f"Error in state modifier: {str(e)}")
+        return state["messages"]
 
 # Create the agent with the Python execution tool
 agent = create_react_agent(