Update example/vllm_v2_weather_agent.py

example/vllm_v2_weather_agent.py

@@ -7,35 +7,35 @@ EXAMPLE OUTPUT:
 
 ****************************************
 RUNNING QUERY: What's the weather for Paris, TX in fahrenheit?
-Turn 1
+Step 1
 ----------------------------------------
 
 Executing: get_geo_coordinates
 Arguments: {'city': 'Paris', 'state': 'TX'}
 Response: The coordinates for Paris, TX are: latitude 33.6609, longitude 95.5555
 
-Turn 2
+Step 2
 ----------------------------------------
 
 Executing: get_current_weather
 Arguments: {'latitude': [33.6609], 'longitude': [95.5555], 'unit': 'fahrenheit'}
 Response: The weather is 85 degrees fahrenheit. It is partly cloudy, with highs in the 90's.
 
-Turn 3
+Step 3
 ----------------------------------------
 Conversation Complete
 
 
 ****************************************
 RUNNING QUERY: Who won the most recent PGA?
-Turn 1
+Step 1
 ----------------------------------------
 
 Executing: no_relevant_function
 Arguments: {'user_query_span': 'Who won the most recent PGA?'}
 Response: No relevant function for your request was found. We will stop here.
 
-Turn 2
+Step 2
 ----------------------------------------
 Conversation Complete
 """
@@ -46,7 +46,7 @@ class WeatherConfig:
     api_key: str = "" # FILL IN WITH YOUR VLLM_ENDPOINT_KEY
     api_base: str = "" # FILL IN WITH YOUR VLLM_ENDPOINT
     model: Optional[str] = None
-    max_turns: int = 5
+    max_steps: int = 5
 
 class WeatherTools:
     """Collection of available tools/functions for the weather agent"""
@@ -192,8 +192,8 @@ class WeatherAgent:
         print ("*" * 40)
         print (f"RUNNING QUERY: {initial_query}")
 
-        for turn in range(self.config.max_turns):
-            print(f"\nTurn {turn + 1}")
+        for step in range(self.config.max_steps):
+            print(f"\nStep {step + 1}")
             print("-" * 40)
 
             response = self.client.chat.completions.create(
@@ -217,8 +217,8 @@ class WeatherAgent:
 
             self.process_tool_calls(message)
 
-        if turn >= self.config.max_turns - 1:
-            print("Maximum turns reached")
+        if step >= self.config.max_steps - 1:
+            print("Maximum steps reached")
 
 def main():
     # Example usage