AkashDataScience commited on
Commit
d4462ce
·
1 Parent(s): 518b3a9
Files changed (1) hide show
  1. app.py +2 -1
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