claude-smart 0.2.47 → 0.2.49

package/plugin/vendor/reflexio/reflexio/server/services/storage/storage_base/governance/_subject_barrier.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+
+from reflexio.models.api_schema.domain.governance import (
+    AuditEvent,
+    PurgeOperation,
+    SubjectWriteBarrier,
+)
+
+
+class SubjectBarrierMixin(ABC):
+    """Backend-neutral subject-erasure-barrier store contract.
+
+    Extracted verbatim from ``_governance.py`` (the SubjectBarrier bucket). The
+    residual ``GovernanceMixin`` ABC stays composed alongside this mixin and
+    holds the remaining rebuild-hide / governance-erase-execution / purge-complete
+    abstract methods.
+    """
+
+    @abstractmethod
+    def begin_subject_erasure_barrier(
+        self, subject_ref: str, purge_id: str
+    ) -> SubjectWriteBarrier:
+        raise NotImplementedError
+
+    @abstractmethod
+    def assert_subject_writable(self, subject_ref: str) -> None:
+        raise NotImplementedError
+
+    @abstractmethod
+    def complete_subject_erasure_barrier_after_empty_check(
+        self, purge_id: str, audit_event: AuditEvent
+    ) -> PurgeOperation:
+        raise NotImplementedError
+
+    @abstractmethod
+    def fail_subject_erasure_barrier(
+        self,
+        subject_ref: str,
+        purge_id: str,
+        error_code: str,
+        error_detail: str,
+    ) -> SubjectWriteBarrier:
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_subject_write_barrier(self, subject_ref: str) -> SubjectWriteBarrier | None:
+        raise NotImplementedError

package/plugin/vendor/reflexio/reflexio/server/services/storage/storage_base/profiles/__init__.py

@@ -0,0 +1,9 @@
+from ._interaction_store import InteractionStoreMixin
+from ._profile_store import ProfileStoreMixin
+from ._search import ProfileSearchMixin
+
+__all__ = [
+    "InteractionStoreMixin",
+    "ProfileSearchMixin",
+    "ProfileStoreMixin",
+]

package/plugin/vendor/reflexio/reflexio/server/services/storage/storage_base/profiles/_interaction_store.py

@@ -0,0 +1,95 @@
+from abc import abstractmethod
+
+from reflexio.models.api_schema.domain import (
+    DeleteUserInteractionRequest,
+    Interaction,
+)
+
+
+class InteractionStoreMixin:
+    """Mixin for interaction-store CRUD methods."""
+
+    @abstractmethod
+    def get_all_interactions(self, limit: int = 100) -> list[Interaction]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_user_interaction(self, user_id: str) -> list[Interaction]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_all_user_ids(self) -> list[str]:
+        """Return distinct user IDs that have stored interactions."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def add_user_interaction(self, user_id: str, interaction: Interaction) -> None:
+        raise NotImplementedError
+
+    @abstractmethod
+    def add_user_interactions_bulk(
+        self,
+        user_id: str,
+        interactions: list[Interaction],
+        *,
+        embeddings_prepared: bool = False,
+    ) -> None:
+        """Add multiple user interactions with batched embedding generation.
+
+        Args:
+            user_id: The user ID
+            interactions: List of interactions to add
+            embeddings_prepared: When True, skip all embedding generation and write
+                whatever embedding is already on each interaction (real or ``[]``).
+                Use on the durable path after ``prepare_interaction_embeddings`` to
+                ensure no network I/O occurs inside an open ``commit_scope``.
+        """
+        raise NotImplementedError
+
+    def prepare_interaction_embeddings(self, interactions: list[Interaction]) -> None:  # noqa: ARG002
+        """Pre-populate interaction.embedding for each interaction without writing to storage.
+
+        Call this before opening a commit_scope to avoid a network round-trip inside
+        the transaction.  Subclasses that generate embeddings client-side MUST override
+        this to call get_embeddings and populate interaction.embedding before the scope
+        is opened; backends where the embedding is generated server-side can leave this
+        as a no-op.
+
+        Args:
+            interactions: Interactions whose embedding fields will be populated in-place.
+        """
+        return
+
+    @abstractmethod
+    def delete_user_interaction(self, request: DeleteUserInteractionRequest) -> None:
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_all_interactions_for_user(self, user_id: str) -> None:
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_all_interactions(self) -> None:
+        """Delete all interactions across all users."""
+        raise NotImplementedError
+
+    @abstractmethod
+    def count_all_interactions(self) -> int:
+        """Count total interactions across all users.
+
+        Returns:
+            int: Total number of interactions
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_oldest_interactions(self, count: int) -> int:
+        """Delete the oldest N interactions based on created_at timestamp.
+
+        Args:
+            count (int): Number of oldest interactions to delete
+
+        Returns:
+            int: Number of interactions actually deleted
+        """
+        raise NotImplementedError

