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:
|