Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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:
|