26 lines
767 B
Python
26 lines
767 B
Python
|
import os
|
||
|
|
from enum import Enum
|
||
|
|
from pathlib import Path
|
||
|
|
from dotenv import load_dotenv
|
||
|
|
|
||
|
|
load_dotenv()
|
||
|
|
root_path = Path(".")
|
||
|
|
|
||
|
|
data_path = root_path / "data"
|
||
|
|
rag_path = data_path / "rag"
|
||
|
|
rag_path.mkdir(exist_ok=True, parents=True)
|
||
|
|
constraint_path = rag_path / "constraints.pkl"
|
||
|
|
constraint_vector_db_path = rag_path / "constraint_vector.db"
|
||
|
|
problem_descriptions_vector_db_path = rag_path / "problem_descriptions_vector.db"
|
||
|
|
objective_descriptions_vector_db_path = rag_path / "objective_descriptions_vector.db"
|
||
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
||
|
|
|
||
|
|
|
||
|
|
class RAGMode(Enum):
|
||
|
|
PROBLEM_DESCRIPTION = "problem_description"
|
||
|
|
CONSTRAINT_OR_OBJECTIVE = "constraint_or_objective"
|
||
|
|
PROBLEM_LABELS = "problem_labels"
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.value
|