claude-smart 0.2.46 → 0.2.48

package/plugin/vendor/reflexio/reflexio/lib/_base.py

@@ -127,6 +127,21 @@ class ReflexioBase:
             raise RuntimeError(STORAGE_NOT_CONFIGURED_MSG)
         return storage
 
+    def get_storage(self) -> BaseStorage:
+        """Return the configured storage backend.
+
+        Public accessor for consumers (including enterprise) that need the
+        underlying ``BaseStorage``. Prefer ``request_context.storage`` inside
+        OSS pipeline code; this exists for callers that hold a ``Reflexio``.
+
+        Returns:
+            BaseStorage: The configured storage backend.
+
+        Raises:
+            RuntimeError: If storage is not configured.
+        """
+        return self._get_storage()
+
     def _get_query_reformulator(self) -> QueryReformulator:
         """Lazily create and cache a QueryReformulator instance.
 

package/plugin/vendor/reflexio/reflexio/lib/_config.py

@@ -29,32 +29,37 @@ class ConfigMixin(ReflexioBase):
             # security). Without this, the marker reaches the readiness check as
             # an unfillable StorageConfigManagedSupabase and fails with
             # "Storage configuration is incomplete".
+            current_storage_config = configurator.get_current_storage_configuration()
             storage_config = config.storage_config
             if storage_config is None or isinstance(
                 storage_config, StorageConfigManagedSupabase
             ):
-                storage_config = configurator.get_current_storage_configuration()
+                storage_config = current_storage_config
                 config.storage_config = storage_config
 
-            # Check if storage config is ready to test
-            if not configurator.is_storage_config_ready_to_test(
-                storage_config=storage_config
-            ):
-                return SetConfigResponse(
-                    success=False, msg="Storage configuration is incomplete"
+            storage_config_changed = storage_config != current_storage_config
+            if storage_config_changed or current_storage_config is None:
+                # Storage validation initializes the backend and may run
+                # migrations, so only do it when the storage target changes.
+                if not configurator.is_storage_config_ready_to_test(
+                    storage_config=storage_config
+                ):
+                    return SetConfigResponse(
+                        success=False, msg="Storage configuration is incomplete"
+                    )
+
+                (
+                    success,
+                    error_msg,
+                ) = configurator.test_and_init_storage_config(
+                    storage_config=storage_config
                 )
 
-            # Test and initialize storage connection
-            (
-                success,
-                error_msg,
-            ) = configurator.test_and_init_storage_config(storage_config=storage_config)
-
-            if not success:
-                return SetConfigResponse(
-                    success=False,
-                    msg=f"Failed to validate storage connection: {error_msg}",
-                )
+                if not success:
+                    return SetConfigResponse(
+                        success=False,
+                        msg=f"Failed to validate storage connection: {error_msg}",
+                    )
 
             # Only set config if validation passed
             configurator.set_config(config)

package/plugin/vendor/reflexio/reflexio/lib/_generation.py

@@ -38,22 +38,23 @@ class GenerationMixin(ReflexioBase):
         """
         if not self._is_storage_configured():
             raise ValueError(STORAGE_NOT_CONFIGURED_MSG)
+        from reflexio.server.extensions import get_service
+        from reflexio.server.services.playbook.aggregation_prompt_processing import (
+            AGGREGATION_PROMPT_PROCESSOR,
+        )
         from reflexio.server.services.playbook.components.aggregator import (
             PlaybookAggregator,
         )
         from reflexio.server.services.playbook.playbook_service_utils import (
             PlaybookAggregatorRequest,
         )
-        from reflexio.server.services.playbook.user_detail_stripping import (
-            create_aggregation_user_detail_stripper,
-        )
 
-        user_detail_stripper = create_aggregation_user_detail_stripper(
-            self.request_context.configurator
-        )
+        aggregation_prompt_processor = get_service(AGGREGATION_PROMPT_PROCESSOR)
         aggregator_kwargs = {}
