Create config_data/config.py

config_data/config.py

@@ -0,0 +1,52 @@
+from dataclasses import dataclass
+import os
+
+
+@dataclass
+class TgBot:
+    token: str
+    admin_ids: list[int]
+
+
+@dataclass
+class DataConfig:
+    dataset: str
+    cls_vec: str
+
+
+@dataclass
+class ModelConfig:
+    bi_checkpoint: str
+    cross_checkpoint: str
+    device: str
+    hf_client: str
+
+
+@dataclass
+class Config:
+    tg_bot: TgBot
+    data: DataConfig
+    model: ModelConfig
+
+
+def load_config(path: str='.env') -> Config:
+
+    env: Env = Env()
+    env.read_env()
+
+    return Config(
+        tg_bot=TgBot(
+            token=os.environ['telegram_token'],
+            admin_ids=os.environ['telegram_admin']
+        ),
+        data=DataConfig(
+            dataset='ekaterinatao/house_md_context3',
+            cls_vec='ekaterinatao/house_md_cls_embeds'
+        ),
+        model=ModelConfig(
+            bi_checkpoint='ekaterinatao/house-md-bot-bert-bi-encoder',
+            cross_checkpoint='ekaterinatao/house-md-bot-bert-cross-encoder',
+            device='cpu',
+            hf_client='ekaterinatao/house_md_bot'
+        )
+    )