Ifeanyi commited on
Commit
b6c8d68
·
verified ·
1 Parent(s): af10061

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +2 -1
retriever.py CHANGED
@@ -1,9 +1,10 @@
1
  from PyPDF2 import PdfReader
2
  from google import genai
3
  import time
 
4
 
5
  # set up authentication
6
- api_key = userdata.get("GEMINI_API_KEY")
7
  client = genai.Client(api_key=api_key)
8
 
9
  # read in documents
 
1
  from PyPDF2 import PdfReader
2
  from google import genai
3
  import time
4
+ import os
5
 
6
  # set up authentication
7
+ api_key = os.getenv("GEMINI_API_KEY")
8
  client = genai.Client(api_key=api_key)
9
 
10
  # read in documents