Spaces:
Sleeping
Sleeping
Update veryfinal.py
Browse files- veryfinal.py +2 -2
veryfinal.py
CHANGED
|
@@ -6,7 +6,7 @@ load_dotenv()
|
|
| 6 |
|
| 7 |
# Imports
|
| 8 |
from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings
|
| 9 |
-
from
|
| 10 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 11 |
from langchain_community.document_loaders import WikipediaLoader
|
| 12 |
from langchain_community.document_loaders import ArxivLoader
|
|
@@ -147,7 +147,7 @@ json_chunks = text_splitter.split_documents(json_docs)
|
|
| 147 |
database = FAISS.from_documents(json_chunks, NVIDIAEmbeddings())
|
| 148 |
|
| 149 |
# Initialize LLM
|
| 150 |
-
llm =
|
| 151 |
|
| 152 |
# Create retriever and retriever tool
|
| 153 |
retriever = database.as_retriever(search_type="similarity", search_kwargs={"k": 3})
|
|
|
|
| 6 |
|
| 7 |
# Imports
|
| 8 |
from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings
|
| 9 |
+
from langchain_groq import ChatGroq
|
| 10 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 11 |
from langchain_community.document_loaders import WikipediaLoader
|
| 12 |
from langchain_community.document_loaders import ArxivLoader
|
|
|
|
| 147 |
database = FAISS.from_documents(json_chunks, NVIDIAEmbeddings())
|
| 148 |
|
| 149 |
# Initialize LLM
|
| 150 |
+
llm = ChatGroq(model="qwen-qwq-32b", temperature=0,api_key=os.getenv("GROQ_API_KEY"))
|
| 151 |
|
| 152 |
# Create retriever and retriever tool
|
| 153 |
retriever = database.as_retriever(search_type="similarity", search_kwargs={"k": 3})
|