12
12
pytestmark = providers .mistral .mark
13
13
14
14
15
- @pytest .fixture (scope = "module" )
15
+ @pytest .fixture ()
16
16
def streaming_model ():
17
17
return MistralModel (
18
18
model_id = "mistral-medium-latest" ,
@@ -24,7 +24,7 @@ def streaming_model():
24
24
)
25
25
26
26
27
- @pytest .fixture (scope = "module" )
27
+ @pytest .fixture ()
28
28
def non_streaming_model ():
29
29
return MistralModel (
30
30
model_id = "mistral-medium-latest" ,
@@ -36,12 +36,12 @@ def non_streaming_model():
36
36
)
37
37
38
38
39
- @pytest .fixture (scope = "module" )
39
+ @pytest .fixture ()
40
40
def system_prompt ():
41
41
return "You are an AI assistant that provides helpful and accurate information."
42
42
43
43
44
- @pytest .fixture (scope = "module" )
44
+ @pytest .fixture ()
45
45
def tools ():
46
46
@strands .tool
47
47
def tool_time () -> str :
@@ -54,12 +54,12 @@ def tool_weather() -> str:
54
54
return [tool_time , tool_weather ]
55
55
56
56
57
- @pytest .fixture (scope = "module" )
57
+ @pytest .fixture ()
58
58
def streaming_agent (streaming_model , tools ):
59
59
return Agent (model = streaming_model , tools = tools )
60
60
61
61
62
- @pytest .fixture (scope = "module" )
62
+ @pytest .fixture ()
63
63
def non_streaming_agent (non_streaming_model , tools ):
64
64
return Agent (model = non_streaming_model , tools = tools )
65
65
@@ -69,7 +69,7 @@ def agent(request):
69
69
return request .getfixturevalue (request .param )
70
70
71
71
72
- @pytest .fixture (scope = "module" )
72
+ @pytest .fixture ()
73
73
def weather ():
74
74
class Weather (BaseModel ):
75
75
"""Extracts the time and weather from the user's message with the exact strings."""
0 commit comments