centive-sdk 2.2.0.dev6__py3-none-any.whl → 2.2.0.dev8__py3-none-any.whl

centive_sdk/config.py

@@ -83,6 +83,12 @@ class ClientConfig(BaseModel):
     max_session_id_chars: int = 200
     # Concurrent websocket connections accepted per user.
     max_connections_per_user: int = 5
+    # How long the connect-time session trigger waits for the browser SDK to
+    # ask for a session itself before firing. Browser SDKs since 1.0 ask the
+    # moment the socket opens, and their request is the one that carries the
+    # page the user has open; firing first meant two Centive calls and two Anam
+    # tokens per connection. 0 restores the old immediate trigger.
+    session_trigger_grace_seconds: float = 0.5
 
     # Page-telemetry frames (type="telemetry") from the FE SDK. These mirror the
     # aria-sdk producer's own caps (MAX_EVENTS_PER_FRAME, MAX_FRAME_BYTES,

centive_sdk/resources/websocket_server.py

@@ -1,3 +1,4 @@
+import asyncio
 import inspect
 import json
 import secrets
@@ -82,6 +83,14 @@ class WebSocketServer:
         # current_page when the FE's user_trigger frame carries no page of its
         # own (older browser SDKs). Bounded by the number of live connections.
         self._connection_last_page: Dict[WebSocketServerProtocol, dict] = {}
+        # One Centive trigger-session call per connection start, not two: the
+        # connect-time automatic trigger is held back for a grace period and
+        # stands down when the client asked itself (_auto_trigger_after_grace);
+        # a request that lands while another trigger for the same connection is
+        # in flight waits for it and receives the same frame.
+        self._connection_trigger_tasks: Dict[WebSocketServerProtocol, asyncio.Task] = {}
+        self._connection_triggered: set = set()
+        self._connection_trigger_inflight: Dict[WebSocketServerProtocol, asyncio.Future] = {}
         # Every session key touched by a connection, so a disconnect saves all of
         # them rather than only the most recent one.
         self._connection_sessions: Dict[WebSocketServerProtocol, Set[str]] = {}
@@ -569,7 +578,9 @@ class WebSocketServer:
                 {"client": client_info, "user_id": user_id},
             )
 
-            await self._trigger_and_emit(websocket, user_id, None, client_info)
+            self._connection_trigger_tasks[websocket] = asyncio.create_task(
+                self._auto_trigger_after_grace(websocket, user_id, client_info)
+            )
 
         try:
             async for message in websocket:
@@ -599,6 +610,11 @@ class WebSocketServer:
             self._clients.discard(websocket)
             self._connection_users.pop(websocket, None)
             self._connection_last_page.pop(websocket, None)
+            self._connection_triggered.discard(websocket)
+            self._connection_trigger_inflight.pop(websocket, None)
+            pending_trigger = self._connection_trigger_tasks.pop(websocket, None)
+            if pending_trigger is not None:
+                pending_trigger.cancel()
             self._telemetry_rate.forget(websocket)
             session_keys = self._connection_sessions.pop(websocket, set())
 
@@ -1408,10 +1424,47 @@ class WebSocketServer:
         # batcher. The frame's own page wins; the last page_view seen on this
         # connection stands in for browser SDKs that predate the field.
         current_page = self._coerce_page(data.get("page")) or self._connection_last_page.get(websocket)
+        # The client asked itself, so the connect-time automatic trigger, if it
+        # is still waiting out its grace period, has nothing left to do.
+        self._connection_triggered.add(websocket)
         await self._trigger_and_emit(
             websocket, user_id, user_trigger, client_info, current_page=current_page
         )
 
