Use durable function tools (#34)

* Upgrade Restate SDK and fix readme

* Fix function tools

Author: gvdongen

openai-agents/examples/app/rollback_agent.py

@@ -1,9 +1,9 @@
 import restate
 
 from typing import Callable
-from agents import Agent, RunContextWrapper, function_tool
+from agents import Agent, RunContextWrapper
 from pydantic import Field, BaseModel, ConfigDict
-from restate.ext.openai import restate_context, DurableRunner, raise_terminal_errors
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.models import HotelBooking, FlightBooking, BookingPrompt, BookingResult
 from app.utils.utils import (
@@ -22,7 +22,7 @@ class BookingContext(BaseModel):
 
 
 # Functions raise terminal errors instead of feeding them back to the agent
-@function_tool(failure_error_function=raise_terminal_errors)
+@durable_function_tool
 async def book_hotel(
     wrapper: RunContextWrapper[BookingContext], booking: HotelBooking
 ) -> BookingResult:
@@ -40,7 +40,7 @@ async def book_hotel(
     )
 
 
-@function_tool(failure_error_function=raise_terminal_errors)
+@durable_function_tool
 async def book_flight(
     wrapper: RunContextWrapper[BookingContext], booking: FlightBooking
 ) -> BookingResult:

openai-agents/template/agent.py

@@ -1,7 +1,7 @@
 import restate
 
-from agents import Agent, function_tool
-from restate.ext.openai import restate_context, DurableRunner
+from agents import Agent
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from utils.utils import (
     fetch_weather,
@@ -11,7 +11,7 @@
 )
 
 
-@function_tool
+@durable_function_tool
 async def get_weather(req: WeatherRequest) -> WeatherResponse:
     """Get the current weather for a given city."""
     # Do durable steps using the Restate context

openai-agents/tour-of-agents/app/durable_agent.py

@@ -6,16 +6,15 @@
     RunConfig,
     RunContextWrapper,
     ModelSettings,
-    function_tool,
 )
 from restate import Service, VirtualObject
-from restate.ext.openai import restate_context, DurableRunner
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.models import WeatherPrompt, WeatherRequest, WeatherResponse
 from app.utils.utils import fetch_weather
 
 
-@function_tool
+@durable_function_tool
 async def get_weather(city: WeatherRequest) -> WeatherResponse:
     """Get the current weather for a given city."""
     return await restate_context().run_typed("get weather", fetch_weather, req=city)

openai-agents/tour-of-agents/app/error_handling.py

@@ -2,20 +2,20 @@
 
 import restate
 
-from agents import Agent, function_tool
+from agents import Agent
 from restate.ext.openai import (
     restate_context,
     DurableRunner,
-    raise_terminal_errors,
     LlmRetryOpts,
+    durable_function_tool,
 )
 
 from app.utils.models import WeatherPrompt, WeatherRequest, WeatherResponse
 from app.utils.utils import fetch_weather
 
 
 # <start_here>
-@function_tool(failure_error_function=raise_terminal_errors)
+@durable_function_tool
 async def get_weather(city: WeatherRequest) -> WeatherResponse:
     """Get the current weather for a given city."""
     return await restate_context().run_typed("get weather", fetch_weather, req=city)

openai-agents/tour-of-agents/app/human_approval_agent.py

@@ -1,7 +1,7 @@
 import restate
 
-from agents import Agent, function_tool
-from restate.ext.openai import restate_context, DurableRunner
+from agents import Agent
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.models import ClaimPrompt
 from app.utils.utils import (
@@ -11,7 +11,7 @@
 
 
 # <start_here>
-@function_tool
+@durable_function_tool
 async def human_approval(claim: InsuranceClaim) -> str:
     """Ask for human approval for high-value claims."""
 

openai-agents/tour-of-agents/app/human_approval_agent_with_timeout.py

@@ -1,8 +1,8 @@
 from datetime import timedelta
 
 import restate
-from agents import Agent, function_tool
-from restate.ext.openai import restate_context, DurableRunner
+from agents import Agent
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.models import ClaimPrompt
 from app.utils.utils import (
@@ -11,7 +11,7 @@
 )
 
 
-@function_tool
+@durable_function_tool
 async def human_approval(claim: InsuranceClaim) -> str:
     """Ask for human approval for high-value claims."""
     # Create an awakeable for human approval

openai-agents/tour-of-agents/app/multi_agent_remote.py

@@ -1,21 +1,21 @@
 import restate
 
-from agents import Agent, function_tool
-from restate.ext.openai import restate_context, DurableRunner
+from agents import Agent
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.utils import InsuranceClaim, run_eligibility_agent, run_fraud_agent
 
 
 # Durable service call to the eligibility agent; persisted and retried by Restate
-@function_tool
+@durable_function_tool
 async def check_eligibility(claim: InsuranceClaim) -> str:
     """Analyze claim eligibility."""
     return await restate_context().service_call(run_eligibility_agent, claim)
 
 
 # <start_here>
 # Durable service call to the fraud agent; persisted and retried by Restate
-@function_tool
+@durable_function_tool
 async def check_fraud(claim: InsuranceClaim) -> str:
     """Analyze the probability of fraud."""
     return await restate_context().service_call(run_fraud_agent, claim)

openai-agents/tour-of-agents/app/parallel_tools_agent.py

@@ -1,7 +1,7 @@
 import restate
 
-from agents import Agent, function_tool
-from restate.ext.openai import restate_context, DurableRunner
+from agents import Agent
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.utils import (
     InsuranceClaim,
@@ -12,7 +12,7 @@
 
 
 # <start_here>
-@function_tool
+@durable_function_tool
 async def calculate_metrics(claim: InsuranceClaim) -> list[str]:
     """Calculate claim metrics."""
     ctx = restate_context()

openai-agents/tour-of-agents/app/sub_workflow_agent.py

@@ -1,6 +1,6 @@
 import restate
-from agents import Agent, function_tool
-from restate.ext.openai import restate_context, DurableRunner
+from agents import Agent
+from restate.ext.openai import restate_context, DurableRunner, durable_function_tool
 
 from app.utils.models import ClaimPrompt
 from app.utils.utils import (
@@ -33,7 +33,7 @@ async def review(ctx: restate.Context, claim: InsuranceClaim) -> str:
 
 
 # <start_here>
-@function_tool
+@durable_function_tool
 async def human_approval(claim: InsuranceClaim) -> str:
     """Ask for human approval for high-value claims."""
     return await restate_context().service_call(review, claim)

openai-agents/tour-of-agents/app/utils/models.py

@@ -1,5 +1,3 @@
-from typing import Optional, Dict, Any
-
 from pydantic.alias_generators import to_camel as camelize
 
 from pydantic import BaseModel, ConfigDict
@@ -12,7 +10,9 @@ class WeatherPrompt(BaseModel):
 
 
 class ClaimPrompt(BaseModel):
-    message: str = "Process my hospital bill of 2024-10-01 for 3000USD for a broken leg at General Hospital."
+    message: str = (
+        "Process my hospital bill of 2024-10-01 for 3000USD for a broken leg at General Hospital."
+    )
 
 
 class ChatMessage(BaseModel):