vprzybylo commited on
Commit
b3d6567
·
1 Parent(s): 0badb2c

Refactor get_secrets function to simplify environment variable retrieval

Browse files
Files changed (1) hide show
  1. 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
- if os.environ.get("SPACE_ID"):
33
- # We're in a Hugging Face Space
34
- return {
35
- "OPENAI_API_KEY": st.secrets["OPENAI_API_KEY"],
36
- "TAVILY_API_KEY": st.secrets.get("TAVILY_API_KEY"),
37
- "LANGCHAIN_API_KEY": st.secrets.get("LANGCHAIN_API_KEY"),
38
- "LANGCHAIN_PROJECT": st.secrets.get("LANGCHAIN_PROJECT", "GridGuide"),
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