+    async def _auto_trigger_after_grace(
+        self, websocket: WebSocketServerProtocol, user_id: str, client_info: str
+    ) -> None:
+        """The connect-time session trigger, held back briefly.
+
+        Browser SDKs since 1.0 ask for a token themselves the moment the socket
+        opens, and that request is the one that carries the page the user has
+        open. Firing the automatic trigger first (and before the message loop
+        even started reading) meant two Centive calls and two Anam tokens per
+        connection, with the opening prompt built from the page-less one. Now
+        the automatic trigger waits config.session_trigger_grace_seconds and
+        stands down when the client has asked in the meantime. A client that
+        never asks (older SDKs, custom frontends) still gets its session, half
+        a second later.
+        """
+        try:
+            grace = max(0.0, float(self._config.session_trigger_grace_seconds or 0))
+            if grace:
+                await asyncio.sleep(grace)
+            if websocket in self._connection_triggered or not self._is_live(websocket):
+                safe_log(
+                    self._logger,
+                    "info",
+                    "automatic session trigger stood down",
+                    {"client": client_info, "user_id": user_id,
+                     "reason": "client asked" if websocket in self._connection_triggered else "connection gone"},
+                )
+                return
+            await self._trigger_and_emit(websocket, user_id, None, client_info)
+        except asyncio.CancelledError:
+            return
+        finally:
+            self._connection_trigger_tasks.pop(websocket, None)
+
     def _remember_page(self, websocket: WebSocketServerProtocol, events: Any) -> None:
         """Keeps the newest page_view's page for this connection (see _handle_user_trigger)."""
         if not isinstance(events, list):
@@ -1444,7 +1497,50 @@ class WebSocketServer:
         client_info: str,
         current_page: Optional[dict] = None,
     ) -> None:
-        """calls trigger_session API and emits response to FE."""
+        """calls trigger_session API and emits response to FE.
+
+        Two triggers for the same connection at once (the automatic one mid
+        call when the client's own request lands) share one Centive call: the
+        second waits for the first and re-emits its frame.
+        """
+        inflight = self._connection_trigger_inflight.get(websocket)
+        if inflight is not None:
+            safe_log(
+                self._logger,
+                "info",
+                "session trigger coalesced with one in flight",
+                {"client": client_info, "user_id": user_id},
+            )
+            try:
+                frame = await asyncio.shield(inflight)
+            except Exception:  # noqa: BLE001 - the first caller already reported it
+                frame = None
+            if frame is not None:
+                try:
+                    await websocket.send(frame)
+                except Exception:  # noqa: BLE001
+                    pass
+            return
+
+        done: asyncio.Future = asyncio.get_running_loop().create_future()
+        self._connection_trigger_inflight[websocket] = done
+        emitted: Optional[str] = None
+        try:
+            emitted = await self._trigger_once(websocket, user_id, user_trigger, client_info, current_page)
+        finally:
+            self._connection_trigger_inflight.pop(websocket, None)
+            if not done.done():
+                done.set_result(emitted)
+
+    async def _trigger_once(
+        self,
+        websocket: WebSocketServerProtocol,
+        user_id: str,
+        user_trigger: Optional[bool],
+        client_info: str,
+        current_page: Optional[dict],
+    ) -> Optional[str]:
+        """One trigger-session call. Returns the success frame it sent, if any."""
         safe_log(
             self._logger,
             "info",
@@ -1481,7 +1577,7 @@ class WebSocketServer:
                     "message": "failed to trigger session",
                     "error": "UPSTREAM_TRIGGER_FAILED",
                 }))
-                return
+                return None
 
             response_data = {
                 "status": response.status,
@@ -1490,7 +1586,8 @@ class WebSocketServer:
                 "time_taken": response.time_taken,
             }
 
-            await websocket.send(json.dumps(response_data))
+            frame = json.dumps(response_data)
+            await websocket.send(frame)
 
             safe_log(
                 self._logger,
@@ -1503,6 +1600,7 @@ class WebSocketServer:
                     "has_session_data": response.session_data is not None,
                 },
             )
+            return frame
 
         except Exception as e:
             # Generic on the wire, detailed in the server's own logs: upstream

