ZarinT commited on
Commit
a099ec5
·
verified ·
1 Parent(s): d93c7e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ def load_environment():
31
  os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.environ["HF_TOKEN"]
32
  if "GOOGLE_API_KEY" not in os.environ:
33
  raise ValueError("GOOGLE_API_KEY not found in environment variables.")
34
- genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
35
 
36
  class GeminiLLM(Runnable):
37
  def __init__(self, model_name="models/gemini-1.5-pro-latest", api_key=None):
@@ -55,7 +55,7 @@ class GeminiLLM(Runnable):
55
 
56
  class GeminiEmbeddings(Embeddings):
57
  def __init__(self, model_name="models/embedding-001", api_key=None):
58
- self.api_key = "AIzaSyB7qS5OEjRDh0WJqAoKlqPmVus1dU-uWOw" or os.environ["GOOGLE_API_KEY"]
59
  if not api_key:
60
  raise ValueError("GOOGLE_API_KEY not found in environment variables.")
61
  os.environ["GOOGLE_API_KEY"] = api_key
 
31
  os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.environ["HF_TOKEN"]
32
  if "GOOGLE_API_KEY" not in os.environ:
33
  raise ValueError("GOOGLE_API_KEY not found in environment variables.")
34
+ genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
35
 
36
  class GeminiLLM(Runnable):
37
  def __init__(self, model_name="models/gemini-1.5-pro-latest", api_key=None):
 
55
 
56
  class GeminiEmbeddings(Embeddings):
57
  def __init__(self, model_name="models/embedding-001", api_key=None):
58
+ self.api_key = api_key or os.environ["GOOGLE_API_KEY"]
59
  if not api_key:
60
  raise ValueError("GOOGLE_API_KEY not found in environment variables.")
61
  os.environ["GOOGLE_API_KEY"] = api_key