Added a test-case for the mistralai api

Author: pad918

tests/test_llm_api.py

@@ -16,6 +16,25 @@
 sys.path.append(".")
 
 
+def test_mistral_model():
+    from langchain_core.messages import HumanMessage
+    from src.utils import utils
+
+    llm = utils.get_llm_model(
+        provider="mistral",
+        model_name="mistral-large-latest",
+        temperature=0.8,
+        base_url=os.getenv("MISTRAL_ENDPOINT", ""),
+        api_key=os.getenv("MISTRAL_API_KEY", "")
+    )
+    message = HumanMessage(
+        content=[
+            {"type": "text", "text": "who are you?"}
+        ]
+    )
+    ai_msg = llm.invoke([message])
+    print(ai_msg.content)
+
 def test_openai_model():
     from langchain_core.messages import HumanMessage
     from src.utils import utils
@@ -128,4 +147,5 @@ def test_ollama_model():
     # test_gemini_model()
     # test_azure_openai_model()
     # test_deepseek_model()
-    test_ollama_model()
+    # test_ollama_model()
+    test_mistral_model()