package/plugin/vendor/reflexio/reflexio/server/services/storage/storage_base/{_profiles.py → profiles/_profile_store.py}

@@ -1,22 +1,15 @@
 from abc import abstractmethod
 
 from reflexio.models.api_schema.domain import (
-    DeleteUserInteractionRequest,
     DeleteUserProfileRequest,
-    Interaction,
     Status,
     UserProfile,
 )
-from reflexio.models.api_schema.retriever_schema import (
-    SearchInteractionRequest,
-    SearchUserProfileRequest,
-)
 
 
-class ProfileMixin:
-    """Mixin for profile and interaction CRUD methods."""
+class ProfileStoreMixin:
+    """Mixin for profile-store CRUD methods."""
 
-    # read methods
     @abstractmethod
     def get_all_profiles(
         self,
@@ -32,10 +25,6 @@ class ProfileMixin:
     ) -> list[UserProfile]:
         raise NotImplementedError
 
-    @abstractmethod
-    def get_all_interactions(self, limit: int = 100) -> list[Interaction]:
-        raise NotImplementedError
-
     @abstractmethod
     def get_user_profile(
         self,
@@ -48,38 +37,37 @@ class ProfileMixin:
         profile_time_to_live: str | None = None,
         start_time: int | None = None,
         end_time: int | None = None,
+        include_expired: bool = False,
     ) -> list[UserProfile]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def get_user_interaction(self, user_id: str) -> list[Interaction]:
-        raise NotImplementedError
-
-    # create or update methods
-    @abstractmethod
-    def add_user_profile(self, user_id: str, user_profiles: list[UserProfile]) -> None:
-        """Add the user profile for a given user id."""
-        raise NotImplementedError
-
-    @abstractmethod
-    def add_user_interaction(self, user_id: str, interaction: Interaction) -> None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def add_user_interactions_bulk(
-        self, user_id: str, interactions: list[Interaction]
-    ) -> None:
-        """Add multiple user interactions with batched embedding generation.
+        """Return profiles for a user, optionally including expired rows.
 
         Args:
-            user_id: The user ID
-            interactions: List of interactions to add
+            user_id: The user whose profiles to return.
+            status_filter: Statuses to include (``None`` element = CURRENT).
+                Defaults to ``[None]`` (CURRENT only) when not supplied.
+            tags: If provided, only return profiles whose ``tags`` overlap.
+            profile_id: If provided, filter to this exact profile id.
+            query: Free-text substring match across content, profile_id, user_id.
+            source: If provided, filter to this exact source value.
+            profile_time_to_live: If provided, filter to this TTL value.
+            start_time: If provided, lower bound on last_modified_timestamp.
+            end_time: If provided, upper bound on last_modified_timestamp.
+            include_expired: When ``False`` (default), rows whose
+                ``expiration_timestamp`` is in the past are excluded
+                (byte-for-byte prior behaviour — every existing caller is
+                unaffected).  Pass ``True`` to omit the expiry filter so
+                EXPIRED tombstones (``expiration_timestamp < now``) are
+                returned.  Required by ``clear_user_data`` to reach every
+                profile a user owns regardless of TTL state.
+
+        Returns:
+            list[UserProfile]: Matching profiles in unspecified order.
         """
         raise NotImplementedError
 
-    # delete methods
     @abstractmethod
-    def delete_user_interaction(self, request: DeleteUserInteractionRequest) -> None:
+    def add_user_profile(self, user_id: str, user_profiles: list[UserProfile]) -> None:
+        """Add the user profile for a given user id."""
         raise NotImplementedError
 
     @abstractmethod
@@ -99,10 +87,6 @@ class ProfileMixin:
         """Replace only the tags of a profile, leaving content and embedding untouched."""
         raise NotImplementedError
 
-    @abstractmethod
-    def delete_all_interactions_for_user(self, user_id: str) -> None:
-        raise NotImplementedError
-
     @abstractmethod
     def delete_all_profiles_for_user(self, user_id: str) -> None:
         raise NotImplementedError
@@ -112,20 +96,6 @@ class ProfileMixin:
         """Delete all profiles across all users."""
         raise NotImplementedError
 
-    @abstractmethod
-    def delete_all_interactions(self) -> None:
-        """Delete all interactions across all users."""
-        raise NotImplementedError
-
-    @abstractmethod
-    def count_all_interactions(self) -> int:
-        """Count total interactions across all users.
-
-        Returns:
-            int: Total number of interactions
-        """
-        raise NotImplementedError
-
     @abstractmethod
     def count_all_profiles(self) -> int:
         """Count total profiles across all users without hydrating rows.
