opencode-semantic-memory 0.1.0__py3-none-any.whl

opencode_memory/config.py

@@ -0,0 +1,89 @@
+"""Configuration management for opencode-memory."""
+
+from pathlib import Path
+from typing import Any
+
+import tomli
+from pydantic import BaseModel, Field
+
+
+class IdentityConfig(BaseModel):
+    """Configuration for user identity."""
+
+    user: str | None = None  # Auto-detected from git if not set
+    instance: str = "gitlab.com"
+    primary_project: str | None = None  # Auto-detected from git if not set
+
+
+class BootConfig(BaseModel):
+    """Configuration for boot context."""
+
+    identity: bool = True
+    active_sessions: bool = True
+    hot_items: bool = True
+    recent_decisions: bool = False
+    unresolved_blockers: bool = True
+    max_hot_items: int = 5
+    recent_decision_days: int = 3
+
+
+class IngestionConfig(BaseModel):
+    """Configuration for data ingestion."""
+
+    watch_paths: list[str] = Field(
+        default_factory=lambda: [
+            "~/.local/share/opencode/opencode.db",
+        ]
+    )
+    db_poll_interval: int = 30
+    # Enable/disable automatic session ingestion (pattern-based, no LLM cost)
+    enabled: bool = True
+    # Enable/disable LLM-based knowledge extraction from old conversations
+    # WARNING: This calls 'opencode run' for each conversation, which uses your
+    # LLM API and may incur significant costs. Runs every 6 hours on up to 50
+    # conversations per cycle.
+    llm_extraction: bool = False  # Off by default due to cost
+    working_directory: str | None = None  # cwd for opencode during LLM extraction
+
+
+class StorageConfig(BaseModel):
+    """Configuration for data storage."""
+
+    path: str = "~/.local/share/opencode-memory"
+
+
+class Config(BaseModel):
+    """Main configuration."""
+
+    identity: IdentityConfig = Field(default_factory=IdentityConfig)
+    boot: BootConfig = Field(default_factory=BootConfig)
+    ingestion: IngestionConfig = Field(default_factory=IngestionConfig)
+    storage: StorageConfig = Field(default_factory=StorageConfig)
+
+    @classmethod
+    def load(cls, config_path: Path | None = None) -> "Config":
+        """Load configuration from file or use defaults."""
+        if config_path is None:
+            config_path = Path.home() / ".config" / "opencode-memory" / "config.toml"
+
+        if config_path.exists():
+            with open(config_path, "rb") as f:
+                data: dict[str, Any] = tomli.load(f)
+            return cls.model_validate(data)
+
+        return cls()
+
+    @property
+    def storage_path(self) -> Path:
+        """Get expanded storage path."""
+        return Path(self.storage.path).expanduser()
+
+    @property
+    def db_path(self) -> Path:
+        """Get path to SQLite database."""
+        return self.storage_path / "memory.db"
+
+    @property
+    def vectors_path(self) -> Path:
+        """Get path to vector store."""
+        return self.storage_path / "vectors"