Skip to content

Commit fe251a7

Browse files
committed
Added new test for mistral models
1 parent 41ddd2b commit fe251a7

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/test_llm_api.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@ def get_env_value(key, provider):
3838
"openai": {"api_key": "OPENAI_API_KEY", "base_url": "OPENAI_ENDPOINT"},
3939
"azure_openai": {"api_key": "AZURE_OPENAI_API_KEY", "base_url": "AZURE_OPENAI_ENDPOINT"},
4040
"gemini": {"api_key": "GOOGLE_API_KEY"},
41-
"deepseek": {"api_key": "DEEPSEEK_API_KEY", "base_url": "DEEPSEEK_ENDPOINT"}
41+
"deepseek": {"api_key": "DEEPSEEK_API_KEY", "base_url": "DEEPSEEK_ENDPOINT"},
42+
"mistral": {"api_key": "MISTRAL_API_KEY", "base_url": "MISTRAL_ENDPOINT"},
4243
}
4344

4445
if provider in env_mappings and key in env_mappings[provider]:
@@ -116,11 +117,16 @@ def test_deepseek_r1_ollama_model():
116117
config = LLMConfig(provider="ollama", model_name="deepseek-r1:14b")
117118
test_llm(config, "How many 'r's are in the word 'strawberry'?")
118119

120+
def test_mistral_model():
121+
config = LLMConfig(provider="mistral", model_name="pixtral-large-latest")
122+
test_llm(config, "Describe this image", "assets/examples/test.png")
123+
119124
if __name__ == "__main__":
120125
# test_openai_model()
121126
# test_gemini_model()
122127
# test_azure_openai_model()
123-
test_deepseek_model()
128+
#test_deepseek_model()
124129
# test_ollama_model()
125130
# test_deepseek_r1_model()
126131
# test_deepseek_r1_ollama_model()
132+
test_mistral_model()

0 commit comments

Comments
 (0)