tykiww commited on
Commit
951ec68
1 Parent(s): c0982f8

Update utilities/setup.py

Browse files
Files changed (1) hide show
  1. utilities/setup.py +10 -1
utilities/setup.py CHANGED
@@ -18,4 +18,13 @@ class get_files:
18
  config_path = os.getenv('CONFIG_PATH')
19
  with open(config_path, 'r') as file:
20
  config = json.load(file)
21
- return config
 
 
 
 
 
 
 
 
 
 
18
  config_path = os.getenv('CONFIG_PATH')
19
  with open(config_path, 'r') as file:
20
  config = json.load(file)
21
+ return config
22
+
23
+ def get_keys():
24
+ "Get keys required to set up loading"
25
+ return {
26
+ "pinecone": os.environ["PINECONE_TOKEN"],
27
+ "huggingface": os.environ["HUGGINGFACE_TOKEN"],
28
+ "openai": "", # leaving this alone for future use.
29
+ }
30
+