agentkit-sdk-python 0.4.3__tar.gz → 0.5.0__tar.gz

{agentkit_sdk_python-0.4.3 → agentkit_sdk_python-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.4.3
+Version: 0.5.0
 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.4.3 → agentkit_sdk_python-0.5.0}/agentkit/apps/agent_server_app/agent_server_app.py

@@ -24,6 +24,7 @@ from fastapi.responses import StreamingResponse
 from google.adk.a2a.utils.agent_to_a2a import to_a2a
 from google.adk.agents.base_agent import BaseAgent
 from google.adk.agents.run_config import RunConfig, StreamingMode
+from google.adk.apps.app import App
 from google.adk.artifacts.in_memory_artifact_service import (
     InMemoryArtifactService,
 )
@@ -42,8 +43,8 @@ from google.adk.utils.context_utils import Aclosing
 from google.genai import types
 from opentelemetry import trace
 from veadk import Agent
-from veadk.memory.short_term_memory import ShortTermMemory
 from veadk.runner import Runner
+from veadk.memory.short_term_memory import ShortTermMemory
 
 from agentkit.apps.agent_server_app.middleware import (
     AgentkitTelemetryHTTPMiddleware,
@@ -55,27 +56,37 @@ logger = logging.getLogger(__name__)
 
 
 class AgentKitAgentLoader(BaseAgentLoader):
-    def __init__(self, agent: BaseAgent) -> None:
+    def __init__(self, agent_or_app: BaseAgent | App) -> None:
         super().__init__()
 
-        self.agent = agent
+        self.agent_or_app = agent_or_app
+        if isinstance(agent_or_app, App):
+            self.root_agent = agent_or_app.root_agent
+            self.app_name = agent_or_app.name or self.root_agent.name
+        else:
+            self.root_agent = agent_or_app
+            self.app_name = agent_or_app.name
 
     @override
-    def load_agent(self, agent_name: str) -> BaseAgent:
-        return self.agent
+    def load_agent(self, agent_name: str) -> BaseAgent | App:
+        if agent_name != self.app_name:
+            raise ValueError(
+                f"Unknown agent '{agent_name}'. Expected '{self.app_name}'."
+            )
+        return self.agent_or_app
 
     @override
     def list_agents(self) -> list[str]:
-        return [self.agent.name]
+        return [self.app_name]
 
     @override
     def list_agents_detailed(self) -> list[dict[str, Any]]:
-        name = self.agent.name
-        description = getattr(self.agent, "description", "") or ""
+        name = self.app_name
+        description = getattr(self.root_agent, "description", "") or ""
         return [
             {
                 "name": name,
-                "root_agent_name": name,
+                "root_agent_name": self.root_agent.name,
                 "description": description,
                 "language": "python",
             }
@@ -85,11 +96,25 @@ class AgentKitAgentLoader(BaseAgentLoader):
 class AgentkitAgentServerApp(BaseAgentkitApp):
     def __init__(
         self,
-        agent: BaseAgent,
-        short_term_memory: BaseSessionService | ShortTermMemory,
+        agent: BaseAgent | App | None = None,
+        short_term_memory: BaseSessionService | ShortTermMemory | None = None,
+        *,
+        app: App | None = None,
     ) -> None:
         super().__init__()
 
+        if short_term_memory is None:
+            raise TypeError("short_term_memory is required.")
+
+        if app is not None and agent is not None:
+            raise TypeError("Only one of 'agent' or 'app' can be provided.")
+
+        entry = app if app is not None else agent
+        if entry is None:
+            raise TypeError("Either 'agent' or 'app' must be provided.")
+
+        root_agent = entry.root_agent if isinstance(entry, App) else entry
+
         _artifact_service = InMemoryArtifactService()
         _credential_service = InMemoryCredentialService()
 
@@ -97,12 +122,12 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
         _eval_set_results_manager = LocalEvalSetResultsManager(agents_dir=".")
 
         self.server = AdkWebServer(
-            agent_loader=AgentKitAgentLoader(agent),
+            agent_loader=AgentKitAgentLoader(entry),
             session_service=short_term_memory
             if isinstance(short_term_memory, BaseSessionService)
             else short_term_memory.session_service,
-            memory_service=agent.long_term_memory
-            if isinstance(agent, Agent) and agent.long_term_memory
+            memory_service=root_agent.long_term_memory
+            if isinstance(root_agent, Agent) and root_agent.long_term_memory
             else InMemoryMemoryService(),
             artifact_service=_artifact_service,
             credential_service=_credential_service,
@@ -111,8 +136,8 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
             agents_dir=".",
         )
 
-        runner = Runner(agent=agent)
-        _a2a_server_app = to_a2a(agent=agent, runner=runner)
+        runner = Runner(agent=root_agent)
+        _a2a_server_app = to_a2a(agent=root_agent, runner=runner)
 
         @asynccontextmanager
         async def lifespan(app: FastAPI):

{agentkit_sdk_python-0.4.3 → agentkit_sdk_python-0.5.0}/agentkit/apps/simple_app/simple_app_handlers.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import asyncio
+import json
 import inspect
 import logging
 import threading
@@ -22,6 +23,7 @@ from abc import ABC, abstractmethod
 from typing import Any, Callable, Optional
 from typing_extensions import override
 
+from starlette.exceptions import HTTPException
 from starlette.requests import Request
 from starlette.responses import JSONResponse, Response, StreamingResponse
 
@@ -33,6 +35,13 @@ from opentelemetry import trace
 logger = logging.getLogger("agentkit." + __name__)
 
 
+class InvalidJSONPayloadError(Exception): ...
+
+
+def _build_error_content(message: str, error_type: str) -> dict[str, dict[str, str]]:
+    return {"error": {"message": message, "type": error_type}}
+
+
 class BaseHandler(ABC):
     def __init__(self, func: Callable | None = None) -> None:
         self.func = func
@@ -46,7 +55,13 @@ class InvokeHandler(BaseHandler):
     async def handle(self, request: Request) -> Response | StreamingResponse:
         if not self.func:
             logger.error("Invoke handler function is not set")
-            return Response(status_code=404)
+            return JSONResponse(
+                status_code=404,
+                content=_build_error_content(
+                    message="Entrypoint function is not set. Please register a function with @app.entrypoint.",
+                    error_type="NotFound",
+                ),
+            )
 
         span = telemetry.tracer.start_span(name="agentkit_invocation")
         ctx = trace.set_span_in_context(span)
@@ -72,13 +87,58 @@ class InvokeHandler(BaseHandler):
                     media_type="text/event-stream",
                 )
 
+            if isinstance(result, Response):
+                logger.info("Returning Starlette response")
+                telemetry.trace_agent_finish("", None)
+                return result
+
             logger.info("Returning non-streaming response")
             safe_json_string = safe_serialize_to_json_string(result)
             telemetry.trace_agent_finish(safe_json_string, None)
+        except InvalidJSONPayloadError as e:
+            logger.warning("Failed to parse JSON payload: %s", e)
+            telemetry.trace_agent_finish("", e)
+            return JSONResponse(
+                status_code=400,
+                content=_build_error_content(
+                    message="Invalid JSON payload. Please provide a valid JSON object in the request body.",
+                    error_type="BadRequest",
+                ),
+            )
+        except HTTPException as e:
+            logger.info("Returning HTTP exception: %s", e.status_code)
+            status_code = int(getattr(e, "status_code", 500) or 500)
+            detail = getattr(e, "detail", "")
+            if status_code < 500:
+                telemetry.trace_agent_finish("", None)
+                if isinstance(detail, str):
+                    message = detail
+                else:
+                    message = safe_serialize_to_json_string(detail)
+                content = _build_error_content(
+                    message=message, error_type="HTTPException"
+                )
+            else:
+                telemetry.trace_agent_finish("", e)
+                content = _build_error_content(
+                    message="An error occurred inside the user-defined entrypoint function (decorated with @app.entrypoint). This is likely caused by the agent application, not AgentKit. Please contact agent administrator and check the Runtime logs.",
+                    error_type="HTTPException",
+                )
+            return JSONResponse(
+                status_code=status_code,
+                content=content,
+                headers=getattr(e, "headers", None),
+            )
         except Exception as e:
-            logger.error("Invoke handler function failed: %s", e)
+            logger.exception("Invoke handler function failed: %s", e)
             telemetry.trace_agent_finish("", e)
-            raise e
+            return JSONResponse(
+                status_code=500,
+                content=_build_error_content(
+                    message="An error occurred inside the user-defined entrypoint function (decorated with @app.entrypoint). This is likely caused by the agent application, not AgentKit. Please contact agent administrator and check the Runtime logs.",
+                    error_type=type(e).__name__,
+                ),
+            )
 
         return Response(safe_json_string, media_type="application/json")
 
@@ -99,7 +159,10 @@ class InvokeHandler(BaseHandler):
             return {}, {}, {"message": "Invoke handler function is not set."}
 
         # parse request
-        payload: dict = await request.json()
+        try:
+            payload: dict = await request.json()
+        except json.JSONDecodeError as e:
+            raise InvalidJSONPayloadError(str(e)) from e
         headers: dict = dict(request.headers)
 
         # parse entrypoint function params

{agentkit_sdk_python-0.4.3 → agentkit_sdk_python-0.5.0}/agentkit/sdk/knowledge/types.py

@@ -100,6 +100,9 @@ class VpcConfigForGetKnowledgeBase(KnowledgeBaseModel):
 
 
 class VpcConfigurationForListKnowledgeBases(KnowledgeBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )

{agentkit_sdk_python-0.4.3 → agentkit_sdk_python-0.5.0}/agentkit/sdk/mcp/types.py

@@ -63,10 +63,22 @@ class AuthorizerConfigurationForListMCPToolsets(MCPBaseModel):
 
 
 class AuthorizerForGetMCPServiceForInboundAuthorizerConfiguration(MCPBaseModel):
+    custom_jwt_authorizer: Optional[CustomJwtAuthorizerForGetMCPService] = Field(
+        default=None, alias="CustomJwtAuthorizer"
+    )
+    identity_authorizer: Optional[IdentityAuthorizerForGetMCPService] = Field(
+        default=None, alias="IdentityAuthorizer"
+    )
     key_auth: Optional[KeyAuthForGetMCPService] = Field(default=None, alias="KeyAuth")
 
 
 class AuthorizerForGetMCPServiceForOutboundAuthorizerConfiguration(MCPBaseModel):
+    custom_jwt_authorizer: Optional[CustomJwtAuthorizerForGetMCPService] = Field(
+        default=None, alias="CustomJwtAuthorizer"
+    )
+    identity_authorizer: Optional[IdentityAuthorizerForGetMCPService] = Field(
+        default=None, alias="IdentityAuthorizer"
+    )
     key_auth: Optional[KeyAuthForGetMCPService] = Field(default=None, alias="KeyAuth")
 
 
@@ -91,9 +103,15 @@ class BackendConfigurationForGetMCPService(MCPBaseModel):
     custom_mcp_configuration: Optional[CustomMcpConfigurationForGetMCPService] = Field(
         default=None, alias="CustomMcpConfiguration"
     )
+    ecs_configuration: Optional[EcsConfigurationForGetMCPService] = Field(
+        default=None, alias="EcsConfiguration"
+    )
     function_configuration: Optional[FunctionConfigurationForGetMCPService] = Field(
         default=None, alias="FunctionConfiguration"
     )
+    vke_configuration: Optional[VkeConfigurationForGetMCPService] = Field(
+        default=None, alias="VkeConfiguration"
+    )
 
 
 class CustomConfigurationForGetMCPService(MCPBaseModel):
@@ -102,6 +120,11 @@ class CustomConfigurationForGetMCPService(MCPBaseModel):
     protocol_type: Optional[str] = Field(default=None, alias="ProtocolType")
 
 
+class CustomJwtAuthorizerForGetMCPService(MCPBaseModel):
+    allowed_clients: Optional[list[str]] = Field(default=None, alias="AllowedClients")
+    discovery_url: Optional[str] = Field(default=None, alias="DiscoveryUrl")
+
+
 class CustomJwtAuthorizerForGetMCPToolset(MCPBaseModel):
     allowed_clients: Optional[list[str]] = Field(default=None, alias="AllowedClients")
     discovery_url: Optional[str] = Field(default=None, alias="DiscoveryUrl")
@@ -124,6 +147,12 @@ class CustomMcpConfigurationForGetMCPService(MCPBaseModel):
     )
 
 
