mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
"""Configuration management for basic-memory."""
|
|
|
|
from pathlib import Path
|
|
|
|
from pydantic import Field, field_validator
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
DATABASE_NAME = "memory.db"
|
|
KNOWLEDGE_DIR_NAME = "knowledge"
|
|
DOCUMENTS_DIR_NAME = "documents"
|
|
DATA_DIR_NAME = "data"
|
|
|
|
|
|
class ProjectConfig(BaseSettings):
|
|
"""Configuration for a specific basic-memory project."""
|
|
|
|
# Default to ~/.basic-memory but allow override with env var
|
|
home: Path = Field(
|
|
default_factory=lambda: Path.home() / ".basic-memory",
|
|
description="Base path for basic-memory files",
|
|
)
|
|
|
|
model_config = SettingsConfigDict(
|
|
env_prefix="BASIC_MEMORY_",
|
|
extra="ignore",
|
|
env_file=".env",
|
|
env_file_encoding="utf-8",
|
|
)
|
|
|
|
@property
|
|
def knowledge_dir(self) -> Path:
|
|
"""Get knowledge directory path."""
|
|
return self.home / KNOWLEDGE_DIR_NAME
|
|
|
|
@property
|
|
def documents_dir(self) -> Path:
|
|
"""Get documents directory path."""
|
|
return self.home / DOCUMENTS_DIR_NAME
|
|
|
|
@property
|
|
def database_path(self) -> Path:
|
|
"""Get SQLite database path."""
|
|
return self.home / DATA_DIR_NAME / DATABASE_NAME
|
|
|
|
@field_validator("home")
|
|
@classmethod
|
|
def ensure_path_exists(cls, v: Path) -> Path:
|
|
"""Ensure project path exists."""
|
|
if not v.exists():
|
|
v.mkdir(parents=True)
|
|
return v
|
|
|
|
|
|
# Load project config
|
|
config = ProjectConfig()
|