Skip to content

Fix: Update mistral tests to avoid shared agents #398

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jul 9, 2025
Merged

Fix: Update mistral tests to avoid shared agents #398

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions tests_integ/models/test_model_mistral.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
pytestmark = providers.mistral.mark


@pytest.fixture(scope="module")
@pytest.fixture()
def streaming_model():
return MistralModel(
model_id="mistral-medium-latest",
Expand All @@ -24,7 +24,7 @@ def streaming_model():
)


@pytest.fixture(scope="module")
@pytest.fixture()
def non_streaming_model():
return MistralModel(
model_id="mistral-medium-latest",
Expand All @@ -36,12 +36,12 @@ def non_streaming_model():
)


@pytest.fixture(scope="module")
@pytest.fixture()
def system_prompt():
return "You are an AI assistant that provides helpful and accurate information."


@pytest.fixture(scope="module")
@pytest.fixture()
def tools():
@strands.tool
def tool_time() -> str:
Expand All @@ -54,12 +54,12 @@ def tool_weather() -> str:
return [tool_time, tool_weather]


@pytest.fixture(scope="module")
@pytest.fixture()
def streaming_agent(streaming_model, tools):
return Agent(model=streaming_model, tools=tools)


@pytest.fixture(scope="module")
@pytest.fixture()
def non_streaming_agent(non_streaming_model, tools):
return Agent(model=non_streaming_model, tools=tools)

Expand All @@ -69,7 +69,7 @@ def agent(request):
return request.getfixturevalue(request.param)


@pytest.fixture(scope="module")
@pytest.fixture()
def weather():
class Weather(BaseModel):
"""Extracts the time and weather from the user's message with the exact strings."""
Expand Down