1 files changed, 21 insertions, 0 deletions
diff --git a/src/config.py b/src/config.py
new file mode 100644
index 0000000..6258a65
--- /dev/null
+++ b/src/config.py
@@ -0,0 +1,21 @@
+from dataclasses import dataclass
+from src.utils import LLM_Model
+@dataclass
+class LinearRAGConfig:
+    dataset_name: str
+    embedding_model: str = "all-mpnet-base-v2"
+    llm_model: LLM_Model = None
+    chunk_token_size: int = 1000
+    chunk_overlap_token_size: int = 100
+    spacy_model: str = "en_core_web_trf"
+    working_dir: str = "./import"
+    batch_size: int = 128
+    max_workers: int = 16
+    retrieval_top_k: int = 5
+    max_iterations: int = 3
+    top_k_sentence: int = 2
+    passage_ratio: float = 1.5
+    passage_node_weight: float = 0.05
+    damping: float = 0.5
+    initial_threshold: float = 0.5
+    iteration_threshold: float = 0.5
+\ No newline at end of file