ragtime-cli 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

src/config.py

@@ -36,11 +36,19 @@ class CodeConfig:
     ])
 
 
+@dataclass
+class ConventionsConfig:
+    """Configuration for convention checking."""
+    files: list[str] = field(default_factory=lambda: [".ragtime/CONVENTIONS.md"])
+    also_search_memories: bool = True
+
+
 @dataclass
 class RagtimeConfig:
     """Main ragtime configuration."""
     docs: DocsConfig = field(default_factory=DocsConfig)
     code: CodeConfig = field(default_factory=CodeConfig)
+    conventions: ConventionsConfig = field(default_factory=ConventionsConfig)
 
     @classmethod
     def load(cls, project_path: Path) -> "RagtimeConfig":
@@ -58,6 +66,7 @@ class RagtimeConfig:
 
         docs_data = data.get("docs", {})
         code_data = data.get("code", {})
+        conventions_data = data.get("conventions", {})
 
         return cls(
             docs=DocsConfig(
@@ -70,6 +79,12 @@ class RagtimeConfig:
                 languages=code_data.get("languages", CodeConfig().languages),
                 exclude=code_data.get("exclude", CodeConfig().exclude),
             ),
+            conventions=ConventionsConfig(
+                files=conventions_data.get("files", ConventionsConfig().files),
+                also_search_memories=conventions_data.get(
+                    "also_search_memories", ConventionsConfig().also_search_memories
+                ),
+            ),
         )
 
     def save(self, project_path: Path) -> None:
@@ -89,6 +104,10 @@ class RagtimeConfig:
                 "languages": self.code.languages,
                 "exclude": self.code.exclude,
             },
+            "conventions": {
+                "files": self.conventions.files,
+                "also_search_memories": self.conventions.also_search_memories,
+            },
         }
 
         with open(config_path, "w") as f: