Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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 =
|
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
|