Upload rml_ai/config.py with huggingface_hub

rml_ai/config.py

@@ -0,0 +1,64 @@
+"""
+Configuration for RML System
+"""
+
+import os
+from typing import Optional
+from dataclasses import dataclass
+
+
+@dataclass
+class RMLConfig:
+    """Configuration for RML system"""
+    
+    # Model paths
+    encoder_model: str = "intfloat/e5-base-v2"
+    decoder_model: str = "microsoft/phi-1_5"
+    
+    # Device configuration
+    device: str = "auto"
+    
+    # Dataset configuration
+    dataset_path: str = "data/rml_data.jsonl"
+    max_entries: int = 1000
+    
+    # Encoding configuration
+    encoder_batch_size: int = 8
+    encoder_max_length: int = 192
+    
+    # Feature flags
+    disable_web_search: bool = True
+    disable_world_knowledge: bool = True
+    
+    def __post_init__(self):
+        """Load configuration from environment variables"""
+        self.encoder_model = os.getenv("RML_ENCODER_MODEL", self.encoder_model)
+        self.decoder_model = os.getenv("RML_DECODER_MODEL", self.decoder_model)
+        self.device = os.getenv("RML_DEVICE", self.device)
+        self.dataset_path = os.getenv("RML_DATASET_PATH", self.dataset_path)
+        self.max_entries = int(os.getenv("RML_API_ENTRIES", self.max_entries))
+        self.encoder_batch_size = int(os.getenv("RML_ENCODER_BATCH_SIZE", self.encoder_batch_size))
+        self.encoder_max_length = int(os.getenv("RML_ENCODER_MAX_LEN", self.encoder_max_length))
+        self.disable_web_search = os.getenv("RML_DISABLE_WEB_SEARCH", "1") == "1"
+        self.disable_world_knowledge = os.getenv("RML_DISABLE_WORLD_KNOWLEDGE", "1") == "1"
+    
+    def to_dict(self) -> dict:
+        """Convert config to dictionary"""
+        return {
+            "encoder_model": self.encoder_model,
+            "decoder_model": self.decoder_model,
+            "device": self.device,
+            "dataset_path": self.dataset_path,
+            "max_entries": self.max_entries,
+            "encoder_batch_size": self.encoder_batch_size,
+            "encoder_max_length": self.encoder_max_length,
+            "disable_web_search": self.disable_web_search,
+            "disable_world_knowledge": self.disable_world_knowledge,
+        }
+    
+    def __str__(self) -> str:
+        """String representation of config"""
+        config_str = "RML Configuration:\n"
+        for key, value in self.to_dict().items():
+            config_str += f"  {key}: {value}\n"
+        return config_str