traigent 0.21.1.dev0__tar.gz → 0.21.3.dev0__tar.gz

{traigent-0.21.1.dev0/traigent.egg-info → traigent-0.21.3.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traigent
-Version: 0.21.1.dev0
+Version: 0.21.3.dev0
 Summary: Enterprise-grade LLM optimization platform with advanced analytics and AI-powered insights
 Author-email: Traigent Team <opensource@traigent.ai>
 License-Expression: AGPL-3.0-only OR LicenseRef-Traigent-Commercial

{traigent-0.21.1.dev0 → traigent-0.21.3.dev0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "traigent"
-version = "0.21.1.dev0"
+version = "0.21.3.dev0"
 authors = [
     {name = "Traigent Team", email = "opensource@traigent.ai"},
 ]

{traigent-0.21.1.dev0 → traigent-0.21.3.dev0}/traigent/analytics/next_steps.py

@@ -21,6 +21,7 @@ Usage:
 
 from __future__ import annotations
 
+import os
 from typing import Any, cast
 
 from traigent.cloud.auth import _build_api_key_auth_headers
@@ -47,6 +48,21 @@ _REQUIRED_RESPONSE_KEYS = {
     "next_steps",
 }  # matches the contract's required set (next_steps_schema.json)
 
+_GUIDANCE_VARIANT_HEADER = "X-Traigent-Guidance-Variant"
+
+
+def _guidance_variant_headers() -> dict[str, str]:
+    """Build the optional guidance-variant request header.
+
+    Reads ``TRAIGENT_GUIDANCE_VARIANT`` and, when set and non-blank, sends its
+    stripped value to request that guidance variant. No client-side
+    allowlisting is applied beyond stripping; the server allowlists.
+    """
+    variant = os.environ.get("TRAIGENT_GUIDANCE_VARIANT", "").strip()
+    if not variant:
+        return {}
+    return {_GUIDANCE_VARIANT_HEADER: variant}
+
 
 class NextStepsClient:
     """Client for retrieving category-level next-step recommendations.
@@ -139,6 +155,11 @@ class NextStepsClient:
         The backend must include the next-steps feature and expose:
         ``GET /api/v1/analytics/experiments/{experiment_run_id}/next-steps``.
 
+        When the ``TRAIGENT_GUIDANCE_VARIANT`` env var is set to a non-blank
+        value, its stripped value is sent as the ``X-Traigent-Guidance-Variant``
+        request header to request that guidance variant. No client-side
+        allowlisting is applied; the server allowlists.
+
         Args:
             experiment_run_id: Experiment run ID to retrieve recommendations for
 
@@ -146,7 +167,10 @@ class NextStepsClient:
             Dict with the backend next-steps contract payload. The response
             includes a ``caveat`` field that callers should display near the
             recommendations. If the backend includes the optional opaque
-            ``posture`` block, it is returned unchanged.
+            ``posture`` block, it is returned unchanged. If the backend
+            includes the optional ``guidance_meta`` block (served_variant,
+            engine, policy_table_sha, smartopt_version, fallback_reason), it
+            is returned unchanged.
 
         Raises:
             httpx.HTTPError: If request fails. A 404 response is raised as
@@ -156,10 +180,12 @@ class NextStepsClient:
                 next-steps contract keys.
         """
         client = self._get_client()
+        headers = _guidance_variant_headers()
 
         try:
             response = await client.get(
-                f"/api/v1/analytics/experiments/{experiment_run_id}/next-steps"
+                f"/api/v1/analytics/experiments/{experiment_run_id}/next-steps",
+                headers=headers or None,
             )
             response.raise_for_status()
         except httpx.HTTPStatusError as exc:

{traigent-0.21.1.dev0 → traigent-0.21.3.dev0}/traigent/cli/next_steps_command.py

@@ -146,6 +146,12 @@ def _print_next_steps_table(payload: dict[str, Any]) -> None:
 
     console.print(f"\n[bold]Caveat:[/bold] {payload.get('caveat', '')}")
 
+    guidance_meta = payload.get("guidance_meta")
+    if isinstance(guidance_meta, dict):
+        served_variant = guidance_meta.get("served_variant", "")
+        engine = guidance_meta.get("engine", "")
+        console.print(f"guidance: served={served_variant} engine={engine}")
+
 
 def _priority_sort_key(row: dict[str, Any]) -> tuple[int, str]:
     priority = row.get("priority")

{traigent-0.21.1.dev0 → traigent-0.21.3.dev0}/traigent/core/orchestrator.py

@@ -820,6 +820,7 @@ class OptimizationOrchestrator:
         self._provider_consecutive_call_failures = 0
         self._provider_failure_first_error: str | None = None
         self._provider_failure_category: str | None = None
+        self._session_finalized = False
         self._logger: OptimizationLogger | None = None
         self._logger_v2: Any | None = None
         self._logger_facade = LoggerFacade()
@@ -2495,6 +2496,7 @@ class OptimizationOrchestrator:
         )
 
         self._status = OptimizationStatus.RUNNING
+        self._session_finalized = False
         self._start_time = time.time()
         self._successful_trials = 0
         self._failed_trials = 0
@@ -3236,12 +3238,18 @@ class OptimizationOrchestrator:
                     )
                 )
 
