refactor(examples): update agent function signatures

- Change return type from Run[Output] to Output directly

- Add .run() method calls to execute agent functions

- Clean up imports ordering

Author: Pierre

examples/02_agent_with_tools.py

@@ -9,12 +9,12 @@
 
 import asyncio
 from datetime import date, datetime
+from zoneinfo import ZoneInfo
 
 from pydantic import BaseModel, Field
-from zoneinfo import ZoneInfo
 
 import workflowai
-from workflowai import Model, Run
+from workflowai import Model
 
 
 def get_current_date() -> str:
@@ -65,7 +65,7 @@ class HistoricalEventOutput(BaseModel):
     model=Model.GEMINI_1_5_FLASH_LATEST,
     tools=[get_current_date, calculate_days_between],
 )
-async def analyze_historical_event(event_input: HistoricalEventInput) -> Run[HistoricalEventOutput]:
+async def analyze_historical_event(event_input: HistoricalEventInput) -> HistoricalEventOutput:
     """
     Find information about historical events and calculate days since they occurred.
 
@@ -88,7 +88,7 @@ async def main():
     # Example: Query about the moon landing
     print("\nExample: First Moon Landing")
     print("-" * 50)
-    result = await analyze_historical_event(
+    result = await analyze_historical_event.run(
         HistoricalEventInput(query="When was the first moon landing?"),
     )
     print(result.output)
@@ -99,7 +99,7 @@ async def main():
     # Example: Query about World War II
     print("\nExample: End of World War II")
     print("-" * 50)
-    result = await analyze_historical_event(
+    result = await analyze_historical_event.run(
         HistoricalEventInput(query="When did World War II end?"),
     )
     print(result.output)