Update src/streamlit_app.py
Browse files- 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 |
-
|
| 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=
|
| 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(
|