+class EcsConfigurationForGetMCPService(MCPBaseModel):
+    instances: Optional[list[InstancesForGetMCPService]] = Field(
+        default=None, alias="Instances"
+    )
+
+
 class EnvsForGetMCPService(MCPBaseModel):
     key: Optional[str] = Field(default=None, alias="Key")
     value: Optional[str] = Field(default=None, alias="Value")
@@ -134,6 +163,13 @@ class FunctionConfigurationForGetMCPService(MCPBaseModel):
     function_name: Optional[str] = Field(default=None, alias="FunctionName")
 
 
+class IdentityAuthorizerForGetMCPService(MCPBaseModel):
+    credential_provider_name: Optional[str] = Field(
+        default=None, alias="CredentialProviderName"
+    )
+    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
+
+
 class InboundAuthorizerConfigurationForGetMCPService(MCPBaseModel):
     authorizer: Optional[
         AuthorizerForGetMCPServiceForInboundAuthorizerConfiguration
@@ -141,6 +177,12 @@ class InboundAuthorizerConfigurationForGetMCPService(MCPBaseModel):
     authorizer_type: Optional[str] = Field(default=None, alias="AuthorizerType")
 
 
+class InstancesForGetMCPService(MCPBaseModel):
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+    ip: Optional[str] = Field(default=None, alias="Ip")
+    port: Optional[int] = Field(default=None, alias="Port")
+
+
 class KeyAuthForGetMCPService(MCPBaseModel):
     api_key_location: Optional[str] = Field(default=None, alias="ApiKeyLocation")
     api_keys: Optional[list[ApiKeysForGetMCPService]] = Field(
@@ -171,6 +213,7 @@ class MCPServiceForGetMCPService(MCPBaseModel):
     )
     backend_type: Optional[str] = Field(default=None, alias="BackendType")
     created_at: Optional[str] = Field(default=None, alias="CreatedAt")
+    failed_log_file_url: Optional[str] = Field(default=None, alias="FailedLogFileUrl")
     inbound_authorizer_configuration: Optional[
         InboundAuthorizerConfigurationForGetMCPService
     ] = Field(default=None, alias="InboundAuthorizerConfiguration")
@@ -416,17 +459,33 @@ class TagsForListMCPToolsets(MCPBaseModel):
     value: Optional[str] = Field(default=None, alias="Value")
 
 
+class VkeConfigurationForGetMCPService(MCPBaseModel):
+    cluster_id: Optional[str] = Field(default=None, alias="ClusterId")
+    namespace: Optional[str] = Field(default=None, alias="Namespace")
+    port: Optional[int] = Field(default=None, alias="Port")
+    service_name: Optional[str] = Field(default=None, alias="ServiceName")
+
+
 class VpcConfigurationForGetMCPService(MCPBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
     vpc_id: Optional[str] = Field(default=None, alias="VpcId")
 
 
 class VpcConfigurationForGetMCPToolset(MCPBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
     vpc_id: Optional[str] = Field(default=None, alias="VpcId")
 
 
 class VpcConfigurationForListMCPServices(MCPBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )
@@ -435,6 +494,9 @@ class VpcConfigurationForListMCPServices(MCPBaseModel):
 
 
 class VpcConfigurationForListMCPToolsets(MCPBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )
@@ -450,9 +512,15 @@ class BackendForCreateMCPService(MCPBaseModel):
     custom_mcp_configuration: Optional[BackendCustomMcpForCreateMCPService] = Field(
         default=None, alias="CustomMcpConfiguration"
     )
+    ecs_configuration: Optional[BackendEcsForCreateMCPService] = Field(
+        default=None, alias="EcsConfiguration"
+    )
     function_configuration: Optional[BackendFunctionForCreateMCPService] = Field(
         default=None, alias="FunctionConfiguration"
     )
+    vke_configuration: Optional[BackendVkeForCreateMCPService] = Field(
+        default=None, alias="VkeConfiguration"
+    )
 
 
 class BackendCustomForCreateMCPService(MCPBaseModel):
@@ -496,11 +564,24 @@ class BackendCustomMcpPublicPackageForCreateMCPService(MCPBaseModel):
     raw_config: Optional[str] = Field(default=None, alias="RawConfig")
 
 
+class BackendEcsForCreateMCPService(MCPBaseModel):
+    instances: Optional[list[BackendEcsInstancesItemForCreateMCPService]] = Field(
+        default=None, alias="Instances"
+    )
+
+
 class BackendFunctionForCreateMCPService(MCPBaseModel):
     function_id: Optional[str] = Field(default=None, alias="FunctionId")
     function_name: Optional[str] = Field(default=None, alias="FunctionName")
 
 
+class BackendVkeForCreateMCPService(MCPBaseModel):
+    cluster_id: Optional[str] = Field(default=None, alias="ClusterId")
+    namespace: Optional[str] = Field(default=None, alias="Namespace")
+    port: Optional[int] = Field(default=None, alias="Port")
+    service_name: Optional[str] = Field(default=None, alias="ServiceName")
+
+
 class InboundAuthorizerForCreateMCPService(MCPBaseModel):
     authorizer: Optional[InboundAuthorizerAuthorizerForCreateMCPService] = Field(
         default=None, alias="Authorizer"
@@ -543,6 +624,9 @@ class NetworkForCreateMCPService(MCPBaseModel):
 
 
 class NetworkVpcForCreateMCPService(MCPBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
     vpc_id: str = Field(..., alias="VpcId")
 
@@ -555,11 +639,21 @@ class OutboundAuthorizerForCreateMCPService(MCPBaseModel):
 
 
 class OutboundAuthorizerAuthorizerForCreateMCPService(MCPBaseModel):
+    identity_authorizer: Optional[
+        OutboundAuthorizerAuthorizerIdentityAuthorizerForCreateMCPService
+    ] = Field(default=None, alias="IdentityAuthorizer")
     key_auth: Optional[OutboundAuthorizerAuthorizerKeyAuthForCreateMCPService] = Field(
         default=None, alias="KeyAuth"
     )
 
 
+class OutboundAuthorizerAuthorizerIdentityAuthorizerForCreateMCPService(MCPBaseModel):
+    credential_provider_name: Optional[str] = Field(
+        default=None, alias="CredentialProviderName"
+    )
+    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
+
+
 class OutboundAuthorizerAuthorizerKeyAuthForCreateMCPService(MCPBaseModel):
     api_keys: Optional[
         list[OutboundAuthorizerAuthorizerKeyAuthApiKeysItemForCreateMCPService]
@@ -584,6 +678,12 @@ class BackendCustomMcpPrivatePackageEnvsItemForCreateMCPService(MCPBaseModel):
     value: Optional[str] = Field(default=None, alias="Value")
 
 
+class BackendEcsInstancesItemForCreateMCPService(MCPBaseModel):
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+    ip: Optional[str] = Field(default=None, alias="Ip")
+    port: Optional[int] = Field(default=None, alias="Port")
+
+
 class InboundAuthorizerAuthorizerKeyAuthApiKeysItemForCreateMCPService(MCPBaseModel):
     key: Optional[str] = Field(default=None, alias="Key")
     name: str = Field(..., alias="Name")
@@ -674,6 +774,9 @@ class NetworkForCreateMCPToolset(MCPBaseModel):
 
 
 class NetworkVpcForCreateMCPToolset(MCPBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )
@@ -855,9 +958,15 @@ class BackendForUpdateMCPService(MCPBaseModel):
     custom_mcp_configuration: Optional[BackendCustomMcpForUpdateMCPService] = Field(
         default=None, alias="CustomMcpConfiguration"
     )
+    ecs_configuration: Optional[BackendEcsForUpdateMCPService] = Field(
+        default=None, alias="EcsConfiguration"
+    )
     function_configuration: Optional[BackendFunctionForUpdateMCPService] = Field(
         default=None, alias="FunctionConfiguration"
     )
+    vke_configuration: Optional[BackendVkeForUpdateMCPService] = Field(
+        default=None, alias="VkeConfiguration"
+    )
 
 
 class BackendCustomForUpdateMCPService(MCPBaseModel):
@@ -901,11 +1010,24 @@ class BackendCustomMcpPublicPackageForUpdateMCPService(MCPBaseModel):
     raw_config: Optional[str] = Field(default=None, alias="RawConfig")
 
 
+class BackendEcsForUpdateMCPService(MCPBaseModel):
+    instances: Optional[list[BackendEcsInstancesItemForUpdateMCPService]] = Field(
+        default=None, alias="Instances"
+    )
+
+
 class BackendFunctionForUpdateMCPService(MCPBaseModel):
     function_id: Optional[str] = Field(default=None, alias="FunctionId")
     function_name: Optional[str] = Field(default=None, alias="FunctionName")
 
 
+class BackendVkeForUpdateMCPService(MCPBaseModel):
+    cluster_id: Optional[str] = Field(default=None, alias="ClusterId")
+    namespace: Optional[str] = Field(default=None, alias="Namespace")
+    port: Optional[int] = Field(default=None, alias="Port")
+    service_name: Optional[str] = Field(default=None, alias="ServiceName")
+
+
 class InboundAuthorizerForUpdateMCPService(MCPBaseModel):
     authorizer: Optional[InboundAuthorizerAuthorizerForUpdateMCPService] = Field(
         default=None, alias="Authorizer"
@@ -943,11 +1065,21 @@ class OutboundAuthorizerForUpdateMCPService(MCPBaseModel):
 
 
 class OutboundAuthorizerAuthorizerForUpdateMCPService(MCPBaseModel):
+    identity_authorizer: Optional[
+        OutboundAuthorizerAuthorizerIdentityAuthorizerForUpdateMCPService
+    ] = Field(default=None, alias="IdentityAuthorizer")
     key_auth: Optional[OutboundAuthorizerAuthorizerKeyAuthForUpdateMCPService] = Field(
         default=None, alias="KeyAuth"
     )
 
 
+class OutboundAuthorizerAuthorizerIdentityAuthorizerForUpdateMCPService(MCPBaseModel):
+    credential_provider_name: Optional[str] = Field(
+        default=None, alias="CredentialProviderName"
+    )
+    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
+
+
 class OutboundAuthorizerAuthorizerKeyAuthForUpdateMCPService(MCPBaseModel):
     api_keys: Optional[
         list[OutboundAuthorizerAuthorizerKeyAuthApiKeysItemForUpdateMCPService]
@@ -961,6 +1093,12 @@ class BackendCustomMcpPrivatePackageEnvsItemForUpdateMCPService(MCPBaseModel):
     value: Optional[str] = Field(default=None, alias="Value")
 
 
+class BackendEcsInstancesItemForUpdateMCPService(MCPBaseModel):
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+    ip: Optional[str] = Field(default=None, alias="Ip")
+    port: Optional[int] = Field(default=None, alias="Port")
+
+
 class InboundAuthorizerAuthorizerKeyAuthApiKeysItemForUpdateMCPService(MCPBaseModel):
     key: Optional[str] = Field(default=None, alias="Key")
     name: str = Field(..., alias="Name")

{agentkit_sdk_python-0.4.3 → agentkit_sdk_python-0.5.0}/agentkit/sdk/memory/types.py

@@ -140,6 +140,9 @@ class VpcConfigForGetMemoryCollection(MemoryBaseModel):
 
 
 class VpcConfigurationForListMemoryCollections(MemoryBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )

{agentkit_sdk_python-0.4.3 → agentkit_sdk_python-0.5.0}/agentkit/sdk/runtime/types.py

@@ -166,11 +166,12 @@ class EnvsForListRuntimes(RuntimeTypeBaseModel):
 
 
 class InstanceItemsForListRuntimeInstances(RuntimeTypeBaseModel):
+    alias_name: Optional[str] = Field(default=None, alias="AliasName")
     create_time: Optional[str] = Field(default=None, alias="CreateTime")
-    function_id: Optional[str] = Field(default=None, alias="FunctionId")
     instance_name: Optional[str] = Field(default=None, alias="InstanceName")
     instance_status: Optional[str] = Field(default=None, alias="InstanceStatus")
     revision_number: Optional[int] = Field(default=None, alias="RevisionNumber")
+    runtime_id: Optional[str] = Field(default=None, alias="RuntimeId")
 
 
 class KeyAuthForGetRuntime(RuntimeTypeBaseModel):
@@ -241,6 +242,9 @@ class TlsConfigurationForListRuntimes(RuntimeTypeBaseModel):
 
 
 class VpcConfigurationForGetRuntime(RuntimeTypeBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )
@@ -249,6 +253,9 @@ class VpcConfigurationForGetRuntime(RuntimeTypeBaseModel):
 
 
 class VpcConfigurationForListRuntimes(RuntimeTypeBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     security_group_ids: Optional[list[str]] = Field(
         default=None, alias="SecurityGroupIds"
     )
@@ -289,6 +296,9 @@ class NetworkForCreateRuntime(RuntimeTypeBaseModel):
 
 
 class NetworkVpcForCreateRuntime(RuntimeTypeBaseModel):
+    enable_shared_internet_access: Optional[bool] = Field(
+        default=None, alias="EnableSharedInternetAccess"
+    )
     subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
     vpc_id: str = Field(..., alias="VpcId")
 
@@ -415,9 +425,9 @@ class GetRuntimeCozeTokenResponse(RuntimeTypeBaseModel):
 
 # GetRuntimeInstanceLogs - Request
 class GetRuntimeInstanceLogsRequest(RuntimeTypeBaseModel):
-    function_id: str = Field(..., alias="FunctionId")
     instance_name: str = Field(..., alias="InstanceName")
     limit: Optional[int] = Field(default=None, alias="Limit")
+    runtime_id: str = Field(..., alias="RuntimeId")
 
 
 # GetRuntimeInstanceLogs - Response