Skip to content

Commit 5f2acc8

Browse files
yaozheng-fangyaozheng-fang
authored
feat: update api version 1215 (#26)
2 parents 26b763c + 34325d2 commit 5f2acc8

File tree

10 files changed

+1375
-1203
lines changed

10 files changed

+1375
-1203
lines changed

‎agentkit/sdk/knowledge/client.py‎

Lines changed: 29 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,12 @@ class AgentkitKnowledgeClient(BaseAgentkitClient):
3939
"""AgentKit Knowledge Base Management Service"""
4040

4141
API_ACTIONS: Dict[str, str] = {
42-
"UpdateKnowledgeBase": "UpdateKnowledgeBase",
43-
"ListKnowledgeBases": "ListKnowledgeBases",
4442
"AddKnowledgeBase": "AddKnowledgeBase",
45-
"GetKnowledgeConnectionInfo": "GetKnowledgeConnectionInfo",
46-
"GetKnowledgeBase": "GetKnowledgeBase",
4743
"DeleteKnowledgeBase": "DeleteKnowledgeBase",
44+
"GetKnowledgeBase": "GetKnowledgeBase",
45+
"GetKnowledgeConnectionInfo": "GetKnowledgeConnectionInfo",
46+
"ListKnowledgeBases": "ListKnowledgeBases",
47+
"UpdateKnowledgeBase": "UpdateKnowledgeBase",
4848
}
4949

5050
def __init__(
@@ -62,31 +62,31 @@ def __init__(
6262
service_name="knowledge",
6363
)
6464

65-
def update_knowledge_base(
66-
self, request: UpdateKnowledgeBaseRequest
67-
) -> UpdateKnowledgeBaseResponse:
65+
def add_knowledge_base(
66+
self, request: AddKnowledgeBaseRequest
67+
) -> AddKnowledgeBaseResponse:
6868
return self._invoke_api(
69-
api_action="UpdateKnowledgeBase",
69+
api_action="AddKnowledgeBase",
7070
request=request,
71-
response_type=UpdateKnowledgeBaseResponse,
71+
response_type=AddKnowledgeBaseResponse,
7272
)
7373

74-
def list_knowledge_bases(
75-
self, request: ListKnowledgeBasesRequest
76-
) -> ListKnowledgeBasesResponse:
74+
def delete_knowledge_base(
75+
self, request: DeleteKnowledgeBaseRequest
76+
) -> DeleteKnowledgeBaseResponse:
7777
return self._invoke_api(
78-
api_action="ListKnowledgeBases",
78+
api_action="DeleteKnowledgeBase",
7979
request=request,
80-
response_type=ListKnowledgeBasesResponse,
80+
response_type=DeleteKnowledgeBaseResponse,
8181
)
8282

83-
def add_knowledge_base(
84-
self, request: AddKnowledgeBaseRequest
85-
) -> AddKnowledgeBaseResponse:
83+
def get_knowledge_base(
84+
self, request: GetKnowledgeBaseRequest
85+
) -> GetKnowledgeBaseResponse:
8686
return self._invoke_api(
87-
api_action="AddKnowledgeBase",
87+
api_action="GetKnowledgeBase",
8888
request=request,
89-
response_type=AddKnowledgeBaseResponse,
89+
response_type=GetKnowledgeBaseResponse,
9090
)
9191

9292
def get_knowledge_connection_info(
@@ -98,20 +98,20 @@ def get_knowledge_connection_info(
9898
response_type=GetKnowledgeConnectionInfoResponse,
9999
)
100100

101-
def get_knowledge_base(
102-
self, request: GetKnowledgeBaseRequest
103-
) -> GetKnowledgeBaseResponse:
101+
def list_knowledge_bases(
102+
self, request: ListKnowledgeBasesRequest
103+
) -> ListKnowledgeBasesResponse:
104104
return self._invoke_api(
105-
api_action="GetKnowledgeBase",
105+
api_action="ListKnowledgeBases",
106106
request=request,
107-
response_type=GetKnowledgeBaseResponse,
107+
response_type=ListKnowledgeBasesResponse,
108108
)
109109

110-
def delete_knowledge_base(
111-
self, request: DeleteKnowledgeBaseRequest
112-
) -> DeleteKnowledgeBaseResponse:
110+
def update_knowledge_base(
111+
self, request: UpdateKnowledgeBaseRequest
112+
) -> UpdateKnowledgeBaseResponse:
113113
return self._invoke_api(
114-
api_action="DeleteKnowledgeBase",
114+
api_action="UpdateKnowledgeBase",
115115
request=request,
116-
response_type=DeleteKnowledgeBaseResponse,
116+
response_type=UpdateKnowledgeBaseResponse,
117117
)

‎agentkit/sdk/knowledge/types.py‎

Lines changed: 105 additions & 108 deletions
Original file line numberDiff line numberDiff line change
@@ -28,29 +28,37 @@ class KnowledgeBaseModel(BaseModel):
2828

2929

3030
# Data Types
31+
class AssociatedRuntimesForGetKnowledgeBase(KnowledgeBaseModel):
32+
id: Optional[str] = Field(default=None, alias="Id")
33+
name: Optional[str] = Field(default=None, alias="Name")
34+
35+
3136
class AssociatedRuntimesForListKnowledgeBases(KnowledgeBaseModel):
3237
id: Optional[str] = Field(default=None, alias="Id")
3338
name: Optional[str] = Field(default=None, alias="Name")
3439

3540

36-
class VpcConfigurationForListKnowledgeBases(KnowledgeBaseModel):
37-
security_group_ids: Optional[list[str]] = Field(
38-
default=None, alias="SecurityGroupIds"
39-
)
40-
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
41+
class ConnectionInfosForGetKnowledgeConnectionInfo(KnowledgeBaseModel):
42+
addr_type: Optional[str] = Field(default=None, alias="AddrType")
43+
auth_key: Optional[str] = Field(default=None, alias="AuthKey")
44+
auth_type: Optional[str] = Field(default=None, alias="AuthType")
45+
base_url: Optional[str] = Field(default=None, alias="BaseUrl")
46+
expire_at: Optional[str] = Field(default=None, alias="ExpireAt")
47+
extra_config: Optional[str] = Field(default=None, alias="ExtraConfig")
48+
region: Optional[str] = Field(default=None, alias="Region")
49+
status: Optional[str] = Field(default=None, alias="Status")
4150
vpc_id: Optional[str] = Field(default=None, alias="VpcId")
51+
vpc_name: Optional[str] = Field(default=None, alias="VpcName")
4252

4353

44-
class NetworkConfigurationForListKnowledgeBases(KnowledgeBaseModel):
45-
enable_private_network: Optional[bool] = Field(
46-
default=None, alias="EnablePrivateNetwork"
47-
)
48-
enable_public_network: Optional[bool] = Field(
49-
default=None, alias="EnablePublicNetwork"
50-
)
51-
vpc_configuration: Optional[VpcConfigurationForListKnowledgeBases] = Field(
52-
default=None, alias="VpcConfiguration"
54+
class KnowledgeBasesForAddKnowledgeBase(KnowledgeBaseModel):
55+
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
56+
message: Optional[str] = Field(default=None, alias="Message")
57+
provider_knowledge_id: Optional[str] = Field(
58+
default=None, alias="ProviderKnowledgeId"
5359
)
60+
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
61+
status: Optional[str] = Field(default=None, alias="Status")
5462

5563

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

7684

77-
class KnowledgeBasesForAddKnowledgeBase(KnowledgeBaseModel):
78-
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
79-
message: Optional[str] = Field(default=None, alias="Message")
80-
provider_knowledge_id: Optional[str] = Field(
81-
default=None, alias="ProviderKnowledgeId"
85+
class NetworkConfigurationForListKnowledgeBases(KnowledgeBaseModel):
86+
enable_private_network: Optional[bool] = Field(
87+
default=None, alias="EnablePrivateNetwork"
88+
)
89+
enable_public_network: Optional[bool] = Field(
90+
default=None, alias="EnablePublicNetwork"
91+
)
92+
vpc_configuration: Optional[VpcConfigurationForListKnowledgeBases] = Field(
93+
default=None, alias="VpcConfiguration"
8294
)
83-
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
84-
status: Optional[str] = Field(default=None, alias="Status")
85-
86-
87-
class ConnectionInfosForGetKnowledgeConnectionInfo(KnowledgeBaseModel):
88-
addr_type: Optional[str] = Field(default=None, alias="AddrType")
89-
auth_key: Optional[str] = Field(default=None, alias="AuthKey")
90-
auth_type: Optional[str] = Field(default=None, alias="AuthType")
91-
base_url: Optional[str] = Field(default=None, alias="BaseUrl")
92-
expire_at: Optional[str] = Field(default=None, alias="ExpireAt")
93-
extra_config: Optional[str] = Field(default=None, alias="ExtraConfig")
94-
region: Optional[str] = Field(default=None, alias="Region")
95-
status: Optional[str] = Field(default=None, alias="Status")
96-
vpc_id: Optional[str] = Field(default=None, alias="VpcId")
97-
vpc_name: Optional[str] = Field(default=None, alias="VpcName")
98-
99-
100-
class AssociatedRuntimesForGetKnowledgeBase(KnowledgeBaseModel):
101-
id: Optional[str] = Field(default=None, alias="Id")
102-
name: Optional[str] = Field(default=None, alias="Name")
10395

10496

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

109101

110-
# UpdateKnowledgeBase - Request
111-
class VpcForUpdateKnowledgeBase(KnowledgeBaseModel):
112-
vpc_id: str = Field(..., alias="VpcId")
113-
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
114-
115-
116-
class UpdateKnowledgeBaseRequest(KnowledgeBaseModel):
117-
description: Optional[str] = Field(default=None, alias="Description")
118-
knowledge_id: str = Field(..., alias="KnowledgeId")
119-
vpc_config: Optional[VpcForUpdateKnowledgeBase] = Field(
120-
default=None, alias="VpcConfig"
121-
)
122-
123-
124-
# UpdateKnowledgeBase - Response
125-
class UpdateKnowledgeBaseResponse(KnowledgeBaseModel):
126-
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
127-
provider_knowledge_id: Optional[str] = Field(
128-
default=None, alias="ProviderKnowledgeId"
129-
)
130-
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
131-
132-
133-
# ListKnowledgeBases - Request
134-
class FiltersItemForListKnowledgeBases(KnowledgeBaseModel):
135-
name: Optional[str] = Field(default=None, alias="Name")
136-
name_contains: Optional[str] = Field(default=None, alias="NameContains")
137-
values: Optional[list[str]] = Field(default=None, alias="Values")
138-
139-
140-
class ListKnowledgeBasesRequest(KnowledgeBaseModel):
141-
create_time_after: Optional[str] = Field(default=None, alias="CreateTimeAfter")
142-
create_time_before: Optional[str] = Field(default=None, alias="CreateTimeBefore")
143-
max_results: Optional[int] = Field(default=None, alias="MaxResults")
144-
next_token: Optional[str] = Field(default=None, alias="NextToken")
145-
page_number: Optional[int] = Field(default=None, alias="PageNumber")
146-
page_size: Optional[int] = Field(default=None, alias="PageSize")
147-
project_name: Optional[str] = Field(default=None, alias="ProjectName")
148-
update_time_after: Optional[str] = Field(default=None, alias="UpdateTimeAfter")
149-
update_time_before: Optional[str] = Field(default=None, alias="UpdateTimeBefore")
150-
filters: Optional[list[FiltersItemForListKnowledgeBases]] = Field(
151-
default=None, alias="Filters"
152-
)
153-
154-
155-
# ListKnowledgeBases - Response
156-
class ListKnowledgeBasesResponse(KnowledgeBaseModel):
157-
knowledge_bases: Optional[list[KnowledgeBasesForListKnowledgeBases]] = Field(
158-
default=None, alias="KnowledgeBases"
102+
class VpcConfigurationForListKnowledgeBases(KnowledgeBaseModel):
103+
security_group_ids: Optional[list[str]] = Field(
104+
default=None, alias="SecurityGroupIds"
159105
)
160-
next_token: Optional[str] = Field(default=None, alias="NextToken")
161-
page_number: Optional[int] = Field(default=None, alias="PageNumber")
162-
page_size: Optional[int] = Field(default=None, alias="PageSize")
163-
total_count: Optional[int] = Field(default=None, alias="TotalCount")
106+
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
107+
vpc_id: Optional[str] = Field(default=None, alias="VpcId")
164108

165109

166110
# AddKnowledgeBase - Request
167111
class KnowledgeBasesItemForAddKnowledgeBase(KnowledgeBaseModel):
112+
description: Optional[str] = Field(default=None, alias="Description")
168113
name: str = Field(..., alias="Name")
169114
provider_knowledge_id: str = Field(..., alias="ProviderKnowledgeId")
170115
provider_type: str = Field(..., alias="ProviderType")
171-
description: Optional[str] = Field(default=None, alias="Description")
172116

173117

174118
class AddKnowledgeBaseRequest(KnowledgeBaseModel):
@@ -185,25 +129,18 @@ class AddKnowledgeBaseResponse(KnowledgeBaseModel):
185129
)
186130

187131

188-
# GetKnowledgeConnectionInfo - Request
189-
class GetKnowledgeConnectionInfoRequest(KnowledgeBaseModel):
132+
# DeleteKnowledgeBase - Request
133+
class DeleteKnowledgeBaseRequest(KnowledgeBaseModel):
190134
knowledge_id: str = Field(..., alias="KnowledgeId")
191135

192136

193-
# GetKnowledgeConnectionInfo - Response
194-
class GetKnowledgeConnectionInfoResponse(KnowledgeBaseModel):
195-
connection_infos: Optional[list[ConnectionInfosForGetKnowledgeConnectionInfo]] = (
196-
Field(default=None, alias="ConnectionInfos")
197-
)
137+
# DeleteKnowledgeBase - Response
138+
class DeleteKnowledgeBaseResponse(KnowledgeBaseModel):
198139
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
199-
message: Optional[str] = Field(default=None, alias="Message")
200-
name: Optional[str] = Field(default=None, alias="Name")
201-
project_name: Optional[str] = Field(default=None, alias="ProjectName")
202140
provider_knowledge_id: Optional[str] = Field(
203141
default=None, alias="ProviderKnowledgeId"
204142
)
205143
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
206-
status: Optional[str] = Field(default=None, alias="Status")
207144

208145

209146
# GetKnowledgeBase - Request
@@ -234,13 +171,73 @@ class GetKnowledgeBaseResponse(KnowledgeBaseModel):
234171
)
235172

236173

237-
# DeleteKnowledgeBase - Request
238-
class DeleteKnowledgeBaseRequest(KnowledgeBaseModel):
174+
# GetKnowledgeConnectionInfo - Request
175+
class GetKnowledgeConnectionInfoRequest(KnowledgeBaseModel):
239176
knowledge_id: str = Field(..., alias="KnowledgeId")
240177

241178

242-
# DeleteKnowledgeBase - Response
243-
class DeleteKnowledgeBaseResponse(KnowledgeBaseModel):
179+
# GetKnowledgeConnectionInfo - Response
180+
class GetKnowledgeConnectionInfoResponse(KnowledgeBaseModel):
181+
connection_infos: Optional[list[ConnectionInfosForGetKnowledgeConnectionInfo]] = (
182+
Field(default=None, alias="ConnectionInfos")
183+
)
184+
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
185+
message: Optional[str] = Field(default=None, alias="Message")
186+
name: Optional[str] = Field(default=None, alias="Name")
187+
project_name: Optional[str] = Field(default=None, alias="ProjectName")
188+
provider_knowledge_id: Optional[str] = Field(
189+
default=None, alias="ProviderKnowledgeId"
190+
)
191+
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
192+
status: Optional[str] = Field(default=None, alias="Status")
193+
194+
195+
# ListKnowledgeBases - Request
196+
class FiltersItemForListKnowledgeBases(KnowledgeBaseModel):
197+
name: Optional[str] = Field(default=None, alias="Name")
198+
name_contains: Optional[str] = Field(default=None, alias="NameContains")
199+
values: Optional[list[str]] = Field(default=None, alias="Values")
200+
201+
202+
class ListKnowledgeBasesRequest(KnowledgeBaseModel):
203+
create_time_after: Optional[str] = Field(default=None, alias="CreateTimeAfter")
204+
create_time_before: Optional[str] = Field(default=None, alias="CreateTimeBefore")
205+
max_results: Optional[int] = Field(default=None, alias="MaxResults")
206+
next_token: Optional[str] = Field(default=None, alias="NextToken")
207+
page_number: Optional[int] = Field(default=None, alias="PageNumber")
208+
page_size: Optional[int] = Field(default=None, alias="PageSize")
209+
project_name: Optional[str] = Field(default=None, alias="ProjectName")
210+
update_time_after: Optional[str] = Field(default=None, alias="UpdateTimeAfter")
211+
update_time_before: Optional[str] = Field(default=None, alias="UpdateTimeBefore")
212+
filters: Optional[list[FiltersItemForListKnowledgeBases]] = Field(
213+
default=None, alias="Filters"
214+
)
215+
216+
217+
# ListKnowledgeBases - Response
218+
class ListKnowledgeBasesResponse(KnowledgeBaseModel):
219+
knowledge_bases: Optional[list[KnowledgeBasesForListKnowledgeBases]] = Field(
220+
default=None, alias="KnowledgeBases"
221+
)
222+
next_token: Optional[str] = Field(default=None, alias="NextToken")
223+
224+
225+
# UpdateKnowledgeBase - Request
226+
class VpcForUpdateKnowledgeBase(KnowledgeBaseModel):
227+
subnet_ids: Optional[list[str]] = Field(default=None, alias="SubnetIds")
228+
vpc_id: str = Field(..., alias="VpcId")
229+
230+
231+
class UpdateKnowledgeBaseRequest(KnowledgeBaseModel):
232+
description: Optional[str] = Field(default=None, alias="Description")
233+
knowledge_id: str = Field(..., alias="KnowledgeId")
234+
vpc_config: Optional[VpcForUpdateKnowledgeBase] = Field(
235+
default=None, alias="VpcConfig"
236+
)
237+
238+
239+
# UpdateKnowledgeBase - Response
240+
class UpdateKnowledgeBaseResponse(KnowledgeBaseModel):
244241
knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
245242
provider_knowledge_id: Optional[str] = Field(
246243
default=None, alias="ProviderKnowledgeId"

0 commit comments

Comments
 (0)