feat: add method commit_model

Author: scream4ik

memstate/storage.py

@@ -30,6 +30,15 @@ def validate(self, typename: str, payload: dict[str, Any]) -> dict[str, Any]:
         except ValidationError as e:
             raise ValidationFailed(str(e))
 
+    def get_type_by_model(self, model_class: type[BaseModel]) -> str | None:
+        """
+        Reverse lookup: finds the registered type name for a given Pydantic class.
+        """
+        for type_name, cls in self._schemas.items():
+            if cls == model_class:
+                return type_name
+        return None
+
 
 class Constraint:
     def __init__(self, singleton_key: str | None = None, immutable: bool = False) -> None:
@@ -121,6 +130,30 @@ def commit(
 
                 raise e
 
+    def commit_model(
+        self,
+        model: BaseModel,
+        session_id: str | None = None,
+        ephemeral: bool = False,
+        actor: str | None = None,
+        reason: str | None = None,
+    ) -> str:
+        """
+        Commit a Pydantic model instance directly.
+        Auto-detects the schema type from the registry.
+        """
+        schema_type = self._schema_registry.get_type_by_model(model.__class__)
+
+        if not schema_type:
+            raise MemoryStoreError(
+                f"Model class '{model.__class__.__name__}' is not registered. "
+                f"Please call memory.register_schema('your_type_name', {model.__class__.__name__}) first."
+            )
+
+        fact = Fact(type=schema_type, payload=model.model_dump())
+
+        return self.commit(fact, session_id=session_id, ephemeral=ephemeral, actor=actor, reason=reason)
+
     def update(self, fact_id: str, patch: dict[str, Any], actor: str | None = None, reason: str | None = None) -> str:
         with self._lock:
             existing = self.storage.load(fact_id)

tests/test_storage.py

@@ -122,3 +122,27 @@ def test_ephemeral_session_discard(memory):
 
     remaining = memory.query(filters={"session_id": session_id})
     assert len(remaining) == 0
+
+
+def test_commit_model_success(memory):
+    schema_name = "user_v1"
+    memory.register_schema(schema_name, User)
+
+    user = User(name="Survivor", age=50)
+
+    fact_id = memory.commit_model(user, actor="system", session_id="session_1")
+
+    saved_fact = memory.storage.load(fact_id)
+
+    assert saved_fact is not None
+    assert saved_fact["type"] == schema_name
+    assert saved_fact["payload"] == {"name": "Survivor", "age": 50}
+
+    assert saved_fact["session_id"] == "session_1"
+
+
+def test_commit_model_raises_on_unregistered(memory):
+    unknown = User(name="Survivor", age=50)
+
+    with pytest.raises(MemoryStoreError, match="is not registered"):
+        memory.commit_model(unknown)