1
1
"""Initialize Temporal OpenAI Agents overrides."""
2
2
3
3
from contextlib import contextmanager
4
- from datetime import timedelta
5
4
from typing import Optional
6
5
7
6
from agents import set_trace_provider
8
7
from agents .run import get_default_agent_runner , set_default_agent_runner
9
8
from agents .tracing import get_trace_provider
10
9
from agents .tracing .provider import DefaultTraceProvider
11
10
12
- from temporalio .common import Priority , RetryPolicy
13
11
from temporalio .contrib .openai_agents ._openai_runner import TemporalOpenAIRunner
14
12
from temporalio .contrib .openai_agents ._temporal_trace_provider import (
15
13
TemporalTraceProvider ,
16
14
)
17
15
from temporalio .contrib .openai_agents .model_parameters import ModelActivityParameters
18
- from temporalio .workflow import ActivityCancellationType , VersioningIntent
19
16
20
17
21
18
@contextmanager
22
19
def set_open_ai_agent_temporal_overrides (
23
- model_params : ModelActivityParameters ,
20
+ model_params : Optional [ ModelActivityParameters ] = None ,
24
21
auto_close_tracing_in_workflows : bool = False ,
25
22
):
26
23
"""Configure Temporal-specific overrides for OpenAI agents.
@@ -41,11 +38,15 @@ def set_open_ai_agent_temporal_overrides(
41
38
42
39
Args:
43
40
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.
44
42
45
43
Returns:
46
44
A context manager that yields the configured TemporalTraceProvider.
47
45
48
46
"""
47
+ if model_params is None :
48
+ model_params = ModelActivityParameters ()
49
+
49
50
if (
50
51
not model_params .start_to_close_timeout
51
52
and not model_params .schedule_to_close_timeout
0 commit comments