agentkit-sdk-python 0.1.6__tar.gz → 0.1.11__tar.gz

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.1.6
+Version: 0.1.11
 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: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Xiangrui Cheng <innsdcc@gmail.com>
+Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Xiangrui Cheng <innsdcc@gmail.com>, Yumeng Bao <baoyumeng.123@gmail.com>
 License:                                  Apache License
                                    Version 2.0, January 2004
                                 http://www.apache.org/licenses/
@@ -226,7 +226,6 @@ Requires-Dist: docker
 Requires-Dist: jinja2
 Requires-Dist: tos>=2.8.7
 Requires-Dist: volcengine
-Requires-Dist: volcengine-python-sdk
 Requires-Dist: pyfiglet>=1.0.2
 Provides-Extra: extensions
 Provides-Extra: toolkit

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/agentkit/apps/a2a_app/a2a_app.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import logging
-import time
 from typing import Callable, override
 
 import uvicorn

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/agentkit/apps/mcp_app/mcp_app.py

@@ -15,7 +15,6 @@
 
 import inspect
 import logging
-import time
 from functools import wraps
 from typing import Any, Callable, override
 

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/agentkit/apps/mcp_app/telemetry.py

@@ -21,8 +21,6 @@ from opentelemetry import trace
 from opentelemetry.trace import get_tracer
 from opentelemetry.metrics import get_meter
 from opentelemetry.trace.span import Span
-from a2a.server.agent_execution.context import RequestContext
-from google.adk.a2a.converters.request_converter import _get_user_id
 
 from agentkit.apps.utils import safe_serialize_to_json_string
 

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/agentkit/apps/simple_app/simple_app_handlers.py

@@ -133,7 +133,10 @@ class InvokeHandler(BaseHandler):
         Returns:
             bytes: SSE-formatted data ready for streaming
         """
-        json_string = safe_serialize_to_json_string(obj)
+        if isinstance(obj, str):
+            json_string = obj
+        else:
+            json_string = safe_serialize_to_json_string(obj)
         sse_data = f"data: {json_string}\n\n"
         return sse_data.encode("utf-8")
 

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/agentkit/apps/simple_app/telemetry.py

@@ -58,7 +58,7 @@ def dont_throw(func):
     def wrapper(*args, **kwargs):
         try:
             return func(*args, **kwargs)
-        except Exception as e:
+        except Exception:
             logger.error(
                 "Agentkit failed to trace in %s, error: %s",
                 func.__name__,

{agentkit_sdk_python-0.1.6 → agentkit_sdk_python-0.1.11}/agentkit/client/__init__.py

@@ -15,12 +15,16 @@
 """
 AgentKit Client Module
 