-        if user_detail_stripper is not None:
-            aggregator_kwargs["user_detail_stripper"] = user_detail_stripper
+        if aggregation_prompt_processor is not None:
+            aggregator_kwargs["aggregation_prompt_processor"] = (
+                aggregation_prompt_processor
+            )
 
         playbook_aggregator = PlaybookAggregator(
             llm_client=self.llm_client,

package/plugin/vendor/reflexio/reflexio/lib/_interactions.py

@@ -53,11 +53,21 @@ class InteractionsMixin(ReflexioBase):
     def publish_interaction(
         self,
         request: PublishUserInteractionRequest | dict,
+        *,
+        use_publish_limiter: bool = True,
+        publish_limiter_wait_forever: bool = True,
+        defer_learning: bool = False,
     ) -> PublishUserInteractionResponse:
         """Publish user interactions.
 
         Args:
             request (Union[PublishUserInteractionRequest, dict]): The publish user interaction request
+            use_publish_limiter: Whether GenerationService should throttle the
+                post-write learning pipeline.
+            publish_limiter_wait_forever: Whether GenerationService should queue
+                indefinitely for that post-write learning limiter.
+            defer_learning: Whether to enqueue post-persist learning instead of
+                running it inline.
 
         Returns:
             PublishUserInteractionResponse: Response containing success status and message
@@ -87,7 +97,12 @@ class InteractionsMixin(ReflexioBase):
             # Convert dict to PublishUserInteractionRequest if needed
             if isinstance(request, dict):
                 request = PublishUserInteractionRequest(**request)
-            result = generation_service.run(request)
+            result = generation_service.run(
+                request,
+                use_publish_limiter=use_publish_limiter,
+                publish_limiter_wait_forever=publish_limiter_wait_forever,
+                defer_learning=defer_learning,
+            )
             after_profiles = _safe_count(storage.count_all_profiles)
             after_playbooks = _safe_count(storage.count_user_playbooks)
             # Don't concatenate warnings into the message field — they

package/plugin/vendor/reflexio/reflexio/lib/_profiles.py

@@ -414,26 +414,22 @@ class ProfilesMixin(ReflexioBase):
                 status_filter = [None]  # Default to current profiles
 
         profiles = self._get_storage().get_user_profile(
-            request.user_id, status_filter=status_filter, tags=request.tags
+            request.user_id,
+            status_filter=status_filter,
+            tags=request.tags,
+            profile_id=request.profile_id,
+            query=request.query,
+            source=request.source,
+            profile_time_to_live=request.profile_time_to_live,
+            start_time=(
+                int(request.start_time.timestamp()) if request.start_time else None
+            ),
+            end_time=int(request.end_time.timestamp()) if request.end_time else None,
         )
         profiles = sorted(
             profiles, key=lambda x: x.last_modified_timestamp, reverse=True
         )
 
-        # Apply time filters
-        if request.start_time:
-            profiles = [
-                p
-                for p in profiles
-                if p.last_modified_timestamp >= int(request.start_time.timestamp())
-            ]
-        if request.end_time:
-            profiles = [
-                p
-                for p in profiles
-                if p.last_modified_timestamp <= int(request.end_time.timestamp())
-            ]
-
         # Apply top_k limit
         if request.top_k:
             profiles = sorted(
@@ -450,6 +446,13 @@ class ProfilesMixin(ReflexioBase):
         self,
         limit: int = 100,
         status_filter: list[Status | None] | None = None,
+        user_id: str | None = None,
+        profile_id: str | None = None,
+        query: str | None = None,
+        source: str | None = None,
+        profile_time_to_live: str | None = None,
+        start_time: int | None = None,
+        end_time: int | None = None,
     ) -> GetUserProfilesResponse:
         """Get all user profiles across all users.
 
@@ -467,7 +470,15 @@ class ProfilesMixin(ReflexioBase):
         if status_filter is None:
             status_filter = [None]  # Default to current profiles
         profiles = self._get_storage().get_all_profiles(
-            limit=limit, status_filter=status_filter
+            limit=limit,
+            status_filter=status_filter,
+            user_id=user_id,
+            profile_id=profile_id,
+            query=query,
+            source=source,
+            profile_time_to_live=profile_time_to_live,
+            start_time=start_time,
+            end_time=end_time,
         )
         profiles = sorted(
             profiles, key=lambda x: x.last_modified_timestamp, reverse=True

package/plugin/vendor/reflexio/reflexio/lib/_search.py

@@ -41,9 +41,27 @@ class SearchMixin(ReflexioBase):
             request = GetAgentSuccessEvaluationResultsRequest(**request)
 
         try:
-            results = self._get_storage().get_agent_success_evaluation_results(
-                limit=request.limit or 100, agent_version=request.agent_version
-            )
+            if request.start_time or request.end_time:
+                results = (
+                    self._get_storage().get_agent_success_evaluation_results_in_window(
+                        from_ts=(
+                            int(request.start_time.timestamp())
+                            if request.start_time
+                            else 0
+                        ),
+                        to_ts=(
+                            int(request.end_time.timestamp())
+                            if request.end_time
+                            else 2**31 - 1
+                        ),
+                        limit=request.limit or 100,
+                        agent_version=request.agent_version,
+                    )
+                )
+            else:
+                results = self._get_storage().get_agent_success_evaluation_results(
+                    limit=request.limit or 100, agent_version=request.agent_version
+                )
             return GetAgentSuccessEvaluationResultsResponse(
                 success=True,
                 agent_success_evaluation_results=results,
@@ -79,6 +97,7 @@ class SearchMixin(ReflexioBase):
                 user_id=request.user_id,
                 request_id=request.request_id,
                 session_id=request.session_id,
+                source=request.source,
                 start_time=(
                     int(request.start_time.timestamp()) if request.start_time else None
                 ),
@@ -104,10 +123,11 @@ class SearchMixin(ReflexioBase):
                     Session(session_id=group_name, requests=request_data_list)
                 )
 
-            # Determine has_more: count total requests returned across all groups
+            # Pagination is session-based (top_k counts sessions): there may be
+            # more pages when this page filled the session limit.
             total_returned = sum(len(s.requests) for s in sessions)
             effective_limit = request.top_k or 100
-            has_more = total_returned >= effective_limit
+            has_more = len(sessions) >= effective_limit
 
             return GetRequestsResponse(
                 success=True,
@@ -162,6 +182,7 @@ class SearchMixin(ReflexioBase):
             storage = self._get_storage()
             prompt_manager = self.request_context.prompt_manager
             retrieval_floor = config.retrieval_floor if config else None
+            recency = getattr(storage, "recency", None)
 
         from reflexio.server.services.unified_search_service import run_unified_search
 
@@ -173,4 +194,5 @@ class SearchMixin(ReflexioBase):
             prompt_manager=prompt_manager,
             pre_retrieval_model_name=pre_retrieval_model_name,
             retrieval_floor=retrieval_floor,
+            recency=recency,
         )

package/plugin/vendor/reflexio/reflexio/lib/_user_playbook.py

@@ -53,10 +53,19 @@ class UserPlaybookMixin(ReflexioBase):
         try:
             user_playbooks = self._get_storage().get_user_playbooks(
                 limit=request.limit or 100,
+                user_playbook_id=request.user_playbook_id,
                 user_id=request.user_id,
+                request_id=request.request_id,
+                query=request.query,
                 playbook_name=request.playbook_name,
                 agent_version=request.agent_version,
                 status_filter=request.status_filter,
+                start_time=(
+                    int(request.start_time.timestamp()) if request.start_time else None
+                ),
+                end_time=(
+                    int(request.end_time.timestamp()) if request.end_time else None
+                ),
                 tags=request.tags,
             )
             return GetUserPlaybooksResponse(

package/plugin/vendor/reflexio/reflexio/models/api_schema/domain/__init__.py

@@ -1,3 +1,4 @@
 from ..common import *  # noqa: F401, F403
 from .entities import *  # noqa: F401, F403
 from .enums import *  # noqa: F401, F403
+from .governance import *  # noqa: F401, F403

package/plugin/vendor/reflexio/reflexio/models/api_schema/domain/enums.py

@@ -46,6 +46,7 @@ class Status(str, Enum):  # noqa: UP042 - CURRENT=None is not compatible with St
     SUPERSEDED = (
         "superseded"  # tombstone: replaced by a new version (superseded_by set)
     )
+    EXPIRED = "expired"  # tombstone: TTL elapsed while active (see expiry reclamation)
 
 
 class OperationStatus(StrEnum):

package/plugin/vendor/reflexio/reflexio/models/api_schema/domain/governance.py

@@ -0,0 +1,117 @@
+from __future__ import annotations
+
+from datetime import UTC, datetime
+from typing import Any, Literal
+
+from pydantic import BaseModel, Field
+
+AuditActorType = Literal["api_token", "jwt", "system"]
+AuditOperation = Literal["READ", "EXPORT", "ERASE", "CREATE", "UPDATE", "DELETE"]
+AuditEntityType = Literal[
+    "profile",
+    "user_playbook",
+    "agent_playbook",
+    "interaction",
+    "request",
+    "session",
+    "agent_success_evaluation_result",
+    "playbook_retrieval_log",
+    "org",
+]
+AuditStatus = Literal["ok", "error"]
+PurgeOperationType = Literal["user_erasure", "org_purge"]
+PurgeScopeType = Literal["user", "org"]
+PurgeStatus = Literal["pending", "running", "failed", "complete"]
+PurgeTargetStatus = Literal["pending", "running", "failed", "complete"]
+SubjectBarrierStatus = Literal["erasing", "erased", "failed"]
+
+__all__ = [
+    "AuditActorType",
+    "AuditOperation",
+    "AuditEntityType",
+    "AuditStatus",
+    "PurgeOperationType",
+    "PurgeScopeType",
+    "PurgeStatus",
+    "PurgeTargetStatus",
+    "SubjectBarrierStatus",
+    "AuditEvent",
+    "PurgeOperation",
+    "PurgeOperationTarget",
+    "SubjectWriteBarrier",
+    "UserExportResult",
+    "UserEraseResult",
+]
+
+
+def _now_epoch() -> int:
+    return int(datetime.now(UTC).timestamp())
+
+
+class AuditEvent(BaseModel):
+    org_id: str
+    actor_type: AuditActorType = "system"
+    actor_ref: str | None = None
+    operation: AuditOperation
+    entity_type: AuditEntityType
+    entity_id: str | None = None
+    subject_ref: str | None = None
+    request_ref: str
+    idempotency_key: str | None = None
+    status: AuditStatus = "ok"
+    detail: dict[str, Any] | None = None
+    created_at: int = Field(default_factory=_now_epoch)
+
+
+class PurgeOperation(BaseModel):
+    purge_id: str
+    org_id: str
+    operation_type: PurgeOperationType
+    scope_type: PurgeScopeType
+    subject_ref: str | None = None
+    request_ref: str
+    idempotency_key: str
+    status: PurgeStatus = "pending"
+    error_code: str | None = None
+    error_detail: str | None = None
+    created_at: int = Field(default_factory=_now_epoch)
+    updated_at: int = Field(default_factory=_now_epoch)
+    completed_at: int | None = None
+
+
+class PurgeOperationTarget(BaseModel):
+    purge_id: str
+    target_name: str
+    target_ref: str = ""
+    phase: str
+    status: PurgeTargetStatus = "pending"
+    detail: dict[str, Any] | None = None
+    deleted_count: int = 0
+    error_detail: str | None = None
+    started_at: int | None = None
+    completed_at: int | None = None
+
+
+class SubjectWriteBarrier(BaseModel):
+    org_id: str
+    subject_ref: str
+    purge_id: str
+    status: SubjectBarrierStatus
+    error_code: str | None = None
+    error_detail: str | None = None
+    created_at: int = Field(default_factory=_now_epoch)
+    updated_at: int = Field(default_factory=_now_epoch)
+
+
+class UserExportResult(BaseModel):
+    subject_ref: str
+    export_id: str
+    bundle: dict[str, Any]
+
+
+class UserEraseResult(BaseModel):
+    subject_ref: str
+    purge_id: str
+    status: PurgeStatus
+    deleted_counts: dict[str, int] = Field(default_factory=dict)
+    rebuilt_agent_playbook_ids: list[int] = Field(default_factory=list)

package/plugin/vendor/reflexio/reflexio/models/api_schema/retriever_schema.py

@@ -187,9 +187,13 @@ class GetInteractionsResponse(BaseModel):
 
 class GetUserProfilesRequest(BaseModel):
     user_id: NonEmptyStr
+    profile_id: str | None = None
+    query: str | None = None
     start_time: datetime | None = None
     end_time: datetime | None = None
     top_k: int | None = Field(default=30, gt=0)
+    source: str | None = None
+    profile_time_to_live: str | None = None
     status_filter: list[Status | None] | None = None
     tags: list[str] | None = None
 
@@ -222,12 +226,23 @@ class SetConfigResponse(BaseModel):
 
 class GetUserPlaybooksRequest(BaseModel):
     limit: int | None = Field(default=100, gt=0)
+    user_playbook_id: int | None = Field(default=None, gt=0)
     user_id: str | None = None
+    request_id: str | None = None
+    query: str | None = None
     playbook_name: str | None = None
     agent_version: str | None = None
+    start_time: datetime | None = None
+    end_time: datetime | None = None
     status_filter: list[Status | None] | None = None
     tags: list[str] | None = None
 
+    @model_validator(mode="after")
+    def check_time_range(self) -> Self:
+        """Validate that end_time is after start_time."""
+        TimeRangeValidatorMixin.validate_time_range(self.start_time, self.end_time)
+        return self
+
 
 class GetUserPlaybooksResponse(BaseModel):
     success: bool
@@ -237,8 +252,12 @@ class GetUserPlaybooksResponse(BaseModel):
 
 class GetAgentPlaybooksRequest(BaseModel):
     limit: int | None = Field(default=100, gt=0)
+    agent_playbook_id: int | None = Field(default=None, gt=0)
+    query: str | None = None
     playbook_name: str | None = None
     agent_version: str | None = None
+    start_time: datetime | None = None
+    end_time: datetime | None = None
     status_filter: list[Status | None] | None = None
     playbook_status_filter: PlaybookStatus | None = None
     tags: list[str] | None = None
@@ -247,6 +266,12 @@ class GetAgentPlaybooksRequest(BaseModel):
     request_id: str | None = None
     session_id: str | None = None
 
+    @model_validator(mode="after")
+    def check_time_range(self) -> Self:
+        """Validate that end_time is after start_time."""
+        TimeRangeValidatorMixin.validate_time_range(self.start_time, self.end_time)
+        return self
+
 
 class GetAgentPlaybooksResponse(BaseModel):
     success: bool
@@ -392,6 +417,14 @@ class SearchAgentPlaybookResponse(BaseModel):
 class GetAgentSuccessEvaluationResultsRequest(BaseModel):
     limit: int | None = Field(default=100, gt=0)
     agent_version: str | None = None
+    start_time: datetime | None = None
+    end_time: datetime | None = None
+
+    @model_validator(mode="after")
+    def check_time_range(self) -> Self:
+        """Validate that end_time is after start_time."""
+        TimeRangeValidatorMixin.validate_time_range(self.start_time, self.end_time)
+        return self
 
 
 class GetAgentSuccessEvaluationResultsResponse(BaseModel):
@@ -404,10 +437,18 @@ class GetRequestsRequest(BaseModel):
     user_id: str | None = None
     request_id: str | None = None
     session_id: str | None = None
+    source: str | None = None
     start_time: datetime | None = None
     end_time: datetime | None = None
-    top_k: int | None = Field(default=30, gt=0)
-    offset: int | None = Field(default=0, ge=0)
+    top_k: int | None = Field(
+        default=30,
+        gt=0,
+        description="Maximum number of sessions to return. Pagination is "
+        "per-session: every returned session includes all of its requests.",
+    )
+    offset: int | None = Field(
+        default=0, ge=0, description="Number of sessions to skip for pagination."
+    )
 
     @model_validator(mode="after")
     def check_time_range(self) -> Self:
@@ -497,7 +538,8 @@ class TimeSeriesDataPoint(BaseModel):
     """A single data point in a time series."""
 
     timestamp: int = Field(gt=0)  # Unix timestamp
-    value: int = Field(ge=0)  # Count or metric value
+    value: float = Field(ge=0)  # Count or metric value
+    count: int | None = Field(default=None, ge=0)  # Optional weight for rate metrics
 
 
 class PeriodStats(BaseModel):

package/plugin/vendor/reflexio/reflexio/models/config_schema.py

@@ -708,6 +708,34 @@ class LineageGCConfig(BaseModel):
     poll_interval_seconds: int = Field(default=86400, gt=0)
 
 
+class ExpiryReclamationConfig(BaseModel):
+    """Direct-delete reclamation of expired plain rows (non-audited).
+
+    Independent of ``lineage_gc``: these rows carry no PII/audit/grace obligation,
+    so they can be reclaimed whenever this is enabled even if tombstone GC is off.
+
+    Opt-in by default (``enabled=False``) so operators control a staged rollout
+    of the direct-delete Class B sweeps and are not surprised by deletions on
+    upgrade.  ``lineage_gc.enabled`` (which defaults to True) is unaffected and
+    continues to drive the Class A profile-expiry and tombstone-GC paths.
+    """
+
+    enabled: bool = False
+
+
+class GovernanceRetentionConfig(BaseModel):
+    """Audit-event retention policy. **Enterprise-only:** reclamation is performed
+    by an enterprise per-org reclamation sweep registered via
+    ``register_per_org_sweep``. In an OSS-only deployment these knobs are accepted
+    but inert (the OSS lineage scheduler does not reclaim audit events) and the
+    server logs a startup warning when retention is enabled.
+    """
+
+    audit_events_retention_enabled: bool = False
+    audit_events_retention_days: int = Field(default=365, gt=0)
+    audit_events_delete_batch_limit: int = Field(default=500, gt=0)
+
+
 @dataclass(frozen=True)
 class EffectivePendingToolCallConfig:
     """Resolved pending-tool-call settings after applying tool overrides."""
@@ -847,6 +875,13 @@ class Config(BaseModel):
     )
     # Tombstone GC job gate (opt-in, off by default — see LineageGCConfig)
     lineage_gc: LineageGCConfig = Field(default_factory=LineageGCConfig)
+    # Direct-delete reclamation of expired plain rows (share links, pending tool calls)
+    expiry_reclamation: ExpiryReclamationConfig = Field(
+        default_factory=ExpiryReclamationConfig
+    )
+    governance_retention: GovernanceRetentionConfig = Field(
+        default_factory=GovernanceRetentionConfig
+    )
     # Optional non-blocking async information tools for classic extraction.
     pending_tool_call_config: PendingToolCallConfig = Field(
         default_factory=PendingToolCallConfig
@@ -905,6 +940,8 @@ class Config(BaseModel):
                 "reflection_config",
                 "playbook_optimizer_config",
                 "lineage_gc",
+                "expiry_reclamation",
+                "governance_retention",
                 "pending_tool_call_config",
                 "retrieval_floor",
             ):