Skip to content

Commit 8d0aff5

Browse files
Merge branch 'webb/openai-agents/run-single-turn' into webb/openai-agents/run-single-turn-streamed
2 parents c96a268 + e44024a commit 8d0aff5

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

sentry_sdk/integrations/openai_agents/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)