-Provides base client infrastructure for all AgentKit services.
+Provides base client infrastructure for all AgentKit and Volcengine services.
 """
 
-from agentkit.client.base_client import BaseAgentkitClient, ApiConfig
+from agentkit.client.base_service_client import BaseServiceClient, ApiConfig
+from agentkit.client.base_agentkit_client import BaseAgentkitClient
+from agentkit.client.base_iam_client import BaseIAMClient
 
 __all__ = [
+    'BaseServiceClient',
     'BaseAgentkitClient',
+    'BaseIAMClient',
     'ApiConfig',
 ]

agentkit_sdk_python-0.1.11/agentkit/client/base_agentkit_client.py

@@ -0,0 +1,92 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Base client for AgentKit services.
+Provides common initialization and API invocation logic.
+"""
+
+from typing import Any, Dict, Union, Optional
+
+from agentkit.client.base_service_client import BaseServiceClient, ApiConfig
+from agentkit.utils.ve_sign import get_volc_agentkit_host_info
+
+
+class BaseAgentkitClient(BaseServiceClient):
+    """
+    Base client for all AgentKit services.
+    
+    This class provides:
+    1. Common credential initialization
+    2. Unified API invocation logic with error handling
+    3. Automatic ApiInfo generation with flexible configuration
+    
+    Subclasses should override API_ACTIONS with either:
+    - Simple dict mapping: {"ActionName": "ActionName"}
+    - Detailed ApiConfig: {"ActionName": ApiConfig(action="ActionName", method="GET", path="/custom")}
+    """
+    
+    # Subclasses should override this with their API action configurations
+    API_ACTIONS: Dict[str, Union[str, ApiConfig]] = {}
+    
+    def __init__(
+        self,
+        access_key: str = "",
+        secret_key: str = "",
+        region: str = "",
+        session_token: str = "",
+        service_name: str = "",
+        header: Optional[Dict[str, Any]] = {"Accept": "application/json"},
+    ) -> None:
+        """
+        Initialize the AgentKit client.
+        
+        Args:
+            access_key: Volcengine access key
+            secret_key: Volcengine secret key
+            region: Volcengine region
+            session_token: Optional session token
+            service_name: Service name for logging (e.g., 'knowledge', 'memory')
+        """
+        super().__init__(
+            access_key=access_key,
+            secret_key=secret_key,
+            region=region,
+            session_token=session_token,
+            service_name=service_name,
+            credential_env_prefix='AGENTKIT',
+            header=header,
+        )
+    
+    def _get_service_config(self) -> Dict[str, str]:
+        """
+        Get AgentKit service configuration.
+        
+        Returns:
+            Dictionary with host, api_version, and service
+        """
+        host, api_version, service = get_volc_agentkit_host_info()
+        return {
+            'host': host,
+            'api_version': api_version,
+            'service': service,
+        }
+    
+    def _get(self, api_action: str, params: Dict[str, Any] = None) -> str:
+        """Legacy method for GET requests."""
+        try:
+            resp = self.get(api_action, params)
+            return resp
+        except Exception as e:
+            raise Exception(f"Failed to {api_action}: {str(e)}")

agentkit_sdk_python-0.1.11/agentkit/client/base_iam_client.py

@@ -0,0 +1,85 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Base client for IAM services.
+Uses the same volcengine.base.Service approach as AgentKit services.
+"""
+
+from typing import Dict, Union
+
+from agentkit.client.base_service_client import BaseServiceClient, ApiConfig
+
+
+class BaseIAMClient(BaseServiceClient):
+    """
+    Base client for IAM services.
+    
+    This class provides the same interface as BaseAgentkitClient but for IAM services:
+    1. Common credential initialization
+    2. Unified API invocation logic with error handling
+    3. Automatic ApiInfo generation with flexible configuration
+    
+    Subclasses should override API_ACTIONS with either:
+    - Simple dict mapping: {"ActionName": "ActionName"}
+    - Detailed ApiConfig: {"ActionName": ApiConfig(action="ActionName", method="GET", path="/custom")}
+    """
+    
+    # Subclasses should override this with their API action configurations
+    API_ACTIONS: Dict[str, Union[str, ApiConfig]] = {}
+    
+    # IAM service specific configuration
+    IAM_API_VERSION = "2018-01-01"
+    IAM_SERVICE_CODE = "iam"
+    IAM_HOST ="open.volcengineapi.com"
+    
+    def __init__(
+        self,
+        access_key: str = "",
+        secret_key: str = "",
+        region: str = "",
+        session_token: str = "",
+        service_name: str = "iam",
+    ) -> None:
+        """
+        Initialize the IAM client.
+        
+        Args:
+            access_key: Volcengine access key
+            secret_key: Volcengine secret key
+            region: Volcengine region
+            session_token: Optional session token
+            service_name: Service name for logging
+        """
+        super().__init__(
+            access_key=access_key,
+            secret_key=secret_key,
+            region=region,
+            session_token=session_token,
+            service_name=service_name,
+            credential_env_prefix='IAM',
+        )
+    
+    def _get_service_config(self) -> Dict[str, str]:
+        """
+        Get IAM service configuration.
+        
+        Returns:
+            Dictionary with host, api_version, and service
+        """
+        return {
+            'host': self.IAM_HOST,
+            'api_version': self.IAM_API_VERSION,
+            'service': self.IAM_SERVICE_CODE,
+        }

agentkit_sdk_python-0.1.6/agentkit/client/base_client.py → agentkit_sdk_python-0.1.11/agentkit/client/base_service_client.py

@@ -13,22 +13,20 @@
 # limitations under the License.
 
 """
