MemoryOS 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

memos/configs/mem_user.py

@@ -0,0 +1,58 @@
+from typing import Any, ClassVar
+
+from pydantic import BaseModel, Field, field_validator, model_validator
+
+from memos.configs.base import BaseConfig
+
+
+class BaseUserManagerConfig(BaseConfig):
+    """Base configuration class for user managers."""
+
+    user_id: str = Field(default="root", description="Default user ID for initialization")
+
+
+class SQLiteUserManagerConfig(BaseUserManagerConfig):
+    """SQLite user manager configuration."""
+
+    db_path: str | None = Field(
+        default=None,
+        description="Path to SQLite database file. If None, uses default path in MEMOS_DIR",
+    )
+
+
+class MySQLUserManagerConfig(BaseUserManagerConfig):
+    """MySQL user manager configuration."""
+
+    host: str = Field(default="localhost", description="MySQL server host")
+    port: int = Field(default=3306, description="MySQL server port")
+    username: str = Field(default="root", description="MySQL username")
+    password: str = Field(default="", description="MySQL password")
+    database: str = Field(default="memos_users", description="MySQL database name")
+    charset: str = Field(default="utf8mb4", description="MySQL charset")
+
+
+class UserManagerConfigFactory(BaseModel):
+    """Factory for user manager configurations."""
+
+    backend: str = Field(default="sqlite", description="Backend for user manager")
+    config: dict[str, Any] = Field(
+        default_factory=dict, description="Configuration for the user manager backend"
+    )
+
+    backend_to_class: ClassVar[dict[str, Any]] = {
+        "sqlite": SQLiteUserManagerConfig,
+        "mysql": MySQLUserManagerConfig,
+    }
+
+    @field_validator("backend")
+    @classmethod
+    def validate_backend(cls, backend: str) -> str:
+        if backend not in cls.backend_to_class:
+            raise ValueError(f"Unsupported user manager backend: {backend}")
+        return backend
+
+    @model_validator(mode="after")
+    def instantiate_config(self):
+        config_class = self.backend_to_class[self.backend]
+        self.config = config_class(**self.config)
+        return self

memos/graph_dbs/base.py

@@ -146,7 +146,7 @@ class BaseGraphDB(ABC):
         """
 
     @abstractmethod
-    def get_by_metadata(self, filters: dict[str, Any]) -> list[str]:
+    def get_by_metadata(self, filters: list[dict[str, Any]]) -> list[str]:
         """
         Retrieve node IDs that match given metadata filters.
 
@@ -162,6 +162,14 @@ class BaseGraphDB(ABC):
             - Can be used for faceted recall or prefiltering before embedding rerank.
         """
 
+    @abstractmethod
+    def get_structure_optimization_candidates(self, scope: str) -> list[dict]:
+        """
+        Find nodes that are likely candidates for structure optimization:
+        - Isolated nodes, nodes with empty background, or nodes with exactly one child.
+        - Plus: the child of any parent node that has exactly one child.
+        """
+
     # Structure Maintenance
     @abstractmethod
     def deduplicate_nodes(self) -> None:

memos/graph_dbs/factory.py

@@ -2,6 +2,7 @@ from typing import Any, ClassVar
 
 from memos.configs.graph_db import GraphDBConfigFactory
 from memos.graph_dbs.base import BaseGraphDB
+from memos.graph_dbs.nebular import NebulaGraphDB
 from memos.graph_dbs.neo4j import Neo4jGraphDB
 from memos.graph_dbs.neo4j_community import Neo4jCommunityGraphDB
 
@@ -12,6 +13,7 @@ class GraphStoreFactory(BaseGraphDB):
     backend_to_class: ClassVar[dict[str, Any]] = {
         "neo4j": Neo4jGraphDB,
         "neo4j-community": Neo4jCommunityGraphDB,
+        "nebular": NebulaGraphDB,
     }
 
     @classmethod