Starting to test mistral too.... not fully working yet

tests/models/test_azureai_studio.py

@@ -11,10 +11,10 @@
 pytestmark = pytest.mark.needs_credentials
 
 
-def test_azureai_openai_chat_smoke(rate_limiter):
-    azureai_studio_endpoint = env_or_fail("AZURE_AI_STUDIO_ENDPOINT")
-    azureai_studio_deployment = env_or_fail("AZURE_AI_STUDIO_DEPLOYMENT")
-    azureai_studio_key = env_or_fail("AZURE_AI_STUDIO_KEY")
+def test_azureai_phi3_chat_smoke(rate_limiter):
+    azureai_studio_endpoint = env_or_fail("AZURE_AI_STUDIO_PHI3_ENDPOINT")
+    azureai_studio_deployment = env_or_fail("AZURE_AI_STUDIO_PHI3_DEPLOYMENT")
+    azureai_studio_key = env_or_fail("AZURE_AI_STUDIO_PHI3_KEY")
 
     lm = models.AzureAIStudioChat(
         azureai_studio_endpoint=azureai_studio_endpoint,
@@ -30,8 +30,35 @@ def test_azureai_openai_chat_smoke(rate_limiter):
         lm += "What is 1 + 1?"
 
     with assistant():
-        lm += gen(max_tokens=10, name="text")
+        lm += gen(max_tokens=10, name="text", temperature=0.5)
         lm += "Pick a number: "
 
     print(str(lm))
     assert len(lm["text"]) > 0
+
+
+def test_azureai_mistral_chat_smoke(rate_limiter):
+    azureai_studio_endpoint = env_or_fail("AZURE_AI_STUDIO_MISTRAL_CHAT_ENDPOINT")
+    azureai_studio_deployment = env_or_fail("AZURE_AI_STUDIO_MISTRAL_CHAT_DEPLOYMENT")
+    azureai_studio_key = env_or_fail("AZURE_AI_STUDIO_MISTRAL_CHAT_KEY")
+
+    lm = models.AzureAIStudioChat(
+        azureai_studio_endpoint=azureai_studio_endpoint,
+        azureai_studio_deployment=azureai_studio_deployment,
+        azureai_studio_key=azureai_studio_key,
+    )
+    assert isinstance(lm, models.AzureAIStudioChat)
+    lm.engine.cache.clear()
+
+    with system():
+        lm += "You are a math wiz."
+
+    with user():
+        lm += "What is 1 + 1?"
+
+    with assistant():
+        lm += gen(max_tokens=10, name="text", temperature=0.5)
+        lm += "Pick a number: "
+
+    print(str(lm))
+    assert len(lm["text"]) < 0