getting locations to myagent

app.py

@@ -80,16 +80,18 @@ with open("prompts.yaml", 'r') as stream:
 class MyAgent(MultiStepAgent):  # defining agent flow
 
     def step(self, memory_step: ActionStep, location: str) -> Union[None, Any]:  # Add location parameter
-        print(kwargs)
-        # 1. Call the tools
+        # 1. Access location from kwargs
+        location = kwargs.get("location")
+        print(f"Location received in step: {location}")  # Add print statement
+        # 2. Call the tools
         currency_info = get_currency_from_timezone(location)
         time_info = get_current_time_in_timezone(location)
 
-        # 2. Print the outputs
+        # 3. Print the outputs
         print(currency_info)
         print(time_info)
 
-        # 3. Final answer
+        # 4. Final answer
         final_answer = f"Task outcome (short version): The currency in {timezone} is {currency_info}. The current time in {timezone} is {time_info}."
         return final_answer  # Return the final answer