add env vars

src/few_shot_funcs.py

@@ -1,3 +1,4 @@
+import os
 import re
 import openai
 import inflect
@@ -9,12 +10,8 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.vectorstores.utils import DistanceStrategy
 
-import os
-
-OPENAI_KEY = ''
 
-openai.api_key = OPENAI_KEY
-os.environ['OPENAI_API_KEY'] = OPENAI_KEY
+openai.api_key = os.environ.get('OPENAI_KEY')
 
 
 # Constants

src/scrape.py

@@ -1,10 +1,11 @@
 import re
+import os
 import requests
 from base64 import b64decode
 from bs4 import BeautifulSoup
 from typing import Dict
 
-Z_KEY = ''
+Z_KEY = os.environ.get('ZYTE_KEY')
 
 
 def zyte_call(url: str) -> bytes: