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

agentkit_sdk_python-0.1.6/MANIFEST.in

@@ -0,0 +1,2 @@
+# Include all resource files recursively
+recursive-include agentkit/toolkit/resources *

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.1.5
+Version: 0.1.6
 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>
 License:                                  Apache License

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

@@ -26,6 +26,8 @@ from starlette.responses import JSONResponse, Response, StreamingResponse
 
 from agentkit.apps.simple_app.telemetry import telemetry
 from agentkit.apps.utils import safe_serialize_to_json_string
+from opentelemetry import context as context_api
+from opentelemetry import trace
 
 logger = logging.getLogger("agentkit." + __name__)
 
@@ -45,37 +47,41 @@ class InvokeHandler(BaseHandler):
             logger.error("Invoke handler function is not set")
             return Response(status_code=404)
 
-        with telemetry.tracer.start_as_current_span(name="invocation") as span:
-            exception = None
-            try:
-                payload, headers, result = await self._process_invoke(request)
-                if inspect.isgenerator(result):
-                    logger.info("Returning streaming response (generator)")
-                    return StreamingResponse(
-                        self._sync_stream_with_error_handling(result),
-                        media_type="text/event-stream",
-                    )
-                elif inspect.isasyncgen(result):
-                    logger.info("Returning streaming response (async generator)")
-                    return StreamingResponse(
-                        self._stream_with_error_handling(result),
-                        media_type="text/event-stream",
-                    )
-
-                safe_json_string = safe_serialize_to_json_string(result)
-            except Exception as e:
-                logger.error("Invoke handler function failed: %s", e)
-                exception = e
-                raise e
-            finally:
-                # TODO: error with stream
-                # handler trace span and metrics
-                try:
-                    telemetry.trace_agent(
-                        self.func, span, payload, headers, safe_json_string, exception
-                    )
-                except Exception as e:
-                    _ = e
+        span = telemetry.tracer.start_span(name="agentkit_invocation")
+        ctx = trace.set_span_in_context(span)
+        context_api.attach(ctx)
+
+        try:
+            payload, headers, result = await self._process_invoke(request)
+
+            # set trace attributes
+            telemetry.trace_agent(self.func, span, payload, headers)
+
+            if inspect.isgenerator(result):
+                logger.info("Returning streaming response (generator)")
+                return StreamingResponse(
+                    self._sync_stream_with_error_handling(result),
+                    media_type="text/event-stream",
+                )
+
+            elif inspect.isasyncgen(result):
+                logger.info("Returning streaming response (async generator)")
+                return StreamingResponse(
+                    self._stream_with_error_handling(result),
+                    media_type="text/event-stream",
+                )
+
+            logger.info("Returning non-streaming response")
+            safe_json_string = safe_serialize_to_json_string(result)
+            telemetry.trace_agent_finish(
+                safe_json_string, None
+            )
+        except Exception as e:
+            logger.error("Invoke handler function failed: %s", e)
+            telemetry.trace_agent_finish(
+                "", e
+            )
+            raise e
 
         return Response(safe_json_string, media_type="application/json")
 
@@ -133,10 +139,14 @@ class InvokeHandler(BaseHandler):
 
     def _sync_stream_with_error_handling(self, generator):
         """Wrap sync generator to handle errors and convert to SSE format."""
+        exception = None
+        last_value = ""
         try:
             for value in generator:
+                last_value = value
                 yield self._convert_to_sse(value)
         except Exception as e:
+            exception = e
             logger.exception("Error in sync streaming")
             error_event = {
                 "error": str(e),
@@ -144,13 +154,23 @@ class InvokeHandler(BaseHandler):
                 "message": "An error occurred during streaming",
             }
             yield self._convert_to_sse(error_event)
+        finally:
+            # finish trace span and record metrics
+            result = safe_serialize_to_json_string(last_value)
+            telemetry.trace_agent_finish(
+                result, exception
+            )
 
     async def _stream_with_error_handling(self, generator):
         """Wrap async generator to handle errors and convert to SSE format."""
+        exception = None
+        last_value = ""
         try:
             async for value in generator:
+                last_value = value
                 yield self._convert_to_sse(value)
         except Exception as e:
+            exception = e
             logger.exception("Error in async streaming")
             error_event = {
                 "error": str(e),
@@ -158,6 +178,12 @@ class InvokeHandler(BaseHandler):
                 "message": "An error occurred during streaming",
             }
             yield self._convert_to_sse(error_event)
+        finally:
+            # finish trace span and record metrics
+            result = safe_serialize_to_json_string(last_value)
+            telemetry.trace_agent_finish(
+                result, exception
+            )
 
 
 class PingHandler(BaseHandler):

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

@@ -15,6 +15,7 @@
 
 import logging
 import time
+import traceback
 from typing import Callable
 
 from opentelemetry import trace
@@ -45,6 +46,28 @@ _GEN_AI_CLIENT_OPERATION_DURATION_BUCKETS = [
 
 logger = logging.getLogger("agentkit." + __name__)
 
+
+def dont_throw(func):
+    """
+    A decorator that wraps the passed in function and logs exceptions instead of throwing them.
+
+    @param func: The function to wrap
+    @return: The wrapper function
+    """
+
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception as e:
+            logger.error(
+                "Agentkit failed to trace in %s, error: %s",
+                func.__name__,
+                traceback.format_exc(),
+            )
+
+    return wrapper
+
+
 class Telemetry:
     def __init__(self):
         self.tracer = get_tracer("agentkit.simple_app")
@@ -56,14 +79,13 @@ class Telemetry:
             explicit_bucket_boundaries_advisory=_GEN_AI_CLIENT_OPERATION_DURATION_BUCKETS,
         )
 
+    @dont_throw
     def trace_agent(
         self,
         func: Callable,
         span: Span,
         payload: dict,
         headers: dict,
-        func_result: str,
-        exception: Exception,
     ) -> None:
         """Get current span and set required attributes."""
 
@@ -92,22 +114,30 @@ class Telemetry:
 
         span.set_attribute(key="gen_ai.input", value=safe_serialize_to_json_string(payload))
 
-        span.set_attribute(key="gen_ai.output", value=func_result)
-
         span.set_attribute(key="gen_ai.span.kind", value="workflow")
         span.set_attribute(key="gen_ai.operation.name", value="invoke_agent")
         span.set_attribute(key="gen_ai.operation.type", value="agent")
-        attributes={"gen_ai_operation_name": "invoke_agent", "gen_ai_operation_type": "agent"}
-
-        if exception:
-            self.handle_exception(span, exception)
-            attributes["error_type"] = exception.__class__.__name__
-
-        # record latency metrics
-        if hasattr(span, "start_time") and self.latency_histogram:
-            duration = (time.time_ns() - span.start_time) / 1e9  # type: ignore
-            self.latency_histogram.record(duration, attributes)
 
+    @dont_throw
+    def trace_agent_finish(
+        self,
+        func_result: str,
+        exception: Exception,
+    ) -> None:
+        span = trace.get_current_span()
+
+        if span and span.is_recording():
+            span.set_attribute(key="gen_ai.output", value=func_result)
+            attributes={"gen_ai_operation_name": "invoke_agent", "gen_ai_operation_type": "agent"}
+            if exception:
+                self.handle_exception(span, exception)
+                attributes["error_type"] = exception.__class__.__name__
+
+            # record latency metrics
+            if hasattr(span, "start_time") and self.latency_histogram:
+                duration = (time.time_ns() - span.start_time) / 1e9  # type: ignore
+                self.latency_histogram.record(duration, attributes)
+            span.end()
 
     @staticmethod
     def handle_exception(span: trace.Span, exception: Exception) -> None:

{agentkit_sdk_python-0.1.5 → agentkit_sdk_python-0.1.6}/agentkit/client/base_client.py

@@ -113,8 +113,8 @@ class BaseAgentkitClient(Service):
                 region=self.region,
                 session_token=self.session_token,
             ),
-            connection_timeout=5,
-            socket_timeout=5,
+            connection_timeout=30,
+            socket_timeout=30,
             scheme="https",
         )
         
