Add action field to session:message payload per API spec 5.1.1

Author: bojieli

src/pine_ai/chat.py

@@ -86,26 +86,40 @@ def leave_session(self, session_id: str) -> None:
         """Leave a session room."""
         self._sio.emit(C2SEvent.SESSION_LEAVE, None, session_id)
 
+    @staticmethod
+    def _build_message_data(
+        content: str,
+        attachments: Optional[list[dict[str, Any]]] = None,
+        referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
+    ) -> dict[str, Any]:
+        """Build the session:message payload per spec 5.1.1."""
+        from datetime import datetime
+        data: dict[str, Any] = {
+            "content": content,
+            "attachments": attachments or [],
+            "referenced_sessions": referenced_sessions or [],
+            "client_now_date": datetime.now().isoformat(),
+        }
+        if action is not None:
+            data["action"] = action
+        return data
+
     async def chat(
         self,
         session_id: str,
         content: str,
         *,
         attachments: Optional[list[dict[str, Any]]] = None,
         referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
     ) -> AsyncGenerator[ChatEvent, None]:
         """Send a message and yield events with stream buffering.
         Production handler reads payload.data as {content, attachments, ...}.
         """
-        from datetime import datetime
         self._sio.emit(
             C2SEvent.SESSION_MESSAGE,
-            {
-                "content": content,
-                "attachments": attachments or [],
-                "referenced_sessions": referenced_sessions or [],
-                "client_now_date": datetime.now().isoformat(),
-            },
+            self._build_message_data(content, attachments, referenced_sessions, action),
             session_id,
         )
         async for event in self._listen(session_id):
@@ -118,17 +132,12 @@ def send_message(
         *,
         attachments: Optional[list[dict[str, Any]]] = None,
         referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
     ) -> None:
         """Fire-and-forget message send (no event listening)."""
-        from datetime import datetime
         self._sio.emit(
             C2SEvent.SESSION_MESSAGE,
-            {
-                "content": content,
-                "attachments": attachments or [],
-                "referenced_sessions": referenced_sessions or [],
-                "client_now_date": datetime.now().isoformat(),
-            },
+            self._build_message_data(content, attachments, referenced_sessions, action),
             session_id,
         )
 

src/pine_ai/client.py

@@ -122,13 +122,15 @@ async def chat(
         *,
         attachments: Optional[list[dict[str, Any]]] = None,
         referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
     ) -> AsyncGenerator[ChatEvent, None]:
         """Send a message and yield buffered events."""
         self._ensure_connected()
         async for event in self._chat.chat(  # type: ignore[union-attr]
             session_id, content,
             attachments=attachments,
             referenced_sessions=referenced_sessions,
+            action=action,
         ):
             yield event
 
@@ -139,13 +141,15 @@ def send_message(
         *,
         attachments: Optional[list[dict[str, Any]]] = None,
         referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
     ) -> None:
         """Send a message without waiting for events (fire-and-forget)."""
         self._ensure_connected()
         self._chat.send_message(  # type: ignore[union-attr]
             session_id, content,
             attachments=attachments,
             referenced_sessions=referenced_sessions,
+            action=action,
         )
 
     async def listen(self, session_id: str) -> AsyncGenerator[ChatEvent, None]:
@@ -239,6 +243,7 @@ def chat_sync(
         *,
         attachments: Optional[list[dict[str, Any]]] = None,
         referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
     ) -> list[ChatEvent]:
         """Send a message and return all events as a list (blocking)."""
         async def _collect() -> list[ChatEvent]:
@@ -247,6 +252,7 @@ async def _collect() -> list[ChatEvent]:
                 session_id, content,
                 attachments=attachments,
                 referenced_sessions=referenced_sessions,
+                action=action,
             ):
                 events.append(event)
             return events
@@ -259,12 +265,14 @@ def send_message(
         *,
         attachments: Optional[list[dict[str, Any]]] = None,
         referenced_sessions: Optional[list[dict[str, str]]] = None,
+        action: Optional[dict[str, Any]] = None,
     ) -> None:
         """Send a message without waiting for events (fire-and-forget)."""
         self._async.send_message(
             session_id, content,
             attachments=attachments,
             referenced_sessions=referenced_sessions,
+            action=action,
         )
 
     def send_form_response(self, session_id: str, message_id: str, form_data: dict[str, Any]) -> None: