tyang4 commited on
Commit
523b285
·
verified ·
1 Parent(s): d6ab3b4

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -25,7 +25,7 @@ import requests
25
  NEO4J_URI = os.getenv("NEO4J_URI")
26
  NEO4J_USERNAME = os.getenv("NEO4J_USERNAME")
27
  NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
28
- openai_api_key = os.getenv("OPENAI_API_KEY")
29
 
30
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
31
  # ============================== DOWNLOAD ==============================
@@ -266,7 +266,7 @@ Return your response strictly as a **JSON object** with the following fields:
266
 
267
  Do not include any explanation or commentary—only return the JSON object.
268
  """
269
- client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
270
  response = client.chat.completions.create(
271
  model="gpt-4o",
272
  messages=[{"role": "user", "content": prompt}],
 
25
  NEO4J_URI = os.getenv("NEO4J_URI")
26
  NEO4J_USERNAME = os.getenv("NEO4J_USERNAME")
27
  NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
28
+ openai_api_key = os.getenv("openai_api_key")
29
 
30
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
31
  # ============================== DOWNLOAD ==============================
 
266
 
267
  Do not include any explanation or commentary—only return the JSON object.
268
  """
269
+ client = openai.OpenAI(api_key=os.getenv("openai_api_key"))
270
  response = client.chat.completions.create(
271
  model="gpt-4o",
272
  messages=[{"role": "user", "content": prompt}],