claude-smart 0.2.45 → 0.2.47

package/plugin/vendor/reflexio/reflexio/client/client.py

@@ -12,6 +12,7 @@ from urllib.parse import urljoin
 
 import aiohttp
 import requests
+from pydantic import ConfigDict
 
 from reflexio.defaults import DEFAULT_AGENT_VERSION
 from reflexio.models.api_schema.retriever_schema import (
@@ -119,6 +120,11 @@ from reflexio.models.config_schema import Config
 
 from .cache import InMemoryCache
 
+
+class _ClientConfigPayload(Config):
+    model_config = ConfigDict(extra="allow")
+
+
 T = TypeVar("T")
 
 logger = logging.getLogger(__name__)
@@ -150,12 +156,12 @@ class ReflexioClient:
 
         Args:
             api_key (str): API key for authentication. Falls back to REFLEXIO_API_KEY env var.
-            url_endpoint (str): Base URL for the API. Falls back to REFLEXIO_API_URL env var,
+            url_endpoint (str): Base URL for the API. Falls back to REFLEXIO_URL env var,
                 then to the default backend URL.
             timeout (int): Default request timeout in seconds (default 300)
         """
         self.base_url = (
-            url_endpoint or os.environ.get("REFLEXIO_API_URL", "") or BACKEND_URL
+            url_endpoint or os.environ.get("REFLEXIO_URL", "") or BACKEND_URL
         )
         self.api_key = api_key or os.environ.get("REFLEXIO_API_KEY", "")
         self.timeout = timeout
@@ -391,8 +397,8 @@ class ReflexioClient:
         wait_for_response: bool = False,
         skip_aggregation: bool = False,
         force_extraction: bool = False,
+        evaluation_only: bool = False,
         override_learning_stall: bool = False,
-        metadata: dict[str, Any] | None = None,
     ) -> PublishUserInteractionResponse:
         """Publish user interactions.
 
@@ -418,7 +424,7 @@ class ReflexioClient:
             interactions: List of interaction data.
             source: The source of the interaction.
             agent_version: The agent version.
-            session_id: Optional session ID for grouping requests.
+            session_id: Required non-empty session ID for grouping requests.
             wait_for_response: If True, the **server** waits for
                 extraction to complete before returning (longer HTTP
                 call, response includes real profile/playbook counts).
@@ -430,15 +436,13 @@ class ReflexioClient:
             force_extraction: If True, bypass all extraction gates
                 (stride_size, cheap pre-filter, LLM should_run) and
                 always run extractors.
+            evaluation_only: If True, store the interaction for
+                session-level evaluation only and permanently exclude it
+                from profile/playbook extraction.
             override_learning_stall: If True, run extraction even when
                 Reflexio has recorded a provider auth/billing stall. Keep
                 this False for automatic hook publishes; use it only for an
                 explicit retry after reauth or limit reset.
-            metadata: Optional per-request annotations stamped by the
-                customer. Mirrored onto the stored ``Request`` row so the
-                eval pipeline can read them back. Conventional key:
-                ``reflexio_retrieval_enabled`` (bool) for F2 group
-                assignment. Values must be JSON-encodable.
 
         Returns:
             PublishUserInteractionResponse: Server response. In
@@ -448,6 +452,9 @@ class ReflexioClient:
                 mode it includes request_id, storage routing, and
                 deltas.
         """
+        if session_id is None or not session_id.strip():
+            raise ValueError("session_id is required and cannot be empty")
+
         interaction_data_list = [
             (
                 InteractionData(**interaction_request)
@@ -464,8 +471,8 @@ class ReflexioClient:
             agent_version=agent_version,
             skip_aggregation=skip_aggregation,
             force_extraction=force_extraction,
+            evaluation_only=evaluation_only,
             override_learning_stall=override_learning_stall,
-            metadata=metadata or {},
         )
         result = self._publish_interaction_sync(
             request, wait_for_response=wait_for_response
@@ -534,6 +541,7 @@ class ReflexioClient:
         source: str | None = None,
         custom_feature: str | None = None,
         extractor_name: str | None = None,
+        tags: list[str] | None = None,
         threshold: float | None = None,
         enable_reformulation: bool | None = None,
         search_mode: SearchMode | None = None,
@@ -551,6 +559,7 @@ class ReflexioClient:
             source (Optional[str]): Filter by source
             custom_feature (Optional[str]): Filter by custom feature
             extractor_name (Optional[str]): Deprecated compatibility field. Accepted but ignored.
+            tags (Optional[list[str]]): Match profiles having any of these tags.
             threshold (Optional[float]): Similarity threshold (default: 0.7)
             enable_reformulation (Optional[bool]): Enable LLM query reformulation (default: False)
 
@@ -569,6 +578,7 @@ class ReflexioClient:
             source=source,
             custom_feature=custom_feature,
             extractor_name=extractor_name,
+            tags=tags,
             threshold=threshold,
             enable_reformulation=enable_reformulation,
             search_mode=search_mode,
@@ -592,6 +602,7 @@ class ReflexioClient:
         source: str | None = None,
         custom_feature: str | None = None,
         extractor_name: str | None = None,
+        tags: list[str] | None = None,
         threshold: float | None = None,
         enable_reformulation: bool | None = None,
         search_mode: SearchMode | None = None,
@@ -614,6 +625,7 @@ class ReflexioClient:
             source=source,
             custom_feature=custom_feature,
             extractor_name=extractor_name,
+            tags=tags,
             threshold=threshold,
             enable_reformulation=enable_reformulation,
             search_mode=search_mode,
@@ -690,6 +702,7 @@ class ReflexioClient:
         start_time: datetime | None = None,
         end_time: datetime | None = None,
         status_filter: list[Status | None] | None = None,
+        tags: list[str] | None = None,
         top_k: int | None = None,
         threshold: float | None = None,
         enable_reformulation: bool | None = None,
@@ -706,6 +719,7 @@ class ReflexioClient:
             start_time (Optional[datetime]): Start time for created_at filter
             end_time (Optional[datetime]): End time for created_at filter
             status_filter (Optional[list[Optional[Status]]]): Filter by status (None for CURRENT, PENDING, ARCHIVED)
+            tags (Optional[list[str]]): Match playbooks having any of these tags.
             top_k (Optional[int]): Maximum number of results to return (default: 10)
             threshold (Optional[float]): Similarity threshold for vector search (default: 0.4)
             enable_reformulation (Optional[bool]): Enable LLM query reformulation (default: False)
@@ -723,6 +737,7 @@ class ReflexioClient:
             start_time=start_time,
             end_time=end_time,
             status_filter=status_filter,
+            tags=tags,
             top_k=top_k,
             threshold=threshold,
             enable_reformulation=enable_reformulation,
@@ -744,6 +759,7 @@ class ReflexioClient:
         end_time: datetime | None = None,
         status_filter: list[Status | None] | None = None,
         playbook_status_filter: PlaybookStatus | None = None,
+        tags: list[str] | None = None,
         top_k: int | None = None,
         threshold: float | None = None,
         enable_reformulation: bool | None = None,
@@ -760,6 +776,7 @@ class ReflexioClient:
             end_time (Optional[datetime]): End time for created_at filter
             status_filter (Optional[list[Optional[Status]]]): Filter by status (None for CURRENT, PENDING, ARCHIVED)
             playbook_status_filter (Optional[PlaybookStatus]): Filter by playbook status (PENDING, APPROVED, REJECTED)
+            tags (Optional[list[str]]): Match playbooks having any of these tags.
             top_k (Optional[int]): Maximum number of results to return (default: 10)
             threshold (Optional[float]): Similarity threshold for vector search (default: 0.4)
             enable_reformulation (Optional[bool]): Enable LLM query reformulation (default: False)
@@ -777,6 +794,7 @@ class ReflexioClient:
             end_time=end_time,
             status_filter=status_filter,
             playbook_status_filter=playbook_status_filter,
+            tags=tags,
             top_k=top_k,
             threshold=threshold,
             enable_reformulation=enable_reformulation,
@@ -1136,10 +1154,15 @@ class ReflexioClient:
         force_refresh: bool = False,
         *,
         user_id: str | None = None,
+        profile_id: str | None = None,
+        query: str | None = None,
         start_time: datetime | None = None,
         end_time: datetime | None = None,
         top_k: int | None = None,
+        source: str | None = None,
+        profile_time_to_live: str | None = None,
         status_filter: list[Status | str | None] | None = None,
+        tags: list[str] | None = None,
     ) -> GetProfilesViewResponse:
         """Get user profiles.
 
@@ -1147,10 +1170,15 @@ class ReflexioClient:
             request (Optional[GetUserProfilesRequest]): The list request object (alternative to kwargs)
             force_refresh (bool, optional): If True, bypass cache and fetch fresh data. Defaults to False.
             user_id (str): The user ID to get profiles for
+            profile_id (Optional[str]): Exact profile ID filter.
+            query (Optional[str]): Case-insensitive text filter across visible fields.
             start_time (Optional[datetime]): Filter by start time
             end_time (Optional[datetime]): Filter by end time
             top_k (Optional[int]): Maximum number of results to return (default: 30)
+            source (Optional[str]): Filter by profile source.
+            profile_time_to_live (Optional[str]): Filter by profile TTL value.
             status_filter (Optional[list[Optional[Union[Status, str]]]]): Filter by profile status. Accepts Status enum or string values (e.g., "archived", "pending").
+            tags (Optional[list[str]]): Match profiles having any of these tags.
 
         Returns:
             GetProfilesViewResponse: Response containing list of profiles
@@ -1171,10 +1199,15 @@ class ReflexioClient:
             request,
             GetUserProfilesRequest,
             user_id=user_id,
+            profile_id=profile_id,
+            query=query,
             start_time=start_time,
             end_time=end_time,
             top_k=top_k,
+            source=source,
+            profile_time_to_live=profile_time_to_live,
             status_filter=converted_status_filter,
+            tags=tags,
         )
 
         # Check cache if not forcing refresh