@@ -191,7 +191,6 @@ class BaseAgentkitClient(Service):
             Exception: If API call fails or returns an error
         """
         # Make API call
-        print(f"request action: {api_action}, body: {json.dumps(request.model_dump(by_alias=True, exclude_none=True))}")
         try:
             res = self.json(
                 api=api_action,
@@ -207,7 +206,6 @@ class BaseAgentkitClient(Service):
         # Parse response
         response_data = json.loads(res)
         
-        print("response_data:", response_data)
         # Check for errors
         metadata = response_data.get("ResponseMetadata", {})
         if metadata.get("Error"):

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

@@ -1,3 +1,17 @@
+# 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.
+
 import asyncio
 import os
 from functools import wraps

agentkit_sdk_python-0.1.6/agentkit/knowledge/knowledge.py

@@ -0,0 +1,108 @@
+# 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.
+
+from typing import Dict
+
+from agentkit.client import BaseAgentkitClient
+from agentkit.utils import get_logger
+
+# 导入自动生成的类型
+from agentkit.knowledge.knowledge_all_types import (
+    ListKnowledgeBasesRequest,
+    ListKnowledgeBasesResponse,
+    AddKnowledgeBaseRequest,
+    AddKnowledgeBaseResponse,
+    GetKnowledgeConnectionInfoRequest,
+    GetKnowledgeConnectionInfoResponse,
+    DeleteKnowledgeBaseRequest,
+    DeleteKnowledgeBaseResponse,
+)
+
+logger = get_logger(__name__)
+
+
+class AgentkitKnowledge(BaseAgentkitClient):
+    """AgentKit Knowledge Base Management Service"""
+    
+    # Define all API actions for this service
+    API_ACTIONS: Dict[str, str] = {
+        "ListKnowledgeBases": "ListKnowledgeBases",
+        "AddKnowledgeBase": "AddKnowledgeBase",
+        "GetKnowledgeConnectionInfo": "GetKnowledgeConnectionInfo",
+        "DeleteKnowledgeBase": "DeleteKnowledgeBase",
+    }
+    
+    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="knowledge",
+        )
+
+    def list_knowledge_bases(
+        self, request: ListKnowledgeBasesRequest
+    ) -> ListKnowledgeBasesResponse:
+        """List all knowledge bases from Volcengine."""
+        return self._invoke_api(
+            api_action="ListKnowledgeBases",
+            request=request,
+            response_type=ListKnowledgeBasesResponse,
+        )
+
+    def add_knowledge_base(
+        self, request: AddKnowledgeBaseRequest
+    ) -> AddKnowledgeBaseResponse:
+        """Add a new knowledge base to Volcengine."""
+        return self._invoke_api(
+            api_action="AddKnowledgeBase",
+            request=request,
+            response_type=AddKnowledgeBaseResponse,
+        )
+
+    def get_knowledge_connection_info(
+        self, request: GetKnowledgeConnectionInfoRequest
+    ) -> GetKnowledgeConnectionInfoResponse:
+        """Get connection information for a specific knowledge base."""
+        return self._invoke_api(
+            api_action="GetKnowledgeConnectionInfo",
+            request=request,
+            response_type=GetKnowledgeConnectionInfoResponse,
+        )
+
+    def delete_knowledge_base(
+        self, request: DeleteKnowledgeBaseRequest
+    ) -> DeleteKnowledgeBaseResponse:
+        """Delete a knowledge base from Volcengine."""
+        return self._invoke_api(
+            api_action="DeleteKnowledgeBase",
+            request=request,
+            response_type=DeleteKnowledgeBaseResponse,
+        )
+
+
+if __name__ == "__main__":
+    # list knowledge bases
+    knowledge = AgentkitKnowledge()
+    req = ListKnowledgeBasesRequest(page_number=1, page_size=10)
+    res = knowledge.list_knowledge_bases(req)
+    print(res)
+    

{agentkit_sdk_python-0.1.5 → agentkit_sdk_python-0.1.6}/agentkit/knowledge/knowledge_all_types.py

@@ -22,24 +22,9 @@ from pydantic import BaseModel, Field
 
 
 # Data Types
-class KnowledgeBasesForListKnowledgeBases(BaseModel):
-    create_time: Optional[str] = Field(default=None, alias="CreateTime")
-    description: Optional[str] = Field(default=None, alias="Description")
-    knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
-    last_update_time: Optional[str] = Field(default=None, alias="LastUpdateTime")
-    name: Optional[str] = Field(default=None, alias="Name")
-    provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId")
-    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
-    status: Optional[str] = Field(default=None, alias="Status")
-    
-    model_config = {
-        "populate_by_name": True,
-        "arbitrary_types_allowed": True
-    }
-
-
 class KnowledgeBasesForAddKnowledgeBase(BaseModel):
     knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
+    message: Optional[str] = Field(default=None, alias="Message")
     provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId")
     provider_type: Optional[str] = Field(default=None, alias="ProviderType")
     status: Optional[str] = Field(default=None, alias="Status")
@@ -57,8 +42,6 @@ class ConnectionInfosForGetKnowledgeConnectionInfo(BaseModel):
     expire_at: Optional[str] = Field(default=None, alias="ExpireAt")
     extra_config: Optional[str] = Field(default=None, alias="ExtraConfig")
     region: Optional[str] = Field(default=None, alias="Region")
-    vpc_id: Optional[str] = Field(default=None, alias="VpcId")
-    vpc_name: Optional[str] = Field(default=None, alias="VpcName")
     
     model_config = {
         "populate_by_name": True,
@@ -66,41 +49,15 @@ class ConnectionInfosForGetKnowledgeConnectionInfo(BaseModel):
     }
 
 
-# ListKnowledgeBases - Request
-class FiltersItem(BaseModel):
+class KnowledgeBasesForListKnowledgeBases(BaseModel):
+    create_time: Optional[str] = Field(default=None, alias="CreateTime")
+    description: Optional[str] = Field(default=None, alias="Description")
+    knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
+    last_update_time: Optional[str] = Field(default=None, alias="LastUpdateTime")
     name: Optional[str] = Field(default=None, alias="Name")
-    name_contains: Optional[str] = Field(default=None, alias="NameContains")
-    values: Optional[list[str]] = Field(default=None, alias="Values")
-    
-    model_config = {
-        "populate_by_name": True,
-        "arbitrary_types_allowed": True
-    }
-
-
-class ListKnowledgeBasesRequest(BaseModel):
-    create_time_after: Optional[str] = Field(default=None, alias="CreateTimeAfter")
-    create_time_before: Optional[str] = Field(default=None, alias="CreateTimeBefore")
-    max_results: Optional[int] = Field(default=None, alias="MaxResults")
-    next_token: Optional[str] = Field(default=None, alias="NextToken")
-    page_number: Optional[int] = Field(default=None, alias="PageNumber")
-    page_size: Optional[int] = Field(default=None, alias="PageSize")
-    update_time_after: Optional[str] = Field(default=None, alias="UpdateTimeAfter")
-    update_time_before: Optional[str] = Field(default=None, alias="UpdateTimeBefore")
-    filters: Optional[list[FiltersItem]] = Field(default=None, alias="Filters")
-    
-    model_config = {
-        "populate_by_name": True,
-        "arbitrary_types_allowed": True
-    }
-
-
-# ListKnowledgeBases - Response
-class ListKnowledgeBasesResponse(BaseModel):
-    knowledge_bases: Optional[list[KnowledgeBasesForListKnowledgeBases]] = Field(default=None, alias="KnowledgeBases")
-    page_number: Optional[int] = Field(default=None, alias="PageNumber")
-    page_size: Optional[int] = Field(default=None, alias="PageSize")
-    total_count: Optional[int] = Field(default=None, alias="TotalCount")
+    provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId")
+    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
+    status: Optional[str] = Field(default=None, alias="Status")
     
     model_config = {
         "populate_by_name": True,
@@ -110,10 +67,10 @@ class ListKnowledgeBasesResponse(BaseModel):
 
 # AddKnowledgeBase - Request
 class KnowledgeBasesItem(BaseModel):
-    name: str = Field(..., alias="Name")
-    provider_knowledge_id: str = Field(..., alias="ProviderKnowledgeId")
-    provider_type: str = Field(..., alias="ProviderType")
-    description: Optional[str] = Field(default=None, alias="Description")
+    name: str = Field(..., alias="Name", description="待导入知识库名称。 - 参数 - N：表示知识库的序号，取值范围：1～10。 - 多个字段名之间用&分隔。 取值： - Viking知识库需填入CollectionName。")
+    provider_knowledge_id: str = Field(..., alias="ProviderKnowledgeId", description="待导入知识库ID。 - 参数 - N：表示知识库的序号，取值范围：1～10。 - 多个字段名之间用&分隔。 取值： - Viking知识库需使用ResourceId。")
+    provider_type: str = Field(..., alias="ProviderType", description="待导入知识库平台类型。 - 参数 - N：表示知识库的序号，取值范围：1～10。 - 多个字段名之间用&分隔。 取值： - VIKINGDB_KNOWLEDGE：Viking知识库。")
+    description: Optional[str] = Field(default=None, alias="Description", description="待导入知识库的描述。 - 参数 - N：表示知识库的序号，取值范围：1～10。 - 不填默认为空。 - 多个字段名之间用&分隔。")
     
     model_config = {
         "populate_by_name": True,
@@ -122,8 +79,8 @@ class KnowledgeBasesItem(BaseModel):
 
 
 class AddKnowledgeBaseRequest(BaseModel):
-    client_token: Optional[str] = Field(default=None, alias="ClientToken")
-    project_name: Optional[str] = Field(default=None, alias="ProjectName")
+    client_token: Optional[str] = Field(default=None, alias="ClientToken", description="保证请求幂等性。由客户端自动生成一个参数值，确保不同请求间该参数值唯一，避免当调用API超时或服务器内部错误时，客户端多次重试导致重复性操作。 取值：仅支持ASCII字符，且不能超过64个字符。")
+    project_name: Optional[str] = Field(default=None, alias="ProjectName", description="知识库所属项目名称，不填默认为default。")
     knowledge_bases: Optional[list[KnowledgeBasesItem]] = Field(default=None, alias="KnowledgeBases")
     
     model_config = {
@@ -134,7 +91,7 @@ class AddKnowledgeBaseRequest(BaseModel):
 
 # AddKnowledgeBase - Response
 class AddKnowledgeBaseResponse(BaseModel):
-    knowledge_bases: Optional[list[KnowledgeBasesForAddKnowledgeBase]] = Field(default=None, alias="KnowledgeBases")
+    knowledge_bases: Optional[list[KnowledgeBasesForAddKnowledgeBase]] = Field(default=None, alias="KnowledgeBases", description="知识库信息。")
     
     model_config = {
         "populate_by_name": True,
@@ -144,7 +101,7 @@ class AddKnowledgeBaseResponse(BaseModel):
 
 # GetKnowledgeConnectionInfo - Request
 class GetKnowledgeConnectionInfoRequest(BaseModel):
-    knowledge_id: str = Field(..., alias="KnowledgeId")
+    knowledge_id: str = Field(..., alias="KnowledgeId", description="知识库ID。")
     
     model_config = {
         "populate_by_name": True,
@@ -154,12 +111,14 @@ class GetKnowledgeConnectionInfoRequest(BaseModel):
 
 # GetKnowledgeConnectionInfo - Response
 class GetKnowledgeConnectionInfoResponse(BaseModel):
-    connection_infos: Optional[list[ConnectionInfosForGetKnowledgeConnectionInfo]] = Field(default=None, alias="ConnectionInfos")
-    error_message: Optional[str] = Field(default=None, alias="ErrorMessage")
-    knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
-    provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId")
-    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
-    status: Optional[str] = Field(default=None, alias="Status")
+    connection_infos: Optional[list[ConnectionInfosForGetKnowledgeConnectionInfo]] = Field(default=None, alias="ConnectionInfos", description="知识库的连接信息。")
+    knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId", description="知识库ID。")
+    message: Optional[str] = Field(default=None, alias="Message", description="状态消息，包含额外的状态信息或错误描述。")
+    name: Optional[str] = Field(default=None, alias="Name", description="知识库名称。")
+    project_name: Optional[str] = Field(default=None, alias="ProjectName", description="已导入的知识库所属项目名称。")
+    provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId", description="已导入的知识库所属项目ID。")
+    provider_type: Optional[str] = Field(default=None, alias="ProviderType", description="已导入的知识库平台类型。取值： - VIKINGDB_KNOWLEDGE：Viking知识库。")
+    status: Optional[str] = Field(default=None, alias="Status", description="知识库状态，取值： - Importing：导入中。 - ImportFailed：导入失败。 - Ready：成功。 - Deleting：删除中。 - DeleteFailed：删除失败。 - Deleted：已删除。 - Error：错误。")
     
     model_config = {
         "populate_by_name": True,
@@ -169,7 +128,7 @@ class GetKnowledgeConnectionInfoResponse(BaseModel):
 
 # DeleteKnowledgeBase - Request
 class DeleteKnowledgeBaseRequest(BaseModel):
-    knowledge_id: str = Field(..., alias="KnowledgeId")
+    knowledge_id: str = Field(..., alias="KnowledgeId", description="知识库ID。")
     
     model_config = {
         "populate_by_name": True,
@@ -179,10 +138,49 @@ class DeleteKnowledgeBaseRequest(BaseModel):
 
 # DeleteKnowledgeBase - Response
 class DeleteKnowledgeBaseResponse(BaseModel):
-    knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
-    provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId")
-    provider_type: Optional[str] = Field(default=None, alias="ProviderType")
-    request_id: Optional[str] = Field(default=None, alias="RequestId")
+    knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId", description="知识库ID。")
+    provider_knowledge_id: Optional[str] = Field(default=None, alias="ProviderKnowledgeId", description="导入的知识库ID。")
+    provider_type: Optional[str] = Field(default=None, alias="ProviderType", description="导入的知识库平台类型。取值： - VIKINGDB_KNOWLEDGE：Viking知识库。")
+    
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+# ListKnowledgeBases - Request
+class FiltersItem(BaseModel):
+    name: Optional[str] = Field(default=None, alias="Name", description="精确过滤的字段名。 - 参数 - N：表示过滤条件的序号，取值范围：1～10。 - 多个字段名之间用&分隔，为“或”的关系。 - 取值： - knowledge_id：知识库ID。 - name：知识库名称。 - status：知识库状态。 - provider_type：导入的知识库平台类型。 ::: tip 当`Filters.N.Name`和`Filters.N.NameContains`同时传入时，仅`Filters.N.Name`生效。 :::")
+    name_contains: Optional[str] = Field(default=None, alias="NameContains", description="模糊过滤的字段名。 - 参数 - N：表示过滤条件的序号，取值范围：1～10。 - 多个字段名之间用&分隔，为“或”的关系。 - 取值： - knowledge_id：知识库ID。 - name：知识库名称。 - status：知识库状态。 - provider_type：导入的知识库平台类型。 ::: tip 当`Filters.N.Name`和`Filters.N.NameContains`同时传入时，仅`Filters.N.Name`生效。 :::")
+    values: Optional[list[str]] = Field(default=None, alias="Values", description="过滤字段的值。 - 参数 - N：表示过滤条件的序号，取值范围：1～10。 - 多个字段名之间用&分隔。 当过滤字段为provider_type时，取值： - VIKINGDB_KNOWLEDGE：Viking知识库。")
+    
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class ListKnowledgeBasesRequest(BaseModel):
+    create_time_after: Optional[str] = Field(default=None, alias="CreateTimeAfter", description="筛选该时间点后创建的知识库。（RFC3339格式）")
+    create_time_before: Optional[str] = Field(default=None, alias="CreateTimeBefore", description="筛选该时间点前创建的知识库。（RFC3339格式）")
+    max_results: Optional[int] = Field(default=None, alias="MaxResults", description="分页查询时设置的每页行数。 - 最小值：1。 - 最大值：100。 - 默认值：10。")
+    next_token: Optional[str] = Field(default=None, alias="NextToken", description="分页查询凭证，用于标记分页的位置，初次调用该接口时无需设置。下次查询时，取值为上一次API调用返回的`NextToken`参数值。")
+    update_time_after: Optional[str] = Field(default=None, alias="UpdateTimeAfter", description="筛选该时间点后更新的知识库。（RFC3339格式）")
+    update_time_before: Optional[str] = Field(default=None, alias="UpdateTimeBefore", description="筛选该时间点前更新的知识库。（RFC3339格式）")
+    filters: Optional[list[FiltersItem]] = Field(default=None, alias="Filters")
+    
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+# ListKnowledgeBases - Response
+class ListKnowledgeBasesResponse(BaseModel):
+    knowledge_bases: Optional[list[KnowledgeBasesForListKnowledgeBases]] = Field(default=None, alias="KnowledgeBases", description="符合要求的知识库信息。")
+    page_number: Optional[int] = Field(default=None, alias="PageNumber")
+    page_size: Optional[int] = Field(default=None, alias="PageSize")
+    total_count: Optional[int] = Field(default=None, alias="TotalCount", description="符合要求的知识库总个数。")
     
     model_config = {
         "populate_by_name": True,