jostlebot commited on
Commit
fcf4e18
·
1 Parent(s): 05a8092

Update environment variable name to ANTHROPIC_KEY consistently across test files

Browse files
Files changed (2) hide show
  1. src/minimal_test.py +7 -7
  2. src/test_api.py +1 -1
src/minimal_test.py CHANGED
@@ -13,26 +13,26 @@ st.write("Available environment variables:", [k for k in os.environ.keys()])
13
  api_key = None
14
 
15
  # Try direct environment variable
16
- if 'ANTHROPIC_API_KEY' in os.environ:
17
- api_key = os.environ['ANTHROPIC_API_KEY']
18
- st.success("Found API key in os.environ['ANTHROPIC_API_KEY']")
19
 
20
  # Try getenv
21
  if not api_key:
22
- api_key = os.getenv('ANTHROPIC_API_KEY')
23
  if api_key:
24
  st.success("Found API key using os.getenv()")
25
 
26
  # Try Streamlit secrets
27
- if not api_key and hasattr(st, 'secrets') and 'ANTHROPIC_API_KEY' in st.secrets:
28
- api_key = st.secrets['ANTHROPIC_API_KEY']
29
  st.success("Found API key in Streamlit secrets")
30
 
31
  if not api_key:
32
  st.error("""
33
  ⚠️ No API key found. Please make sure:
34
  1. You've added the secret in Hugging Face Space settings
35
- 2. The secret is named exactly 'ANTHROPIC_API_KEY'
36
  3. The Space has been rebuilt after adding the secret
37
  """)
38
  st.stop()
 
13
  api_key = None
14
 
15
  # Try direct environment variable
16
+ if 'ANTHROPIC_KEY' in os.environ:
17
+ api_key = os.environ['ANTHROPIC_KEY']
18
+ st.success("Found API key in os.environ['ANTHROPIC_KEY']")
19
 
20
  # Try getenv
21
  if not api_key:
22
+ api_key = os.getenv('ANTHROPIC_KEY')
23
  if api_key:
24
  st.success("Found API key using os.getenv()")
25
 
26
  # Try Streamlit secrets
27
+ if not api_key and hasattr(st, 'secrets') and 'ANTHROPIC_KEY' in st.secrets:
28
+ api_key = st.secrets['ANTHROPIC_KEY']
29
  st.success("Found API key in Streamlit secrets")
30
 
31
  if not api_key:
32
  st.error("""
33
  ⚠️ No API key found. Please make sure:
34
  1. You've added the secret in Hugging Face Space settings
35
+ 2. The secret is named exactly 'ANTHROPIC_KEY'
36
  3. The Space has been rebuilt after adding the secret
37
  """)
38
  st.stop()
src/test_api.py CHANGED
@@ -6,7 +6,7 @@ st.title("Anthropic API Key Test")
6
 
7
  try:
8
  # Get API key from environment
9
- api_key = os.getenv('ANTHROPIC_API_KEY')
10
  if not api_key:
11
  st.error("❌ No API key found in environment variables")
12
  st.stop()
 
6
 
7
  try:
8
  # Get API key from environment
9
+ api_key = os.getenv('ANTHROPIC_KEY')
10
  if not api_key:
11
  st.error("❌ No API key found in environment variables")
12
  st.stop()