{centive_sdk-2.2.0.dev6.dist-info → centive_sdk-2.2.0.dev8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: centive-sdk
-Version: 2.2.0.dev6
+Version: 2.2.0.dev8
 Summary: Python SDK for Centive backend integration
 Project-URL: Homepage, https://github.com/TheAgenticAI/centive-backend-sdk
 Project-URL: Repository, https://github.com/TheAgenticAI/centive-backend-sdk

{centive_sdk-2.2.0.dev6.dist-info → centive_sdk-2.2.0.dev8.dist-info}/RECORD

@@ -4,7 +4,7 @@ centive_sdk/_logging.py,sha256=z8Kdi4vJdko0nkMeySGSHwPkEB2E7ucklc9FeJCbyOw,3518
 centive_sdk/_retry.py,sha256=Q1cIFZyimj0RS4x14FEOseqpORCSydvJyzGiIPfDn7E,1755
 centive_sdk/async_client.py,sha256=RRoCJrvdeLwQz5UJv2EjmlgTq7vntSV9ELMWHyNwJ5I,9355
 centive_sdk/client.py,sha256=Evtsng8cqqRSoV54MI2IKbEW1q0vPOkxlz9zM0evH7k,3249
-centive_sdk/config.py,sha256=HX0N7UCu9QLS5bICW_iME1LeBBG83Za2-cYoV-9VgLA,6187
+centive_sdk/config.py,sha256=1GoHkwJebL1HknSa-y4oFhxDAaG0boq0JY0csNV-pfE,6612
 centive_sdk/exceptions.py,sha256=wFy-hU0TJ0dHa6TEDI7FIQu9Lbg5aMeWxW9E-IMMUbI,1841
 centive_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 centive_sdk/models/__init__.py,sha256=-ledNpbOi1Q14ELhvlDbGPF37N3yv5XJ-QsH2I6UMVQ,598
@@ -16,8 +16,8 @@ centive_sdk/resources/async_telemetry.py,sha256=wT12g9rG2dp_ml5YpAYCxobLlhfcsyy-
 centive_sdk/resources/message_accumulator.py,sha256=Or3sMGkXG9upeQ9sm8YWL_V5vvsxtnE9tu52VBUUGgY,19261
 centive_sdk/resources/sessions.py,sha256=Ipg4bpmpejA5WoALbT49ah0c3I7x4OgPn9tSmfW8JAw,16741
 centive_sdk/resources/telemetry_batcher.py,sha256=OkOv2Iir6E_9mdtAKvelol3RaveGSlH-ckIoE2cQYCc,14629
-centive_sdk/resources/websocket_server.py,sha256=fGOWxz4lvshGFnWFgf-euQYie6mZUohaniidnWA0KvI,61860
-centive_sdk-2.2.0.dev6.dist-info/METADATA,sha256=YeBnZ3uGkNme6CG2LcZYKEgHprUp5O0b7XRjGwgYgF0,36427
-centive_sdk-2.2.0.dev6.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-centive_sdk-2.2.0.dev6.dist-info/licenses/LICENSE,sha256=9dFibPr7z9srhWwd35lQL2MX1Kx2qivEZdVQb9L8Q-Q,1064
-centive_sdk-2.2.0.dev6.dist-info/RECORD,,
+centive_sdk/resources/websocket_server.py,sha256=RTkM1YyD7afNWs4KpoboCo-NI2TFgk8IkzaH0KAKR5A,66518
+centive_sdk-2.2.0.dev8.dist-info/METADATA,sha256=5DM9IFShg7CIqzt5jQ8goRocZXmOCJtELTUWVUgVSZY,36427
+centive_sdk-2.2.0.dev8.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+centive_sdk-2.2.0.dev8.dist-info/licenses/LICENSE,sha256=9dFibPr7z9srhWwd35lQL2MX1Kx2qivEZdVQb9L8Q-Q,1064
+centive_sdk-2.2.0.dev8.dist-info/RECORD,,