explicity pass the GOOGle-api key to llm and embedding model to resolve the 503 Illegal metadata and timeout errors

app.py

@@ -11,6 +11,11 @@ from langchain_core.runnables import RunnablePassthrough
 from langchain_core.output_parsers import StrOutputParser
 import tempfile
 
+# Set the Google API key from environment variables
+if "GOOGLE_API_KEY" not in os.environ:
+    raise Exception("Please set the GOOGLE_API_KEY environment variable.")
+
+google_api_key = os.environ.get("GOOGLE_API_KEY")
 
 # Constants
 LLM_MODEL = "gemini-1.5-flash"
@@ -50,7 +55,7 @@ class PDFChatbot:
             docs = text_splitter.create_documents([text])
             print("Text split into chunks successfully.")
 
-            embeddings = GoogleGenerativeAIEmbeddings(model=EMBEDDING_MODEL)
+            embeddings = GoogleGenerativeAIEmbeddings(model=EMBEDDING_MODEL, google_api_key=google_api_key)
             self.state.db = await Chroma.afrom_documents(
                 documents=docs,
                 embedding=embeddings,
@@ -74,8 +79,8 @@ class PDFChatbot:
             return
 
         print("Database is ready. Retrieving relevant documents...")
-        retriever = self.state.db.as_retriever()
-        llm = ChatGoogleGenerativeAI(model=LLM_MODEL, temperature=0.7)
+        retriever = self.state.db.as_ retriever()
+        llm = ChatGoogleGenerativeAI(model=LLM_MODEL, temperature=0.7, google_api_key=google_api_key)
 
         prompt_template = PromptTemplate(
             template="""
@@ -110,10 +115,6 @@ class SessionState:
     def is_db_ready(self):
         return self.db is not None
 
-# Set the Google API key from environment variables
-if "GOOGLE_API_KEY" not in os.environ:
-    raise Exception("Please set the GOOGLE_API_KEY environment variable.")
-
 with gr.Blocks(title="PDF Chatbot") as demo:
     chatbot = PDFChatbot()