WebashalarForML commited on
Commit
78c0ec0
·
verified ·
1 Parent(s): c9a2f8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -42,13 +42,14 @@ print("DATABASE URI:", DATABASE_URI)
42
 
43
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")
44
  os.environ["GROQ_API_KEY"] = GROQ_API_KEY
 
45
 
46
  # Use ChatGroq LLM (which does not require a Hugging Face API token)
47
  from langchain_groq import ChatGroq
48
  from langchain_mistralai.chat_models import ChatMistralAI
49
 
50
- ll = ChatMistralAI(api_key=api_key)
51
- llm = ChatGroq(model="llama3-70b-8192")
52
 
53
  @tool
54
  def db_query_tool(query: str) -> str:
 
42
 
43
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")
44
  os.environ["GROQ_API_KEY"] = GROQ_API_KEY
45
+ os.environ["MISTRAL_API_KEY"] = MISTRAL_API_KEY
46
 
47
  # Use ChatGroq LLM (which does not require a Hugging Face API token)
48
  from langchain_groq import ChatGroq
49
  from langchain_mistralai.chat_models import ChatMistralAI
50
 
51
+ llm = ChatMistralAI(model="mistral-large-latest")
52
+ #llm = ChatGroq(model="llama3-70b-8192")
53
 
54
  @tool
55
  def db_query_tool(query: str) -> str: