Spaces:
Sleeping
Sleeping
Commit
·
d4462ce
1
Parent(s):
518b3a9
Minor fix
Browse files
app.py
CHANGED
@@ -6,6 +6,7 @@ from langchain_core.prompts import PromptTemplate
|
|
6 |
from langchain_pinecone import PineconeVectorStore
|
7 |
from langchain_community.vectorstores import FAISS
|
8 |
from langchain_community.vectorstores import LanceDB
|
|
|
9 |
from langchain_text_splitters import CharacterTextSplitter
|
10 |
from langchain_community.document_loaders import PyPDFLoader
|
11 |
from langchain_google_genai import GoogleGenerativeAIEmbeddings, GoogleGenerativeAI
|
@@ -23,7 +24,7 @@ prompt_template = """
|
|
23 |
|
24 |
prompt = PromptTemplate(template = prompt_template, input_variables = ["context", "question"])
|
25 |
|
26 |
-
chain = prompt | gemini
|
27 |
|
28 |
index_name = "langchain-test-index"
|
29 |
|
|
|
6 |
from langchain_pinecone import PineconeVectorStore
|
7 |
from langchain_community.vectorstores import FAISS
|
8 |
from langchain_community.vectorstores import LanceDB
|
9 |
+
from google.api_core.exceptions import ResourceExhausted
|
10 |
from langchain_text_splitters import CharacterTextSplitter
|
11 |
from langchain_community.document_loaders import PyPDFLoader
|
12 |
from langchain_google_genai import GoogleGenerativeAIEmbeddings, GoogleGenerativeAI
|
|
|
24 |
|
25 |
prompt = PromptTemplate(template = prompt_template, input_variables = ["context", "question"])
|
26 |
|
27 |
+
chain = prompt | gemini.with_retry(retry_if_exception_type=(ResourceExhausted,))
|
28 |
|
29 |
index_name = "langchain-test-index"
|
30 |
|