Spaces:
Runtime error
Runtime error
File size: 2,483 Bytes
40babb9 9a4251f 40babb9 66c3046 40babb9 9843d03 e23c4ea 40babb9 e23c4ea e6651df e23c4ea a1fbe22 e23c4ea 37e48bb 9a4251f c5478e8 9a4251f 9843d03 9cfa4f1 9843d03 e23c4ea 04cad44 40babb9 04cad44 40babb9 04cad44 40babb9 04cad44 40babb9 04cad44 40babb9 04cad44 9a4251f 37e48bb 9a4251f d0b1aef | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | import os
import tempfile
from dotenv import load_dotenv
#******************************************************************************************************************************************************************
class envConfiguration:
ready = False;
def __init__ (self):
load_dotenv()
#sets the values into local env
def set_config(self, openai_key, openai_model, openai_temp, pinecone_key, pinecone_env, pinecone_index):
os.environ["OPEN_AI_KEY"] = openai_key
os.environ["OPEN_AI_MODEL"] = openai_model
os.environ["OPEN_AI_TEMP"] = str(openai_temp)
os.environ["PINECONE_KEY"] = pinecone_key
os.environ["PINECONE_ENV"] = pinecone_env
os.environ["PINECONE_IND"] = pinecone_index
os.environ["INPUT_DIR"] = os.path.join(tempfile.gettempdir(),
'rkk-document-gpt',
'input'
)
os.environ["TEMPLATE_DIR"] = os.path.join(tempfile.gettempdir(),
'rkk-document-gpt',
'template'
)
os.environ["OUTPUT_DIR"] = os.path.join(tempfile.gettempdir(),
'rkk-document-gpt',
'output'
)
os.makedirs(os.environ.get("INPUT_DIR"), exist_ok = True)
os.makedirs(os.environ.get("TEMPLATE_DIR"), exist_ok = True)
os.makedirs(os.environ.get("OUTPUT_DIR"), exist_ok = True)
self.ready = True
return None
def openai_key(self):
return os.environ.get("OPEN_AI_KEY")
def openai_model(self):
return os.environ.get("OPEN_AI_MODEL")
def openai_temp(self):
return os.environ.get("OPEN_AI_TEMP")
def pinecone_key(self):
return os.environ.get("PINECONE_KEY")
def pinecone_env(self):
return os.environ.get("PINECONE_ENV")
def pinecone_index(self):
return os.environ.get("PINECONE_IND")
def input_dir(self):
return os.environ.get("INPUT_DIR")
def template_dir(self):
return os.environ.get("TEMPLATE_DIR")
def output_dir(self):
return os.environ.get("OUTPUT_DIR")
def is_ready(self):
return self.ready |