shahidshaikh commited on
Commit
183a0f1
·
verified ·
1 Parent(s): 148c386

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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-large-latest"), apiKey=os.getenv("MISTRAL_API_KEY"), 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")
 
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")