Update Agent.py

Agent.py

@@ -99,7 +99,7 @@ agent = initialize_agent(
     verbose=True
 )
 
-def run_agent(state: GaiaState) -> GaiaState:
+def run_agent(state: MessagesState) -> MessagesState:
     result = agent.run(state.input)
     state.output = result
     state.intermediate_steps.append(result)