wassname/stampy-chat
1
0
Fork 0
mirror of https://github.com/wassname/stampy-chat.git synced 2026-09-10 12:40:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make env var defs fail-fast

Browse Source
This commit is contained in:
henri123lemoine
2023-06-27 07:48:18 -04:00
parent 49e9f08cf5
commit 67f0dafd14
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/dataset/pinecone_db_handler.py
+2 -4
View File
@@ -17,10 +17,8 @@ class PineconeDB:
):
self.index_name = PINECONE_INDEX_NAME
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
PINECONE_ENVIRONMENT = os.getenv("PINECONE_ENVIRONMENT")
assert PINECONE_API_KEY, "PINECONE_API_KEY environment variable not set."
assert PINECONE_ENVIRONMENT, "PINECONE_LOCATION environment variable not set."
PINECONE_API_KEY = os.environ["PINECONE_API_KEY"]
PINECONE_ENVIRONMENT = os.environ["PINECONE_ENVIRONMENT"]
pinecone.init(
api_key = PINECONE_API_KEY,
src/main.py
+1 -1
View File
@@ -8,7 +8,7 @@ else:
raise Exception("'src/.env' not found. Rename the 'src/.env.example' file and fill in values.")
import openai
openai.api_key = os.environ.get('OPENAI_API_KEY')
openai.api_key = os.environ['OPENAI_API_KEY']
import logging
logging.basicConfig(level=logging.INFO)