Added a test-case for the mistralai api

This commit is contained in:
Måns Abrahamsson
2025-01-24 14:14:56 +01:00
parent 68446577f4
commit 46bbd559d2

View File

@@ -16,6 +16,25 @@ import sys
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 @@ if __name__ == '__main__':
# test_gemini_model()
# test_azure_openai_model()
# test_deepseek_model()
test_ollama_model()
# test_ollama_model()
test_mistral_model()