tyang4 commited on
Commit
d6ab3b4
·
verified ·
1 Parent(s): 4522dd0

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
- api_key = os.getenv("OPENAI_API_KEY")
29
 
30
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
31
  # ============================== DOWNLOAD ==============================
@@ -62,7 +62,7 @@ def create_driver():
62
 
63
  driver = create_driver()
64
  # ============================== SIMPLE GPT HELPER ==============================
65
- openai_client = openai.OpenAI(api_key=api_key)
66
 
67
  def gpt_chat(sys_msg: str, user_msg: str, **kwargs):
68
  rsp = openai_client.chat.completions.create(
 
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 ==============================
 
62
 
63
  driver = create_driver()
64
  # ============================== SIMPLE GPT HELPER ==============================
65
+ openai_client = openai.OpenAI(api_key=openai_api_key)
66
 
67
  def gpt_chat(sys_msg: str, user_msg: str, **kwargs):
68
  rsp = openai_client.chat.completions.create(