giovannefeitosa's picture
Initial commit
f7db77c
raw
history blame
716 Bytes
import os
class Configs:
def __init__(self):
# environment variables
os.environ["PROJECT_ROOT"] = os.getcwd()
# openai
self.OPENAI_KEY = ""
self.chatCompletionModel = "gpt-3.5-turbo"
self.embeddingsModel = "text-embedding-ada-002"
# generated files
self.generatedDatasetPath = f"{os.environ['PROJECT_ROOT']}/io/generated/dataset.json"
self.generatedEmbeddingsPath = f"{os.environ['PROJECT_ROOT']}/io/generated/embeddings.json"
# spacy
self.spacyModel = 'en_core_web_sm'
# model
self.generatedModelPath = f"{os.environ['PROJECT_ROOT']}/io/generated/model.sklearn"
configs = Configs()