@@ -1182,10 +1215,15 @@ class ReflexioClient:
             cached_result = self._cache.get(
                 "get_profiles",
                 user_id=req.user_id,
+                profile_id=req.profile_id,
+                query=req.query,
                 start_time=req.start_time,
                 end_time=req.end_time,
                 top_k=req.top_k,
+                source=req.source,
+                profile_time_to_live=req.profile_time_to_live,
                 status_filter=req.status_filter,
+                tags=req.tags,
             )
             if cached_result is not None:
                 return cached_result
@@ -1203,10 +1241,15 @@ class ReflexioClient:
             "get_profiles",
             result,
             user_id=req.user_id,
+            profile_id=req.profile_id,
+            query=req.query,
             start_time=req.start_time,
             end_time=req.end_time,
             top_k=req.top_k,
+            source=req.source,
+            profile_time_to_live=req.profile_time_to_live,
             status_filter=req.status_filter,
+            tags=req.tags,
         )
 
         return result
@@ -1233,6 +1276,13 @@ class ReflexioClient:
         self,
         limit: int = 100,
         status_filter: str | 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,
     ) -> GetProfilesViewResponse:
         """Get all user profiles across all users.
 
@@ -1241,6 +1291,13 @@ class ReflexioClient:
             status_filter (str, optional): Filter by profile status. Accepts
                 ``"current"``, ``"pending"``, or ``"archived"``. If ``None``
                 (the default), profiles with any status are returned.
+            user_id (str, optional): Filter by exact user ID.
+            profile_id (str, optional): Filter by exact profile ID.
+            query (str, optional): Case-insensitive text filter across visible fields.
+            source (str, optional): Filter by exact profile source.
+            profile_time_to_live (str, optional): Filter by profile TTL value.
+            start_time (int, optional): Minimum last-modified epoch seconds.
+            end_time (int, optional): Maximum last-modified epoch seconds.
 
         Returns:
             GetProfilesViewResponse: Response containing all user profiles
@@ -1250,6 +1307,21 @@ class ReflexioClient:
         params: dict[str, str | int] = {"limit": limit}
         if status_filter:
             params["status_filter"] = status_filter
+        params.update(
+            {
+                key: value
+                for key, value in {
+                    "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,
+                }.items()
+                if value is not None
+            }
+        )
         response = self._make_request(
             "GET",
             f"/api/get_all_profiles?{urlencode(params)}",
@@ -1265,7 +1337,10 @@ class ReflexioClient:
         Returns:
             dict: Response containing success status and message
         """
