File tree Expand file tree Collapse file tree 1 file changed +15
-2
lines changed Expand file tree Collapse file tree 1 file changed +15
-2
lines changed Original file line number Diff line number Diff line change 22
33from __future__ import annotations
44
5+ from typing import Generator
56import pytest
7+ from pytest_mock import MockerFixture , AsyncMockType
8+
9+ type AgentFixtures = Generator [
10+ tuple [
11+ AsyncMockType ,
12+ AsyncMockType ,
13+ ],
14+ None ,
15+ None ,
16+ ]
617
718
819@pytest .fixture (name = "prepare_agent_mocks" , scope = "function" )
9- def prepare_agent_mocks_fixture (mocker ):
20+ def prepare_agent_mocks_fixture (
21+ mocker : MockerFixture ,
22+ ) -> AgentFixtures :
1023 """Prepare for mock for the LLM agent.
1124
1225 Provides common mocks for AsyncLlamaStackClient and AsyncAgent
1326 with proper agent_id setup to avoid initialization errors.
1427
15- Returns :
28+ Yields :
1629 tuple: (mock_client, mock_agent)
1730 """
1831 mock_client = mocker .AsyncMock ()
You can’t perform that action at this time.
0 commit comments