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})
|