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