girishwangikar commited on
Commit
d25a56b
Β·
verified Β·
1 Parent(s): e66953a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -9,11 +9,20 @@ from langchain_community.vectorstores import FAISS
9
  from langchain_community.document_loaders import PyPDFLoader
10
  from langchain_community.embeddings import HuggingFaceEmbeddings
11
  from dotenv import load_dotenv
 
12
 
13
  load_dotenv() # Load the GROQ API KEY
14
  GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
15
 
16
- llm = ChatGroq(temperature=0, model_name='llama-3.1-8b-instant', groq_api_key=GROQ_API_KEY)
 
 
 
 
 
 
 
 
17
 
18
  prompt = ChatPromptTemplate.from_template("""
19
  Answer the questions based on the provided context only.
 
9
  from langchain_community.document_loaders import PyPDFLoader
10
  from langchain_community.embeddings import HuggingFaceEmbeddings
11
  from dotenv import load_dotenv
12
+ from pydantic import ConfigDict
13
 
14
  load_dotenv() # Load the GROQ API KEY
15
  GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
16
 
17
+ # Configure Pydantic to allow arbitrary types
18
+ config = ConfigDict(arbitrary_types_allowed=True)
19
+
20
+ llm = ChatGroq(
21
+ temperature=0,
22
+ model_name='llama-3.1-8b-instant',
23
+ groq_api_key=GROQ_API_KEY,
24
+ model_config=config
25
+ )
26
 
27
  prompt = ChatPromptTemplate.from_template("""
28
  Answer the questions based on the provided context only.