From b1ea2a791524fb157662dc8989183c528020e467 Mon Sep 17 00:00:00 2001
From: Ryan Melvin <rmelvin@uabmc.edu>
Date: Thu, 4 May 2023 16:26:43 -0500
Subject: [PATCH] docker secerets for openai api
---
app.py | 16 ++++++++++++++++
docker-compose.yml | 12 ------------
2 files changed, 16 insertions(+), 12 deletions(-)
diff --git a/app.py b/app.py
index 518544d..728e704 100644
--- a/app.py
+++ b/app.py
@@ -9,6 +9,22 @@ from langchain.prompts import (
SystemMessagePromptTemplate,
HumanMessagePromptTemplate,
)
+import os
+
+def manage_sensitive(name):
+ secret_fpath = f'/run/secrets/{name}'
+ existence = os.path.exists(secret_fpath)
+
+ if existence:
+ v2 = open(secret_fpath).read().rstrip('\n')
+ return v2
+
+ if not existence:
+ return KeyError(f'{name}')
+
+OPENAI_API_KEY = manage_sensitive("openai_api_key")
+os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+
st.set_page_config(
page_title="MPOG Helper ",
page_icon="🤖",
diff --git a/docker-compose.yml b/docker-compose.yml
index ae09531..d98a770 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -10,19 +10,7 @@ services:
environment:
- PYTHONUNBUFFERED=1
secrets:
- - db_server
- - db_name
- - db_user
- - db_password
- openai_api_key
secrets:
- db_server:
- file: secrets/db_server.txt
- db_name:
- file: secrets/db_name.txt
- db_user:
- file: secrets/db_user.txt
- db_password:
- file: secrets/db_password.txt
openai_api_key:
file: secrets/openai_api_key.txt
--
GitLab