claude-smart 0.2.45 → 0.2.46

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,10 @@ 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 +155,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 +396,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 +423,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 +435,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 +451,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 +470,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 +540,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 +558,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 +577,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 +601,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 +624,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 +701,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 +718,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 +736,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 +758,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 +775,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 +793,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,
@@ -1140,6 +1157,7 @@ class ReflexioClient:
         end_time: datetime | None = None,
         top_k: int | None = None,
         status_filter: list[Status | str | None] | None = None,
+        tags: list[str] | None = None,
     ) -> GetProfilesViewResponse:
         """Get user profiles.
 
@@ -1151,6 +1169,7 @@ class ReflexioClient:
             end_time (Optional[datetime]): Filter by end time
             top_k (Optional[int]): Maximum number of results to return (default: 30)
             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
@@ -1175,6 +1194,7 @@ class ReflexioClient:
             end_time=end_time,
             top_k=top_k,
             status_filter=converted_status_filter,
+            tags=tags,
         )
 
         # Check cache if not forcing refresh
@@ -1186,6 +1206,7 @@ class ReflexioClient:
                 end_time=req.end_time,
                 top_k=req.top_k,
                 status_filter=req.status_filter,
+                tags=req.tags,
             )
             if cached_result is not None:
                 return cached_result
@@ -1207,6 +1228,7 @@ class ReflexioClient:
             end_time=req.end_time,
             top_k=req.top_k,
             status_filter=req.status_filter,
+            tags=req.tags,
         )
 
         return result
@@ -1265,7 +1287,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 +1312,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 +1336,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.
@@ -1353,6 +1378,7 @@ class ReflexioClient:
         playbook_name: str | None = None,
         agent_version: str | None = None,
         status_filter: list[Status | None] | None = None,
+        tags: list[str] | None = None,
     ) -> GetUserPlaybooksViewResponse:
         """Get user playbooks.
 
@@ -1363,6 +1389,7 @@ class ReflexioClient:
             playbook_name (Optional[str]): Filter by playbook name
             agent_version (Optional[str]): Filter by agent version
             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
@@ -1375,6 +1402,7 @@ class ReflexioClient:
             playbook_name=playbook_name,
             agent_version=agent_version,
             status_filter=status_filter,
+            tags=tags,
         )
         response = self._make_request(
             "POST",
@@ -1628,6 +1656,7 @@ class ReflexioClient:
         agent_version: str | None = None,
         status_filter: list[Status | None] | None = None,
         playbook_status_filter: PlaybookStatus | None = None,
+        tags: list[str] | None = None,
     ) -> GetAgentPlaybooksViewResponse:
         """Get agent playbooks.
 
@@ -1639,6 +1668,7 @@ class ReflexioClient:
             agent_version (Optional[str]): Filter by agent version
             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
@@ -1651,6 +1681,7 @@ class ReflexioClient:
             agent_version=agent_version,
             status_filter=status_filter,
             playbook_status_filter=playbook_status_filter,
+            tags=tags,
         )
 
         # Check cache if not forcing refresh
@@ -1659,8 +1690,10 @@ class ReflexioClient:
                 "get_agent_playbooks",
                 limit=req.limit,
                 playbook_name=req.playbook_name,
+                agent_version=req.agent_version,
                 status_filter=req.status_filter,
                 playbook_status_filter=req.playbook_status_filter,
+                tags=req.tags,
             )
             if cached_result is not None:
                 return cached_result
@@ -1679,8 +1712,10 @@ class ReflexioClient:
             result,
             limit=req.limit,
             playbook_name=req.playbook_name,
+            agent_version=req.agent_version,
             status_filter=req.status_filter,
             playbook_status_filter=req.playbook_status_filter,
+            tags=req.tags,
         )
 
         return result
@@ -2212,6 +2247,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 +2273,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 +2295,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)