-Base client for AgentKit services.
-Provides common initialization and API invocation logic.
+Base service client that provides common implementation for all Volcengine services.
+This is the top-level base class for all service clients.
 """
 
 import json
-from typing import Any, Dict, Type, TypeVar, TYPE_CHECKING, Union, Optional
+from typing import Any, Dict, Type, TypeVar, Union, Optional
 from dataclasses import dataclass
 
 from volcengine.ApiInfo import ApiInfo
 from volcengine.base.Service import Service
 from volcengine.Credentials import Credentials
 from volcengine.ServiceInfo import ServiceInfo
-from agentkit.utils.ve_sign import get_volc_ak_sk_region, get_volc_agentkit_host_info
 
-if TYPE_CHECKING:
-    from pydantic import BaseModel
+from agentkit.utils.ve_sign import get_volc_ak_sk_region
 
 T = TypeVar('T')
 
@@ -49,18 +47,18 @@ class ApiConfig:
             self.header = {}
 
 
-class BaseAgentkitClient(Service):
+class BaseServiceClient(Service):
     """
-    Base client for all AgentKit services.
+    Base class for all Volcengine service clients.
     
     This class provides:
-    1. Common credential initialization
-    2. Unified API invocation logic with error handling
-    3. Automatic ApiInfo generation with flexible configuration
+    1. Unified interface for all Volcengine services (AgentKit, IAM, etc.)
+    2. Common implementation using volcengine.base.Service
+    3. Shared credential management and API invocation logic
     
-    Subclasses should override API_ACTIONS with either:
-    - Simple dict mapping: {"ActionName": "ActionName"}
-    - Detailed ApiConfig: {"ActionName": ApiConfig(action="ActionName", method="GET", path="/custom")}
+    Subclasses should:
+    1. Override API_ACTIONS with their API action configurations
+    2. Implement _get_service_config() to provide service-specific configuration
     """
     
     # Subclasses should override this with their API action configurations
@@ -73,20 +71,23 @@ class BaseAgentkitClient(Service):
         region: str = "",
         session_token: str = "",
         service_name: str = "",
+        credential_env_prefix: str = "",
+        header: Optional[Dict[str, Any]] = {"Accept": "application/json"},
     ) -> None:
         """
-        Initialize the AgentKit client.
+        Initialize the service client.
         
         Args:
             access_key: Volcengine access key
             secret_key: Volcengine secret key
             region: Volcengine region
             session_token: Optional session token
-            service_name: Service name for logging (e.g., 'knowledge', 'memory')
+            service_name: Service name for logging
+            credential_env_prefix: Environment variable prefix for credentials (e.g., 'AGENTKIT', 'IAM')
         """
         # Validate and get credentials
         if not any([access_key, secret_key, region]):
-            access_key, secret_key, region = get_volc_ak_sk_region('AGENTKIT')
+            access_key, secret_key, region = get_volc_ak_sk_region(credential_env_prefix)
         else:
             if not all([access_key, secret_key, region]):
                 raise ValueError(
@@ -94,18 +95,23 @@ class BaseAgentkitClient(Service):
                     "missing access key, secret key or region"
                 )
         
+        # Store credentials and service info
         self.access_key = access_key
         self.secret_key = secret_key
         self.region = region
         self.session_token = session_token
+        self.service_name = service_name
+        
+        # Get service-specific configuration from subclass
+        config = self._get_service_config()
+        self.host = config['host']
+        self.api_version = config['api_version']
+        self.service = config['service']
         
-        # Get service configuration
-        self.host, self.api_version, self.service = get_volc_agentkit_host_info()
-
         # Create ServiceInfo
         self.service_info = ServiceInfo(
             host=self.host,
-            header={'Accept': 'application/json'},
+            header=header,
             credentials=Credentials(
                 ak=self.access_key,
                 sk=self.secret_key,
@@ -122,7 +128,21 @@ class BaseAgentkitClient(Service):
         self.api_info = self._build_api_info()
         
         # Initialize parent Service class
-        super().__init__(service_info=self.service_info, api_info=self.api_info)
+        Service.__init__(self, service_info=self.service_info, api_info=self.api_info)
+    
+    def _get_service_config(self) -> Dict[str, str]:
+        """
+        Get service-specific configuration.
+        
+        Subclasses must override this method to provide:
+        - host: API endpoint host
+        - api_version: API version string
+        - service: Service name for signing
+        
+        Returns:
+            Dictionary with 'host', 'api_version', and 'service' keys
+        """
+        raise NotImplementedError("Subclasses must implement _get_service_config()")
     
     def _build_api_info(self) -> Dict[str, ApiInfo]:
         """
