Skip to content

feat: update api version 1215 #26

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
yaozheng-fang merged 1 commit into from
Dec 16, 2025
Merged

feat: update api version 1215 #26

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
58 changes: 29 additions & 29 deletions agentkit/sdk/knowledge/client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,12 +39,12 @@ class AgentkitKnowledgeClient(BaseAgentkitClient):
"""AgentKit Knowledge Base Management Service"""

API_ACTIONS: Dict[str, str] = {
"UpdateKnowledgeBase": "UpdateKnowledgeBase",
"ListKnowledgeBases": "ListKnowledgeBases",
"AddKnowledgeBase": "AddKnowledgeBase",
"GetKnowledgeConnectionInfo": "GetKnowledgeConnectionInfo",
"GetKnowledgeBase": "GetKnowledgeBase",
"DeleteKnowledgeBase": "DeleteKnowledgeBase",
"GetKnowledgeBase": "GetKnowledgeBase",
"GetKnowledgeConnectionInfo": "GetKnowledgeConnectionInfo",
"ListKnowledgeBases": "ListKnowledgeBases",
"UpdateKnowledgeBase": "UpdateKnowledgeBase",
}

def __init__(
Expand All @@ -62,31 +62,31 @@ def __init__(
service_name="knowledge",
)

def update_knowledge_base(
self, request: UpdateKnowledgeBaseRequest
) -> UpdateKnowledgeBaseResponse:
def add_knowledge_base(
self, request: AddKnowledgeBaseRequest
) -> AddKnowledgeBaseResponse:
return self._invoke_api(
api_action="UpdateKnowledgeBase",
api_action="AddKnowledgeBase",
request=request,
response_type=UpdateKnowledgeBaseResponse,
response_type=AddKnowledgeBaseResponse,
)

def list_knowledge_bases(
self, request: ListKnowledgeBasesRequest
) -> ListKnowledgeBasesResponse:
def delete_knowledge_base(
self, request: DeleteKnowledgeBaseRequest
) -> DeleteKnowledgeBaseResponse:
return self._invoke_api(
api_action="ListKnowledgeBases",
api_action="DeleteKnowledgeBase",
request=request,
response_type=ListKnowledgeBasesResponse,
response_type=DeleteKnowledgeBaseResponse,
)

def add_knowledge_base(
self, request: AddKnowledgeBaseRequest
) -> AddKnowledgeBaseResponse:
def get_knowledge_base(
self, request: GetKnowledgeBaseRequest
) -> GetKnowledgeBaseResponse:
return self._invoke_api(
api_action="AddKnowledgeBase",
api_action="GetKnowledgeBase",
request=request,
response_type=AddKnowledgeBaseResponse,
response_type=GetKnowledgeBaseResponse,
)

def get_knowledge_connection_info(
Expand All @@ -98,20 +98,20 @@ def get_knowledge_connection_info(
response_type=GetKnowledgeConnectionInfoResponse,
)

def get_knowledge_base(
self, request: GetKnowledgeBaseRequest
) -> GetKnowledgeBaseResponse:
def list_knowledge_bases(
self, request: ListKnowledgeBasesRequest
) -> ListKnowledgeBasesResponse:
return self._invoke_api(
api_action="GetKnowledgeBase",
api_action="ListKnowledgeBases",
request=request,
response_type=GetKnowledgeBaseResponse,
response_type=ListKnowledgeBasesResponse,
)

def delete_knowledge_base(
self, request: DeleteKnowledgeBaseRequest
) -> DeleteKnowledgeBaseResponse:
def update_knowledge_base(
self, request: UpdateKnowledgeBaseRequest
) -> UpdateKnowledgeBaseResponse:
return self._invoke_api(
api_action="DeleteKnowledgeBase",
api_action="UpdateKnowledgeBase",
request=request,
response_type=DeleteKnowledgeBaseResponse,
response_type=UpdateKnowledgeBaseResponse,
)
213 changes: 105 additions & 108 deletions agentkit/sdk/knowledge/types.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,29 +28,37 @@ class KnowledgeBaseModel(BaseModel):


# Data Types
class AssociatedRuntimesForGetKnowledgeBase(KnowledgeBaseModel):
id: Optional[str] = Field(default=None, alias="Id")
name: Optional[str] = Field(default=None, alias="Name")


class AssociatedRuntimesForListKnowledgeBases(KnowledgeBaseModel):
id: Optional[str] = Field(default=None, alias="Id")
name: Optional[str] = Field(default=None, alias="Name")


class VpcConfigurationForListKnowledgeBases(KnowledgeBaseModel):
security_group_ids: Optional[list[str]] = Field(
default=None, alias="SecurityGroupIds"
)
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
class ConnectionInfosForGetKnowledgeConnectionInfo(KnowledgeBaseModel):
addr_type: Optional[str] = Field(default=None, alias="AddrType")
auth_key: Optional[str] = Field(default=None, alias="AuthKey")
auth_type: Optional[str] = Field(default=None, alias="AuthType")
base_url: Optional[str] = Field(default=None, alias="BaseUrl")
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")
status: Optional[str] = Field(default=None, alias="Status")
vpc_id: Optional[str] = Field(default=None, alias="VpcId")
vpc_name: Optional[str] = Field(default=None, alias="VpcName")


class NetworkConfigurationForListKnowledgeBases(KnowledgeBaseModel):
enable_private_network: Optional[bool] = Field(
default=None, alias="EnablePrivateNetwork"
)
enable_public_network: Optional[bool] = Field(
default=None, alias="EnablePublicNetwork"
)
vpc_configuration: Optional[VpcConfigurationForListKnowledgeBases] = Field(
default=None, alias="VpcConfiguration"
class KnowledgeBasesForAddKnowledgeBase(KnowledgeBaseModel):
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")


class KnowledgeBasesForListKnowledgeBases(KnowledgeBaseModel):
Expand All @@ -74,101 +82,37 @@ class KnowledgeBasesForListKnowledgeBases(KnowledgeBaseModel):
status: Optional[str] = Field(default=None, alias="Status")


class KnowledgeBasesForAddKnowledgeBase(KnowledgeBaseModel):
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"
class NetworkConfigurationForListKnowledgeBases(KnowledgeBaseModel):
enable_private_network: Optional[bool] = Field(
default=None, alias="EnablePrivateNetwork"
)
enable_public_network: Optional[bool] = Field(
default=None, alias="EnablePublicNetwork"
)
vpc_configuration: Optional[VpcConfigurationForListKnowledgeBases] = Field(
default=None, alias="VpcConfiguration"
)
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
status: Optional[str] = Field(default=None, alias="Status")


class ConnectionInfosForGetKnowledgeConnectionInfo(KnowledgeBaseModel):
addr_type: Optional[str] = Field(default=None, alias="AddrType")
auth_key: Optional[str] = Field(default=None, alias="AuthKey")
auth_type: Optional[str] = Field(default=None, alias="AuthType")
base_url: Optional[str] = Field(default=None, alias="BaseUrl")
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")
status: Optional[str] = Field(default=None, alias="Status")
vpc_id: Optional[str] = Field(default=None, alias="VpcId")
vpc_name: Optional[str] = Field(default=None, alias="VpcName")


class AssociatedRuntimesForGetKnowledgeBase(KnowledgeBaseModel):
id: Optional[str] = Field(default=None, alias="Id")
name: Optional[str] = Field(default=None, alias="Name")


class VpcConfigForGetKnowledgeBase(KnowledgeBaseModel):
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
vpc_id: Optional[str] = Field(default=None, alias="VpcId")


# UpdateKnowledgeBase - Request
class VpcForUpdateKnowledgeBase(KnowledgeBaseModel):
vpc_id: str = Field(..., alias="VpcId")
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")


class UpdateKnowledgeBaseRequest(KnowledgeBaseModel):
description: Optional[str] = Field(default=None, alias="Description")
knowledge_id: str = Field(..., alias="KnowledgeId")
vpc_config: Optional[VpcForUpdateKnowledgeBase] = Field(
default=None, alias="VpcConfig"
)


# UpdateKnowledgeBase - Response
class UpdateKnowledgeBaseResponse(KnowledgeBaseModel):
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")


# ListKnowledgeBases - Request
class FiltersItemForListKnowledgeBases(KnowledgeBaseModel):
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")


class ListKnowledgeBasesRequest(KnowledgeBaseModel):
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")
project_name: Optional[str] = Field(default=None, alias="ProjectName")
update_time_after: Optional[str] = Field(default=None, alias="UpdateTimeAfter")
update_time_before: Optional[str] = Field(default=None, alias="UpdateTimeBefore")
filters: Optional[list[FiltersItemForListKnowledgeBases]] = Field(
default=None, alias="Filters"
)


# ListKnowledgeBases - Response
class ListKnowledgeBasesResponse(KnowledgeBaseModel):
knowledge_bases: Optional[list[KnowledgeBasesForListKnowledgeBases]] = Field(
default=None, alias="KnowledgeBases"
class VpcConfigurationForListKnowledgeBases(KnowledgeBaseModel):
security_group_ids: Optional[list[str]] = Field(
default=None, alias="SecurityGroupIds"
)
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")
total_count: Optional[int] = Field(default=None, alias="TotalCount")
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
vpc_id: Optional[str] = Field(default=None, alias="VpcId")


# AddKnowledgeBase - Request
class KnowledgeBasesItemForAddKnowledgeBase(KnowledgeBaseModel):
description: Optional[str] = Field(default=None, alias="Description")
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")


class AddKnowledgeBaseRequest(KnowledgeBaseModel):
Expand All @@ -185,25 +129,18 @@ class AddKnowledgeBaseResponse(KnowledgeBaseModel):
)


# GetKnowledgeConnectionInfo - Request
class GetKnowledgeConnectionInfoRequest(KnowledgeBaseModel):
# DeleteKnowledgeBase - Request
class DeleteKnowledgeBaseRequest(KnowledgeBaseModel):
knowledge_id: str = Field(..., alias="KnowledgeId")


# GetKnowledgeConnectionInfo - Response
class GetKnowledgeConnectionInfoResponse(KnowledgeBaseModel):
connection_infos: Optional[list[ConnectionInfosForGetKnowledgeConnectionInfo]] = (
Field(default=None, alias="ConnectionInfos")
)
# DeleteKnowledgeBase - Response
class DeleteKnowledgeBaseResponse(KnowledgeBaseModel):
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
message: Optional[str] = Field(default=None, alias="Message")
name: Optional[str] = Field(default=None, alias="Name")
project_name: Optional[str] = Field(default=None, alias="ProjectName")
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")


# GetKnowledgeBase - Request
Expand Down Expand Up @@ -234,13 +171,73 @@ class GetKnowledgeBaseResponse(KnowledgeBaseModel):
)


# DeleteKnowledgeBase - Request
class DeleteKnowledgeBaseRequest(KnowledgeBaseModel):
# GetKnowledgeConnectionInfo - Request
class GetKnowledgeConnectionInfoRequest(KnowledgeBaseModel):
knowledge_id: str = Field(..., alias="KnowledgeId")


# DeleteKnowledgeBase - Response
class DeleteKnowledgeBaseResponse(KnowledgeBaseModel):
# GetKnowledgeConnectionInfo - Response
class GetKnowledgeConnectionInfoResponse(KnowledgeBaseModel):
connection_infos: Optional[list[ConnectionInfosForGetKnowledgeConnectionInfo]] = (
Field(default=None, alias="ConnectionInfos")
)
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
message: Optional[str] = Field(default=None, alias="Message")
name: Optional[str] = Field(default=None, alias="Name")
project_name: Optional[str] = Field(default=None, alias="ProjectName")
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")


# ListKnowledgeBases - Request
class FiltersItemForListKnowledgeBases(KnowledgeBaseModel):
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")


class ListKnowledgeBasesRequest(KnowledgeBaseModel):
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")
project_name: Optional[str] = Field(default=None, alias="ProjectName")
update_time_after: Optional[str] = Field(default=None, alias="UpdateTimeAfter")
update_time_before: Optional[str] = Field(default=None, alias="UpdateTimeBefore")
filters: Optional[list[FiltersItemForListKnowledgeBases]] = Field(
default=None, alias="Filters"
)


# ListKnowledgeBases - Response
class ListKnowledgeBasesResponse(KnowledgeBaseModel):
knowledge_bases: Optional[list[KnowledgeBasesForListKnowledgeBases]] = Field(
default=None, alias="KnowledgeBases"
)
next_token: Optional[str] = Field(default=None, alias="NextToken")


# UpdateKnowledgeBase - Request
class VpcForUpdateKnowledgeBase(KnowledgeBaseModel):
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
vpc_id: str = Field(..., alias="VpcId")


class UpdateKnowledgeBaseRequest(KnowledgeBaseModel):
description: Optional[str] = Field(default=None, alias="Description")
knowledge_id: str = Field(..., alias="KnowledgeId")
vpc_config: Optional[VpcForUpdateKnowledgeBase] = Field(
default=None, alias="VpcConfig"
)


# UpdateKnowledgeBase - Response
class UpdateKnowledgeBaseResponse(KnowledgeBaseModel):
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
provider_knowledge_id: Optional[str] = Field(
default=None, alias="ProviderKnowledgeId"
Expand Down
Loading