File tree Expand file tree Collapse file tree
sentry_sdk/integrations/openai_agents Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -124,12 +124,12 @@ def new_wrapped_get_model(
124124 agents .run_internal .run_loop .get_model = new_wrapped_get_model
125125
126126 @wraps (run_loop .run_single_turn )
127- async def patched_run_single_turn (
127+ async def new_wrapped_run_single_turn (
128128 * args : "Any" , ** kwargs : "Any"
129129 ) -> "SingleStepResult" :
130130 return await _run_single_turn (run_loop .run_single_turn , * args , ** kwargs )
131131
132- agents .run .run_single_turn = patched_run_single_turn
132+ agents .run .run_single_turn = new_wrapped_run_single_turn
133133
134134 @wraps (run_loop .run_single_turn_streamed )
135135 async def new_wrapped_run_single_turn_streamed (
@@ -174,7 +174,7 @@ async def old_wrapped_get_all_tools(
174174 @wraps (AgentRunner ._get_model .__func__ )
175175 def old_wrapped_get_model (
176176 cls : "agents.Runner" , agent : "agents.Agent" , run_config : "agents.RunConfig"
177- ) -> "list[ agents.Model] " :
177+ ) -> "agents.Model" :
178178 return _get_model (original_get_model , agent , run_config )
179179
180180 agents .run .AgentRunner ._get_model = classmethod (old_wrapped_get_model )
You can’t perform that action at this time.
0 commit comments