Skip to content

Commit bea39ad

Browse files
committed
Adjust create_activity_tool test expectations
1 parent a3c10c7 commit bea39ad

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/openai_agents/test_context.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,6 @@ def test_activity_as_tool_creates_function_tool(self, mock_function_tool, mock_f
103103
# Verify function_schema was called correctly
104104
mock_function_schema.assert_called_once_with(
105105
func=mock_activity_func._function._func,
106-
name_override="test_activity",
107106
docstring_style=None,
108107
description_override="Custom description",
109108
use_docstring_info=True,
@@ -159,6 +158,7 @@ def test_activity_as_tool_run_activity_with_retry(self, mock_function_tool, mock
159158

160159
mock_activity_func = Mock()
161160
mock_activity_func._function._name = "test_activity"
161+
mock_activity_func._function._trigger = None
162162
mock_activity_func._function._func = lambda x: x
163163

164164
mock_schema = Mock()
@@ -201,6 +201,7 @@ def test_activity_as_tool_run_activity_without_retry(self, mock_function_tool, m
201201

202202
mock_activity_func = Mock()
203203
mock_activity_func._function._name = "test_activity"
204+
mock_activity_func._function._trigger = None
204205
mock_activity_func._function._func = lambda x: x
205206

206207
mock_schema = Mock()

0 commit comments

Comments
 (0)