-                session_summary = self.backend_session_manager.finalize_session(
-                    session_id,
-                    self._status,
-                    certified_selection=self._build_certified_selection_report(),
-                    session_aggregation=agg_payload,
-                )
+                session_summary = None
+                if not getattr(self, "_session_finalized", False):
+                    certified_selection = self._build_certified_selection_report()
+                    try:
+                        session_summary = self.backend_session_manager.finalize_session(
+                            session_id,
+                            self._status,
+                            certified_selection=certified_selection,
+                            session_aggregation=agg_payload,
+                        )
+                    finally:
+                        self._session_finalized = True
                 self.backend_session_manager.attach_session_metadata(
                     result, session_id, session_summary
                 )
@@ -3343,6 +3351,37 @@ class OptimizationOrchestrator:
             f"total cost: ${cost_status.accumulated_cost_usd:.4f}"
         )
 
+    def _finalize_failed_backend_session(
+        self,
+        session_id: str | None,
+        failure: BaseException,
+    ) -> None:
+        """Best-effort terminal finalize for exception exits."""
+        if (
+            getattr(self, "_session_finalized", False)
+            or not self.backend_session_manager.backend_tracking_enabled
+            or backend_egress_disabled(self.traigent_config)
+            or session_id is None
+        ):
+            return
+
+        terminal_reason = self._stop_reason or type(failure).__name__
+        try:
+            self.backend_session_manager.finalize_session(
+                session_id,
+                OptimizationStatus.FAILED,
+            )
+        except Exception as finalize_error:
+            logger.warning(
+                "Failed to finalize backend session %s as FAILED after %s: %s",
+                session_id,
+                terminal_reason,
+                finalize_error,
+                exc_info=True,
+            )
+        finally:
+            self._session_finalized = True
+
     async def _finalize_user_cancelled_optimization(
         self,
         session_id: str | None,
@@ -3498,14 +3537,20 @@ class OptimizationOrchestrator:
                 session_id, session_span, interrupt
             )
 
-        except OptimizationError:
+        except OptimizationError as e:
             self._status = OptimizationStatus.FAILED
+            if not self._stop_reason:
+                self._stop_reason = "error"
+            self._finalize_failed_backend_session(session_id, e)
             raise
         except Exception as e:
             from traigent.knobs import ResolutionError
 
             self._status = OptimizationStatus.FAILED
+            if not self._stop_reason:
+                self._stop_reason = "error"
             logger.error(f"Optimization {self._optimization_id} failed: {e}")
+            self._finalize_failed_backend_session(session_id, e)
             if isinstance(e, ResolutionError):
                 # RFC 0001 §3.4: the typed fail-closed governance rejection
                 # IS the public contract — callers distinguish a stale

{traigent-0.21.1.dev0 → traigent-0.21.3.dev0/traigent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traigent
-Version: 0.21.1.dev0
+Version: 0.21.3.dev0
 Summary: Enterprise-grade LLM optimization platform with advanced analytics and AI-powered insights
 Author-email: Traigent Team <opensource@traigent.ai>
 License-Expression: AGPL-3.0-only OR LicenseRef-Traigent-Commercial