agentkit-sdk-python 0.7.6__py3-none-any.whl → 0.7.8__py3-none-any.whl

agentkit/frameworks/langgraph.py

@@ -37,6 +37,7 @@ from agentkit.frameworks._common import (
 
 
 LANGGRAPH_PENDING_INTERRUPT_STATE_KEY = "agentkit:langgraph:pending_interrupt"
+_OUTPUT_FIELD_KEYS = ("answer", "output", "final", "response", "text", "messages")
 
 
 def _method_kwargs(method: Any, kwargs: dict[str, Any]) -> dict[str, Any]:
@@ -96,7 +97,7 @@ def _update_output_text(data: Any) -> str:
     if not isinstance(data, dict):
         return chunk_to_text(data)
 
-    for key in ("answer", "output", "text", "content", "messages"):
+    for key in _OUTPUT_FIELD_KEYS:
         if key in data:
             text = chunk_to_text(data[key])
             if text:
@@ -106,7 +107,7 @@ def _update_output_text(data: Any) -> str:
     for value in data.values():
         if not isinstance(value, dict):
             continue
-        for key in ("answer", "output", "text", "content", "messages"):
+        for key in _OUTPUT_FIELD_KEYS:
             if key in value:
                 text = chunk_to_text(value[key])
                 if text:
@@ -325,6 +326,8 @@ class LangGraphAgentkitBridge(BaseAgent):
             elif isinstance(item, dict) and isinstance(item.get("type"), str):
                 mode = item["type"]
                 data = item.get("data")
+            elif isinstance(item, dict):
+                mode = "updates"
 
             if mode == "messages":
                 message = data[0] if isinstance(data, tuple) and data else data

agentkit/sdk/tools/client.py

@@ -22,22 +22,32 @@ from agentkit.client import BaseAgentkitClient
 from .types import (
     CreateSessionRequest,
     CreateSessionResponse,
+    CreateSessionSnapshotRequest,
+    CreateSessionSnapshotResponse,
     CreateToolRequest,
     CreateToolResponse,
     DeleteSessionRequest,
     DeleteSessionResponse,
+    DeleteSessionSnapshotRequest,
+    DeleteSessionSnapshotResponse,
     DeleteToolRequest,
     DeleteToolResponse,
     GetSessionLogsRequest,
     GetSessionLogsResponse,
     GetSessionRequest,
     GetSessionResponse,
+    GetSessionSnapshotRequest,
+    GetSessionSnapshotResponse,
     GetToolRequest,
     GetToolResponse,
+    ListSessionSnapshotsRequest,
+    ListSessionSnapshotsResponse,
     ListSessionsRequest,
     ListSessionsResponse,
     ListToolsRequest,
     ListToolsResponse,
+    ResumeSessionFromSnapshotRequest,
+    ResumeSessionFromSnapshotResponse,
     SetSessionTtlRequest,
     SetSessionTtlResponse,
     UpdateToolRequest,
@@ -47,16 +57,22 @@ from .types import (
 
 class AgentkitToolsClient(BaseAgentkitClient):
     """AgentKit Tools Management Service"""
+
     API_ACTIONS: Dict[str, str] = {
         "CreateSession": "CreateSession",
+        "CreateSessionSnapshot": "CreateSessionSnapshot",
         "CreateTool": "CreateTool",
         "DeleteSession": "DeleteSession",
+        "DeleteSessionSnapshot": "DeleteSessionSnapshot",
         "DeleteTool": "DeleteTool",
         "GetSession": "GetSession",
         "GetSessionLogs": "GetSessionLogs",
+        "GetSessionSnapshot": "GetSessionSnapshot",
         "GetTool": "GetTool",
+        "ListSessionSnapshots": "ListSessionSnapshots",
         "ListSessions": "ListSessions",
         "ListTools": "ListTools",
+        "ResumeSessionFromSnapshot": "ResumeSessionFromSnapshot",
         "SetSessionTtl": "SetSessionTtl",
         "UpdateTool": "UpdateTool",
     }
@@ -76,7 +92,6 @@ class AgentkitToolsClient(BaseAgentkitClient):
             service_name="tools",
         )
 
-
     def create_session(self, request: CreateSessionRequest) -> CreateSessionResponse:
         return self._invoke_api(
             api_action="CreateSession",
@@ -84,6 +99,15 @@ class AgentkitToolsClient(BaseAgentkitClient):
             response_type=CreateSessionResponse,
         )
 
+    def create_session_snapshot(
+        self, request: CreateSessionSnapshotRequest
+    ) -> CreateSessionSnapshotResponse:
+        return self._invoke_api(
+            api_action="CreateSessionSnapshot",
+            request=request,
+            response_type=CreateSessionSnapshotResponse,
+        )
+
     def create_tool(self, request: CreateToolRequest) -> CreateToolResponse:
         return self._invoke_api(
             api_action="CreateTool",
@@ -98,6 +122,15 @@ class AgentkitToolsClient(BaseAgentkitClient):
             response_type=DeleteSessionResponse,
         )
 
+    def delete_session_snapshot(
+        self, request: DeleteSessionSnapshotRequest
+    ) -> DeleteSessionSnapshotResponse:
+        return self._invoke_api(
+            api_action="DeleteSessionSnapshot",
+            request=request,
+            response_type=DeleteSessionSnapshotResponse,
+        )
+
     def delete_tool(self, request: DeleteToolRequest) -> DeleteToolResponse:
         return self._invoke_api(
             api_action="DeleteTool",
@@ -112,13 +145,24 @@ class AgentkitToolsClient(BaseAgentkitClient):
             response_type=GetSessionResponse,
         )
 
-    def get_session_logs(self, request: GetSessionLogsRequest) -> GetSessionLogsResponse:
+    def get_session_logs(
+        self, request: GetSessionLogsRequest
+    ) -> GetSessionLogsResponse:
         return self._invoke_api(
             api_action="GetSessionLogs",
             request=request,
             response_type=GetSessionLogsResponse,
         )
 
+    def get_session_snapshot(
+        self, request: GetSessionSnapshotRequest
+    ) -> GetSessionSnapshotResponse:
+        return self._invoke_api(
+            api_action="GetSessionSnapshot",
+            request=request,
+            response_type=GetSessionSnapshotResponse,
+        )
+
     def get_tool(self, request: GetToolRequest) -> GetToolResponse:
         return self._invoke_api(
             api_action="GetTool",
@@ -126,6 +170,15 @@ class AgentkitToolsClient(BaseAgentkitClient):
             response_type=GetToolResponse,
         )
 
+    def list_session_snapshots(
+        self, request: ListSessionSnapshotsRequest
+    ) -> ListSessionSnapshotsResponse:
+        return self._invoke_api(
+            api_action="ListSessionSnapshots",
+            request=request,
+            response_type=ListSessionSnapshotsResponse,
+        )
+
     def list_sessions(self, request: ListSessionsRequest) -> ListSessionsResponse:
         return self._invoke_api(
             api_action="ListSessions",
@@ -140,6 +193,15 @@ class AgentkitToolsClient(BaseAgentkitClient):
             response_type=ListToolsResponse,
         )
 
+    def resume_session_from_snapshot(
+        self, request: ResumeSessionFromSnapshotRequest
+    ) -> ResumeSessionFromSnapshotResponse:
+        return self._invoke_api(
+            api_action="ResumeSessionFromSnapshot",
+            request=request,
+            response_type=ResumeSessionFromSnapshotResponse,
+        )
+
     def set_session_ttl(self, request: SetSessionTtlRequest) -> SetSessionTtlResponse:
         return self._invoke_api(
             api_action="SetSessionTtl",