@@ -38,7 +38,8 @@ def get_env_value(key, provider):
38
38
"openai" : {"api_key" : "OPENAI_API_KEY" , "base_url" : "OPENAI_ENDPOINT" },
39
39
"azure_openai" : {"api_key" : "AZURE_OPENAI_API_KEY" , "base_url" : "AZURE_OPENAI_ENDPOINT" },
40
40
"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" },
42
43
}
43
44
44
45
if provider in env_mappings and key in env_mappings [provider ]:
@@ -116,11 +117,16 @@ def test_deepseek_r1_ollama_model():
116
117
config = LLMConfig (provider = "ollama" , model_name = "deepseek-r1:14b" )
117
118
test_llm (config , "How many 'r's are in the word 'strawberry'?" )
118
119
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
+
119
124
if __name__ == "__main__" :
120
125
# test_openai_model()
121
126
# test_gemini_model()
122
127
# test_azure_openai_model()
123
- test_deepseek_model ()
128
+ # test_deepseek_model()
124
129
# test_ollama_model()
125
130
# test_deepseek_r1_model()
126
131
# test_deepseek_r1_ollama_model()
132
+ test_mistral_model ()
0 commit comments