-        config = self._convert_to_model(config, Config)  # type: ignore[reportAssignmentType]
+        config = self._convert_to_model(  # type: ignore[reportAssignmentType]
+            config,
+            _ClientConfigPayload,
+        )
         return self._make_request(
             "POST",
             "/api/set_config",
@@ -1287,7 +1362,7 @@ class ReflexioClient:
 
         Args:
             partial: Top-level fields to overlay on the existing config.
-                Unknown keys are dropped server-side by Pydantic.
+                Unknown keys are rejected server-side.
 
         Returns:
             dict: ``{"success": bool, "msg": str}`` from the server.
@@ -1311,7 +1386,7 @@ class ReflexioClient:
             "GET",
             "/api/get_config",
         )
-        return Config(**response)
+        return _ClientConfigPayload(**response)
 
     def invalidate_cache(self, org_id: str | None = None) -> dict:
         """Explicitly evict the server-side per-org Reflexio cache entry.
@@ -1349,20 +1424,32 @@ class ReflexioClient:
         request: GetUserPlaybooksRequest | dict | None = None,
         *,
         limit: int | None = None,
+        user_playbook_id: int | None = None,
         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,
     ) -> GetUserPlaybooksViewResponse:
         """Get user playbooks.
 
         Args:
             request (Optional[GetUserPlaybooksRequest]): The get request object (alternative to kwargs)
             limit (Optional[int]): Maximum number of results to return (default: 100)
