Skip to content

Commit 802979f

Browse files
Merge branch 'webb/openai-agents/run-single-turn-streamed' into webb/openai-agents/execute-handoffs
2 parents 9c9973a + 8d0aff5 commit 802979f

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
@@ -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)

0 commit comments

Comments
 (0)