Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -23,7 +23,7 @@ def _llm():
|
|
| 23 |
provider = os.getenv("LLM_PROVIDER", "groq").lower()
|
| 24 |
if provider == "mistral":
|
| 25 |
from langchain_mistralai import ChatMistralAI
|
| 26 |
-
return ChatMistralAI(model=os.getenv("MISTRAL_BIG", "mistral-
|
| 27 |
elif provider == "huggingface":
|
| 28 |
from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
|
| 29 |
repo_id = os.getenv("HF_MODEL_ID", "mistralai/Mistral-Small-24B-Instruct-v0.1")
|
|
|
|
| 23 |
provider = os.getenv("LLM_PROVIDER", "groq").lower()
|
| 24 |
if provider == "mistral":
|
| 25 |
from langchain_mistralai import ChatMistralAI
|
| 26 |
+
return ChatMistralAI(model=os.getenv("MISTRAL_BIG", "mistral-small-latest"), temperature=0)
|
| 27 |
elif provider == "huggingface":
|
| 28 |
from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
|
| 29 |
repo_id = os.getenv("HF_MODEL_ID", "mistralai/Mistral-Small-24B-Instruct-v0.1")
|