+            user_playbook_id (Optional[int]): Exact user playbook ID filter.
             user_id (Optional[str]): Filter by user ID
+            request_id (Optional[str]): Filter by generating request ID.
+            query (Optional[str]): Case-insensitive text filter across visible fields.
             playbook_name (Optional[str]): Filter by playbook name
             agent_version (Optional[str]): Filter by agent version
+            start_time (Optional[datetime]): Filter by start time.
+            end_time (Optional[datetime]): Filter by end time.
             status_filter (Optional[list[Optional[Status]]]): Filter by status
+            tags (Optional[list[str]]): Match playbooks having any of these tags.
 
         Returns:
             GetUserPlaybooksViewResponse: Response containing user playbooks
@@ -1371,10 +1458,16 @@ class ReflexioClient:
             request,
             GetUserPlaybooksRequest,
             limit=limit,
+            user_playbook_id=user_playbook_id,
             user_id=user_id,
+            request_id=request_id,
+            query=query,
             playbook_name=playbook_name,
             agent_version=agent_version,
+            start_time=start_time,
+            end_time=end_time,
             status_filter=status_filter,
+            tags=tags,
         )
         response = self._make_request(
             "POST",
@@ -1624,10 +1717,15 @@ class ReflexioClient:
         force_refresh: bool = False,
         *,
         limit: int | None = None,
+        agent_playbook_id: int | 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,
         playbook_status_filter: PlaybookStatus | None = None,
+        tags: list[str] | None = None,
     ) -> GetAgentPlaybooksViewResponse:
         """Get agent playbooks.
 
@@ -1635,10 +1733,15 @@ class ReflexioClient:
             request (Optional[GetAgentPlaybooksRequest]): The get request object (alternative to kwargs)
             force_refresh (bool, optional): If True, bypass cache and fetch fresh data. Defaults to False.
             limit (Optional[int]): Maximum number of results to return (default: 100)
+            agent_playbook_id (Optional[int]): Exact agent playbook ID filter.
+            query (Optional[str]): Case-insensitive text filter across visible fields.
             playbook_name (Optional[str]): Filter by playbook name
             agent_version (Optional[str]): Filter by agent version
+            start_time (Optional[datetime]): Filter by start time.
+            end_time (Optional[datetime]): Filter by end time.
             status_filter (Optional[list[Optional[Status]]]): Filter by status
             playbook_status_filter (Optional[PlaybookStatus]): Filter by playbook status (default: APPROVED)
+            tags (Optional[list[str]]): Match playbooks having any of these tags.
 
         Returns:
             GetAgentPlaybooksViewResponse: Response containing agent playbooks
@@ -1647,10 +1750,15 @@ class ReflexioClient:
             request,
             GetAgentPlaybooksRequest,
             limit=limit,
+            agent_playbook_id=agent_playbook_id,
+            query=query,
             playbook_name=playbook_name,
             agent_version=agent_version,
+            start_time=start_time,
+            end_time=end_time,
             status_filter=status_filter,
             playbook_status_filter=playbook_status_filter,
+            tags=tags,
         )
 
         # Check cache if not forcing refresh
@@ -1658,9 +1766,15 @@ class ReflexioClient:
             cached_result = self._cache.get(
                 "get_agent_playbooks",
                 limit=req.limit,
+                agent_playbook_id=req.agent_playbook_id,
+                query=req.query,
                 playbook_name=req.playbook_name,
+                agent_version=req.agent_version,
+                start_time=req.start_time,
+                end_time=req.end_time,
                 status_filter=req.status_filter,
                 playbook_status_filter=req.playbook_status_filter,
+                tags=req.tags,
             )
             if cached_result is not None:
                 return cached_result
@@ -1678,9 +1792,15 @@ class ReflexioClient:
             "get_agent_playbooks",
             result,
             limit=req.limit,
+            agent_playbook_id=req.agent_playbook_id,
+            query=req.query,
             playbook_name=req.playbook_name,
+            agent_version=req.agent_version,
+            start_time=req.start_time,
+            end_time=req.end_time,
             status_filter=req.status_filter,
             playbook_status_filter=req.playbook_status_filter,
+            tags=req.tags,
         )
 
         return result
@@ -1691,6 +1811,8 @@ class ReflexioClient:
         *,
         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 = None,
@@ -1701,6 +1823,8 @@ class ReflexioClient:
             request (Optional[GetRequestsRequest]): The get request object (alternative to kwargs)
             user_id (Optional[str]): Filter by user ID
             request_id (Optional[str]): Filter by request ID
+            session_id (Optional[str]): Filter by session ID
+            source (Optional[str]): Filter by request source
             start_time (Optional[datetime]): Filter by start time
             end_time (Optional[datetime]): Filter by end time
             top_k (Optional[int]): Maximum number of results to return (default: 30)
@@ -1713,6 +1837,8 @@ class ReflexioClient:
             GetRequestsRequest,
             user_id=user_id,
             request_id=request_id,
+            session_id=session_id,
+            source=source,
             start_time=start_time,
             end_time=end_time,
             top_k=top_k,
@@ -1730,6 +1856,8 @@ class ReflexioClient:
         *,
         limit: int | None = None,
         agent_version: str | None = None,
+        start_time: datetime | None = None,
+        end_time: datetime | None = None,
     ) -> GetEvaluationResultsViewResponse:
         """Get agent success evaluation results.
 
