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 @@ -122,12 +122,12 @@ def new_wrapped_get_model(
122122 agents .run_internal .run_loop .get_model = new_wrapped_get_model
123123
124124 @wraps (run_loop .run_single_turn )
125- async def patched_run_single_turn (
125+ async def new_wrapped_run_single_turn (
126126 * args : "Any" , ** kwargs : "Any"
127127 ) -> "SingleStepResult" :
128128 return await _run_single_turn (run_loop .run_single_turn , * args , ** kwargs )
129129
130- agents .run .run_single_turn = patched_run_single_turn
130+ agents .run .run_single_turn = new_wrapped_run_single_turn
131131
132132 @wraps (run_loop .run_single_turn_streamed )
133133 async def new_wrapped_run_single_turn_streamed (
@@ -158,7 +158,7 @@ async def old_wrapped_get_all_tools(
158158 @wraps (AgentRunner ._get_model .__func__ )
159159 def old_wrapped_get_model (
160160 cls : "agents.Runner" , agent : "agents.Agent" , run_config : "agents.RunConfig"
161- ) -> "list[ agents.Model] " :
161+ ) -> "agents.Model" :
162162 return _get_model (original_get_model , agent , run_config )
163163
164164 agents .run .AgentRunner ._get_model = classmethod (old_wrapped_get_model )
You can’t perform that action at this time.
0 commit comments