josondev commited on
Commit
e85946a
·
verified ·
1 Parent(s): 9ff1f1d

Update veryfinal.py

Browse files
Files changed (1) hide show
  1. 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 langchain_google_genai import ChatGoogleGenerativeAI
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 = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
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})