modified: llama_github/llm_integration/initial_load.py

Author: Jet Xu

llama_github/llm_integration/initial_load.py

@@ -67,11 +67,11 @@ def __init__(self,
         elif mistral_api_key is not None and mistral_api_key != "" and self.llm is None:
             logger.info("Initializing Codestral API...")
             from langchain_mistralai.chat_models import ChatMistralAI
-            self.llm = ChatMistralAI(mistral_api_key=mistral_api_key, model="mistral-large-latest")
+            self.llm = ChatMistralAI(mistral_api_key=mistral_api_key, model="open-mistral-nemo")
             self.llm_simple = ChatMistralAI(
                 mistral_api_key=mistral_api_key, 
                 model="mistral-small-latest",
-                temperature=0.26
+                temperature=0.2
             )
             self.model_type = "OpenAI"
         elif openai_api_key is not None and openai_api_key != "" and self.llm is None: