agentkit-sdk-python 0.7.7__tar.gz → 0.7.9__tar.gz

{agentkit_sdk_python-0.7.7 → agentkit_sdk_python-0.7.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.7.7
+Version: 0.7.9
 Summary: Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure.
 Author-email: Xiangrui Cheng <innsdcc@gmail.com>, Yumeng Bao <baoyumeng.123@gmail.com>, Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>
 License:                                  Apache License

{agentkit_sdk_python-0.7.7 → agentkit_sdk_python-0.7.9}/agentkit/sdk/identity/auth.py

@@ -41,7 +41,7 @@ def requires_api_key(*, provider_name: str, into: str = "api_key"):
             endpoint = resolve_endpoint("identity")
             access_key = creds.access_key
             secret_key = creds.secret_key
-            session_token = ""
+            session_token = getattr(creds, "session_token", None) or None
 
             response = ve_request(
                 request_body={
@@ -49,9 +49,9 @@ def requires_api_key(*, provider_name: str, into: str = "api_key"):
                     "IdentityToken": "identity_token",
                 },
                 action="GetResourceApiKey",
-                header={"X-Security-Token": session_token} if session_token else {},
                 ak=access_key,
                 sk=secret_key,
+                session_token=session_token,
                 version=endpoint.api_version,
                 service=endpoint.service,
                 host=endpoint.host,

{agentkit_sdk_python-0.7.7 → agentkit_sdk_python-0.7.9}/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",