11"""Initialize Temporal OpenAI Agents overrides."""
22
33from contextlib import contextmanager
4- from datetime import timedelta
54from typing import Optional
65
76from agents import set_trace_provider
87from agents .run import get_default_agent_runner , set_default_agent_runner
98from agents .tracing import get_trace_provider
109from agents .tracing .provider import DefaultTraceProvider
1110
12- from temporalio .common import Priority , RetryPolicy
1311from temporalio .contrib .openai_agents ._openai_runner import TemporalOpenAIRunner
1412from temporalio .contrib .openai_agents ._temporal_trace_provider import (
1513 TemporalTraceProvider ,
1614)
1715from temporalio .contrib .openai_agents .model_parameters import ModelActivityParameters
18- from temporalio .workflow import ActivityCancellationType , VersioningIntent
1916
2017
2118@contextmanager
2219def set_open_ai_agent_temporal_overrides (
23- model_params : ModelActivityParameters ,
20+ model_params : Optional [ ModelActivityParameters ] = None ,
2421 auto_close_tracing_in_workflows : bool = False ,
2522):
2623 """Configure Temporal-specific overrides for OpenAI agents.
@@ -41,11 +38,15 @@ def set_open_ai_agent_temporal_overrides(
4138
4239 Args:
4340 model_params: Configuration parameters for Temporal activity execution of model calls.
41+ auto_close_tracing_in_workflows: If set to true, close tracing spans immediately.
4442
4543 Returns:
4644 A context manager that yields the configured TemporalTraceProvider.
4745
4846 """
47+ if model_params is None :
48+ model_params = ModelActivityParameters ()
49+
4950 if (
5051 not model_params .start_to_close_timeout
5152 and not model_params .schedule_to_close_timeout
0 commit comments