Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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.
|