Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
vprzybylo
commited on
Commit
·
b3d6567
1
Parent(s):
0badb2c
Refactor get_secrets function to simplify environment variable retrieval
Browse files- app/src/ui/app.py +7 -18
app/src/ui/app.py
CHANGED
@@ -29,24 +29,13 @@ sys.path.append(str(src_path))
|
|
29 |
# Get secrets from Hugging Face Space
|
30 |
def get_secrets():
|
31 |
"""Get secrets from Hugging Face Space or local environment."""
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
"LANGCHAIN_TRACING_V2": st.secrets.get("LANGCHAIN_TRACING_V2", "true"),
|
40 |
-
}
|
41 |
-
else:
|
42 |
-
# We're running locally, use environment variables
|
43 |
-
return {
|
44 |
-
"OPENAI_API_KEY": os.environ.get("OPENAI_API_KEY"),
|
45 |
-
"TAVILY_API_KEY": os.environ.get("TAVILY_API_KEY"),
|
46 |
-
"LANGCHAIN_API_KEY": os.environ.get("LANGCHAIN_API_KEY"),
|
47 |
-
"LANGCHAIN_PROJECT": os.environ.get("LANGCHAIN_PROJECT", "GridGuide"),
|
48 |
-
"LANGCHAIN_TRACING_V2": os.environ.get("LANGCHAIN_TRACING_V2", "true"),
|
49 |
-
}
|
50 |
|
51 |
|
52 |
# Set up environment variables from secrets
|
|
|
29 |
# Get secrets from Hugging Face Space
|
30 |
def get_secrets():
|
31 |
"""Get secrets from Hugging Face Space or local environment."""
|
32 |
+
|
33 |
+
return {
|
34 |
+
"OPENAI_API_KEY": os.getenv("OPENAI_API_KEY"),
|
35 |
+
"LANGCHAIN_API_KEY": os.getenv("LANGCHAIN_API_KEY"),
|
36 |
+
"LANGCHAIN_PROJECT": os.getenv("LANGCHAIN_PROJECT", "grid-code-assistant"),
|
37 |
+
"LANGCHAIN_TRACING_V2": os.getenv("LANGCHAIN_TRACING_V2", "true"),
|
38 |
+
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
|
40 |
|
41 |
# Set up environment variables from secrets
|