File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
tests/contrib/openai_agents Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change 11import datetime
22import uuid
33from datetime import timedelta
4- from typing import Any , Optional , cast
4+ from typing import Any , Optional
55
66from agents import Span , Trace , TracingProcessor
77from agents .tracing import get_trace_provider
1313 TestModelProvider ,
1414 set_open_ai_agent_temporal_overrides ,
1515)
16- from temporalio .contrib .openai_agents ._temporal_trace_provider import (
17- TemporalTraceProvider ,
18- )
1916from temporalio .contrib .pydantic import pydantic_data_converter
2017from tests .contrib .openai_agents .test_openai import ResearchWorkflow , TestResearchModel
2118from tests .helpers import new_worker
@@ -51,7 +48,7 @@ async def test_tracing(client: Client):
5148 client = Client (** new_config )
5249
5350 with set_open_ai_agent_temporal_overrides ():
54- provider = cast ( TemporalTraceProvider , get_trace_provider () )
51+ provider = get_trace_provider ()
5552
5653 processor = MemoryTracingProcessor ()
5754 provider .set_processors ([processor ])
You can’t perform that action at this time.
0 commit comments