@@ -140,17 +110,27 @@ class ProfileMixin:
         """
         raise NotImplementedError
 
-    @abstractmethod
-    def delete_oldest_interactions(self, count: int) -> int:
-        """Delete the oldest N interactions based on created_at timestamp.
+    def count_user_profiles_by_status(
+        self, user_ids: list[str], status: Status | None
+    ) -> int:
+        """Count non-expired profiles for concrete users with the given status.
+
+        Storage backends may override this with a single query. The default keeps
+        the storage contract backward-compatible for out-of-tree backends.
 
         Args:
-            count (int): Number of oldest interactions to delete
+            user_ids: Concrete user IDs to include. Empty list returns 0.
+            status: Profile status to match; None means CURRENT.
 
         Returns:
-            int: Number of interactions actually deleted
+            int: Number of matching profiles
         """
-        raise NotImplementedError
+        if not user_ids:
+            return 0
+        return sum(
+            len(self.get_user_profile(user_id=user_id, status_filter=[status]))
+            for user_id in user_ids
+        )
 
     @abstractmethod
     def update_all_profiles_status(
@@ -171,6 +151,24 @@ class ProfileMixin:
         """
         raise NotImplementedError
 
+    @abstractmethod
+    def expire_active_profiles(self, *, now: int, limit: int = 1000) -> int:
+        """Tombstone active profiles whose TTL has elapsed.
+
+        Selects ``status IS NULL AND expiration_timestamp < now`` and transitions
+        each to ``status=EXPIRED, retired_at=now``, emitting a ``status_change``
+        lineage event (reason ``ttl-expired``). Returns the number tombstoned.
+
+        Args:
+            now: Current epoch timestamp (seconds). Profiles with
+                ``expiration_timestamp < now`` are tombstoned.
+            limit: Maximum number of profiles to tombstone in one call (default 1000).
+
+        Returns:
+            int: Number of profiles tombstoned.
+        """
+        raise NotImplementedError
+
     @abstractmethod
     def get_profiles_by_ids(
         self,
@@ -363,21 +361,3 @@ class ProfileMixin:
                 commit-atomic legacy change-log "removed" entry.
         """
         raise NotImplementedError
-
-    # Search methods
-    @abstractmethod
-    def search_interaction(
-        self,
-        search_interaction_request: SearchInteractionRequest,
-        query_embedding: list[float] | None = None,
-    ) -> list[Interaction]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def search_user_profile(
-        self,
-        search_user_profile_request: SearchUserProfileRequest,
-        status_filter: list[Status | None] | None = None,
-        query_embedding: list[float] | None = None,
-    ) -> list[UserProfile]:
-        raise NotImplementedError

package/plugin/vendor/reflexio/reflexio/server/services/storage/storage_base/profiles/_search.py

@@ -0,0 +1,32 @@
+from abc import abstractmethod
+
+from reflexio.models.api_schema.domain import (
+    Interaction,
+    Status,
+    UserProfile,
+)
+from reflexio.models.api_schema.retriever_schema import (
+    SearchInteractionRequest,
+    SearchUserProfileRequest,
+)
+
+
+class ProfileSearchMixin:
+    """Mixin for interaction + profile search methods."""
+
+    @abstractmethod
+    def search_interaction(
+        self,
+        search_interaction_request: SearchInteractionRequest,
+        query_embedding: list[float] | None = None,
+    ) -> list[Interaction]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def search_user_profile(
+        self,
+        search_user_profile_request: SearchUserProfileRequest,
+        status_filter: list[Status | None] | None = None,
+        query_embedding: list[float] | None = None,
+    ) -> list[UserProfile]:
+        raise NotImplementedError

package/plugin/vendor/reflexio/reflexio/server/services/storage/storage_base/_governance.py

@@ -1,148 +0,0 @@
-from __future__ import annotations
-
-from abc import ABC, abstractmethod
-from typing import Literal
-
-from reflexio.models.api_schema.domain.governance import (
-    AuditEvent,
-    PurgeOperation,
-    PurgeOperationTarget,
-    SubjectWriteBarrier,
-)
-from reflexio.models.config_schema import GovernanceRetentionConfig
-
-
-class GovernanceMixin(ABC):
-    """Mixin for backend-neutral governance storage primitives."""
-
-    @abstractmethod
-    def append_audit_event(self, event: AuditEvent) -> bool:
-        raise NotImplementedError
-
-    @abstractmethod
-    def list_audit_events(
-        self, subject_ref: str | None = None, *, org_id: str | None = None
-    ) -> list[AuditEvent]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def begin_purge_operation(
-        self,
-        purge_id: str,
-        idempotency_key: str,
-        operation_type: Literal["user_erasure", "org_purge"],
-        scope_type: Literal["user", "org"],
-        subject_ref: str | None,
-        request_ref: str,
-    ) -> PurgeOperation:
-        raise NotImplementedError
-
-    @abstractmethod
-    def record_purge_target(
-        self,
-        purge_id: str,
-        target_name: str,
-        phase: str,
-        status: Literal["pending", "running", "failed", "complete"],
-        target_ref: str = "",
-        detail: dict[str, object] | None = None,
-        deleted_count: int = 0,
-        error_detail: str | None = None,
-    ) -> None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def list_purge_targets(
-        self, purge_id: str, phase: str | None = None
-    ) -> list[PurgeOperationTarget]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def purge_targets_prepared(self, purge_id: str) -> bool:
-        raise NotImplementedError
-
-    @abstractmethod
-    def prepare_governance_erase_targets(
-        self,
-        purge_id: str,
-        user_id: str,
-        owned_user_playbook_ids: set[int] | None = None,
-    ) -> None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def hide_governance_agent_playbooks_for_rebuild(self, purge_id: str) -> list[int]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def apply_governance_user_data_delete(
-        self, purge_id: str, user_id: str
-    ) -> dict[str, int]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def apply_governance_agent_playbook_rebuild(
-        self,
-        purge_id: str,
-        agent_playbook_id: int,
-        remaining_source_windows: list[dict[str, object]],
-        content: str | None,
-        trigger: str | None,
-        rationale: str | None,
-        blocking_issue: dict[str, object] | None,
-        expanded_terms: str | None,
-        tags: list[str] | None,
-    ) -> None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def complete_purge_operation_with_audit(
-        self, purge_id: str, audit_event: AuditEvent
-    ) -> PurgeOperation:
-        raise NotImplementedError
-
-    @abstractmethod
-    def begin_subject_erasure_barrier(
-        self, subject_ref: str, purge_id: str
-    ) -> SubjectWriteBarrier:
-        raise NotImplementedError
-
-    @abstractmethod
-    def assert_subject_writable(self, subject_ref: str) -> None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def complete_subject_erasure_barrier_after_empty_check(
-        self, purge_id: str, audit_event: AuditEvent
-    ) -> PurgeOperation:
-        raise NotImplementedError
-
-    @abstractmethod
-    def fail_subject_erasure_barrier(
-        self,
-        subject_ref: str,
-        purge_id: str,
-        error_code: str,
-        error_detail: str,
-    ) -> SubjectWriteBarrier:
-        raise NotImplementedError
-
-    @abstractmethod
-    def get_subject_write_barrier(
-        self, subject_ref: str
-    ) -> SubjectWriteBarrier | None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def fail_purge_operation(
-        self, purge_id: str, error_code: str, error_detail: str
-    ) -> PurgeOperation:
-        raise NotImplementedError
-
-    @abstractmethod
-    def get_purge_operation(self, purge_id: str) -> PurgeOperation:
-        raise NotImplementedError
-
-    @abstractmethod
-    def gc_governance_retention(self, *, config: GovernanceRetentionConfig) -> int:
-        raise NotImplementedError