Return Weather object from get_weather

Author: AnatoliB

samples-v2/openai_agents/function_app.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+import json
 import os
 import azure.functions as func
 from openai import AsyncAzureOpenAI
@@ -8,6 +9,7 @@
 from model_activity import create_invoke_model_activity
 from azure.identity import DefaultAzureCredential
 from tools import activity_as_tool
+from pydantic import BaseModel
 
 
 #region Regular Azure OpenAI setup
@@ -132,19 +134,25 @@ def weather_expert(context):
     result = Runner.run_sync(agent, "What is the weather in Tokio?")
     return result.final_output
 
-@dataclass
-class Weather:
+class Weather(BaseModel):
     city: str
     temperature_range: str
     conditions: str
 
+    def to_json(self) -> str:
+        return self.model_dump_json()
+
+    @staticmethod
+    def from_json(data: str) -> "Weather":
+        return Weather(**json.loads(data))
+
 @app.activity_trigger(input_name="city")
-async def get_weather(city: str) -> str:
+async def get_weather(city: str) -> Weather:
     """
     Get the weather for a given city.
     """
     weather = Weather(city=city, temperature_range="14-20C", conditions="Sunny with wind.")
-    return f"Weather in {weather.city}: {weather.temperature_range}, {weather.conditions}"
+    return weather
 
 
 @function_tool

samples-v2/openai_agents/requirements.txt

@@ -7,3 +7,4 @@ azure-functions-durable
 azure-identity
 openai
 openai-agents
+pydantic