@@ -1737,6 +1865,8 @@ class ReflexioClient:
             request (Optional[GetAgentSuccessEvaluationResultsRequest]): The get request object (alternative to kwargs)
             limit (Optional[int]): Maximum number of results to return (default: 100)
             agent_version (Optional[str]): Filter by agent version
+            start_time (Optional[datetime]): Filter by start time
+            end_time (Optional[datetime]): Filter by end time
 
         Returns:
             GetEvaluationResultsViewResponse: Response containing agent success evaluation results
@@ -1746,6 +1876,8 @@ class ReflexioClient:
             GetAgentSuccessEvaluationResultsRequest,
             limit=limit,
             agent_version=agent_version,
+            start_time=start_time,
+            end_time=end_time,
         )
         response = self._make_request(
             "POST",
@@ -2212,6 +2344,9 @@ class ReflexioClient:
         enable_agent_answer: bool | None = None,
         conversation_history: list[ConversationTurn] | list[dict] | None = None,
         search_mode: SearchMode | str | None = None,
+        request_id: str | None = None,
+        session_id: str | None = None,
+        interaction_id: int | None = None,
     ) -> UnifiedSearchViewResponse:
         """Search across all entity types (profiles, agent playbooks, user playbooks).
 
@@ -2235,6 +2370,9 @@ class ReflexioClient:
                 the configured search backend supports it (default: False).
             conversation_history (Optional[list[ConversationTurn] | list[dict]]): Prior conversation turns for context-aware query reformulation. Accepts ConversationTurn objects or dicts with "role" and "content" keys.
             search_mode (Optional[SearchMode | str]): Search mode to use. Accepts SearchMode enum or string value ("vector", "fts", "hybrid").
+            request_id (Optional[str]): Caller correlation id for the search turn.
+            session_id (Optional[str]): Caller session id for the search turn.
+            interaction_id (Optional[int]): Caller interaction id for the search turn.
 
         Returns:
             UnifiedSearchViewResponse: Combined search results from all entity types
@@ -2254,6 +2392,9 @@ class ReflexioClient:
             enable_agent_answer=enable_agent_answer,
             conversation_history=conversation_history,
             search_mode=search_mode,
+            request_id=request_id,
+            session_id=session_id,
+            interaction_id=interaction_id,
         )
         response = self._make_request("POST", "/api/search", json=req.model_dump())
         return UnifiedSearchViewResponse(**response)