File size: 716 Bytes
f7db77c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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()
|