Skip to content

Commit c7d10a2

Browse files
committed
fixed tests for new openai-agents
1 parent 1d57c99 commit c7d10a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integrations/openai_agents/test_openai_agents.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,13 +80,13 @@ def test_agent():
8080

8181

8282
@pytest.fixture
83-
def test_agent_custom_model():
83+
def test_agent_custom_model() -> Agent:
8484
"""Create a real Agent instance for testing."""
8585
return Agent(
8686
name="test_agent_custom_model",
8787
instructions="You are a helpful test assistant.",
8888
# the model could be agents.OpenAIChatCompletionsModel()
89-
model=MagicMock(model="my-custom-model"),
89+
model=MagicMock(spec=agents.Model, model="my-custom-model"),
9090
model_settings=ModelSettings(
9191
max_tokens=100,
9292
temperature=0.7,

0 commit comments

Comments
 (0)