@@ -161,12 +181,6 @@ class BaseAgentkitClient(Service):
                     f"expected str or ApiConfig, got {type(action_config)}"
                 )
         return api_info
-    def _get(self, api_action: str, params: Dict[str, Any] = None) -> str:
-        try:
-            resp = self.get(api_action, params)
-            return resp
-        except Exception as e:
-            raise Exception(f"Failed to {api_action}: {str(e)}")
     
     def _invoke_api(
         self,
@@ -179,7 +193,7 @@ class BaseAgentkitClient(Service):
         Unified API invocation with error handling.
         
         Args:
-            api_action: The API action name (e.g., 'ListKnowledgeBases')
+            api_action: The API action name (e.g., 'GetUser', 'ListRuntimes')
             request: The request object (Pydantic model)
             response_type: The response type to parse into
             params: Additional query parameters
@@ -213,5 +227,4 @@ class BaseAgentkitClient(Service):
             raise Exception(f"Failed to {api_action}: {error_msg}")
         
         # Return typed response
-        # print(json.dumps(response_data, indent=2, ensure_ascii=False))
         return response_type(**response_data.get('Result', {}))

agentkit_sdk_python-0.1.11/agentkit/sdk/__init__.py

@@ -0,0 +1,14 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+

agentkit_sdk_python-0.1.11/agentkit/sdk/account/client.py

@@ -0,0 +1,110 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated from API JSON definition
+# Do not edit manually
+
+from __future__ import annotations
+
+from typing import Dict, List, Optional
+from agentkit.client import BaseAgentkitClient
+from agentkit.sdk.account.types import (
+    ListAccountLinkedServicesRequest,
+    ListAccountLinkedServicesResponse,
+)
+
+
+class AgentkitAccountClient(BaseAgentkitClient):
+    """AgentKit Account Management Service"""
+    API_ACTIONS: Dict[str, str] = {
+        "ListAccountLinkedServices": "ListAccountLinkedServices",
+    }
+
+    def __init__(
+        self,
+        access_key: str = "",
+        secret_key: str = "",
+        region: str = "",
+        session_token: str = "",
+    ) -> None:
+        super().__init__(
+            access_key=access_key,
+            secret_key=secret_key,
+            region=region,
+            session_token=session_token,
+            service_name="account",
+        )
+
+
+    def list_account_linked_services(self, request: ListAccountLinkedServicesRequest) -> ListAccountLinkedServicesResponse:
+        return self._invoke_api(
+            api_action="ListAccountLinkedServices",
+            request=request,
+            response_type=ListAccountLinkedServicesResponse,
+        )
+
+    def get_service_status(self, service_name: str) -> Optional[str]:
+        """
+        Query the LinkServices status for a specific service.
+        
+        Args:
+            service_name: The name of the service to query (e.g., 'ark', 'vefaas')
+            
+        Returns:
+            The status string ('Enabled' or 'Disabled'), or None if service not found.
+        """
+        print(1)
+        response = self.list_account_linked_services(ListAccountLinkedServicesRequest())
+        print(2)
+        if not response.service_statuses:
+            return None
+        for svc in response.service_statuses:
+            if svc.service_name == service_name:
+                return svc.status
+        return None
+
+    def get_services_status(self, service_names: List[str]) -> Dict[str, Optional[str]]:
+        """
+        Query the LinkServices status for multiple services.
+        
+        Args:
+            service_names: List of service names to query (e.g., ['ark', 'vefaas', 'cr'])
+            
+        Returns:
+            A dict mapping service_name -> status ('Enabled'/'Disabled'/None if not found).
+        """
+        response = self.list_account_linked_services(ListAccountLinkedServicesRequest())
+        status_map: Dict[str, Optional[str]] = {name: None for name in service_names}
+        if response.service_statuses:
+            for svc in response.service_statuses:
+                if svc.service_name in status_map:
+                    status_map[svc.service_name] = svc.status
+        return status_map
+
+    def has_disabled_services(self) -> List[str]:
+        """
+        Check if any returned service has a disabled status.
+        
+        Returns:
+            A list of service names that are disabled (status != 'Enabled').
+            Returns empty list if all services are enabled.
+        """
+        response = self.list_account_linked_services(ListAccountLinkedServicesRequest())
+        disabled = []
+        if response.service_statuses:
+            for svc in response.service_statuses:
+                if svc.status != "Enabled":
+                    disabled.append(svc.service_name)
+        return disabled
+

agentkit_sdk_python-0.1.11/agentkit/sdk/account/types.py

@@ -0,0 +1,45 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated from API JSON definition
+# Do not edit manually
+
+from __future__ import annotations
+
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class AccountBaseModel(BaseModel):
+    """AgentKit auto-generated base model"""
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+# Data Types
+class ServiceStatusesForListAccountLinkedServices(AccountBaseModel):
+    service_name: Optional[str] = Field(default=None, alias="ServiceName")
+    status: Optional[str] = Field(default=None, alias="Status")
+
+
+# ListAccountLinkedServices - Request
+class ListAccountLinkedServicesRequest(AccountBaseModel):
+    pass
+
+
+# ListAccountLinkedServices - Response
+class ListAccountLinkedServicesResponse(AccountBaseModel):
+    service_statuses: Optional[list[ServiceStatusesForListAccountLinkedServices]] = Field(default=None, alias="ServiceStatuses")
+

{agentkit_sdk_python-0.1.6/agentkit → agentkit_sdk_python-0.1.11/agentkit/sdk}/identity/auth.py

@@ -18,7 +18,7 @@ from functools import wraps
 from typing import Any, Callable
 
 from agentkit.utils.credential import get_credential_from_vefaas_iam
-from agentkit.utils.ve_sign import ve_request
+from agentkit.utils.ve_sign import ve_request, get_identity_host_info
 
 
 def requires_api_key(*, provider_name: str, into: str = "api_key"):
@@ -37,6 +37,7 @@ def requires_api_key(*, provider_name: str, into: str = "api_key"):
             access_key = os.getenv("VOLCENGINE_ACCESS_KEY")
             secret_key = os.getenv("VOLCENGINE_SECRET_KEY")
             session_token = ""
+            host, version, service, region = get_identity_host_info()
 
             if not (access_key and secret_key):
                 cred = get_credential_from_vefaas_iam()
@@ -53,10 +54,10 @@ def requires_api_key(*, provider_name: str, into: str = "api_key"):
                 header={"X-Security-Token": session_token} if session_token else {},
                 ak=access_key,
                 sk=secret_key,
-                version="2023-10-01",
-                service="cis_test",
-                host="open.volcengineapi.com",
-                region="cn-beijing",
+                version=version,
+                service=service,
+                host=host,
+                region=region,
             )
 
             try:

{agentkit_sdk_python-0.1.6/agentkit → agentkit_sdk_python-0.1.11/agentkit/sdk}/knowledge/__init__.py

@@ -11,9 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-from agentkit.knowledge.knowledge import AgentkitKnowledge
-from agentkit.knowledge.knowledge_all_types import (
+from .client import AgentkitKnowledge
+from .types import (
     ListKnowledgeBasesRequest,
     ListKnowledgeBasesResponse,
     AddKnowledgeBaseRequest,
@@ -25,8 +24,6 @@ from agentkit.knowledge.knowledge_all_types import (
     KnowledgeBasesForListKnowledgeBases,
     KnowledgeBasesForAddKnowledgeBase,
     ConnectionInfosForGetKnowledgeConnectionInfo,
-    KnowledgeBasesItem,
-    FiltersItem,
 )
 
 __all__ = [
@@ -44,4 +41,4 @@ __all__ = [
     "ConnectionInfosForGetKnowledgeConnectionInfo",
     "KnowledgeBasesItem",
     "FiltersItem",
-]
+]