Skip to content

Commit eb061a7

Browse files
authored
Merge pull request #16 from volcengine/fix/toolkit-sdk-bugfix
fix: bugfix for toolkit sdk
2 parents 7e0241b + d5544d3 commit eb061a7

File tree

25 files changed

+472
-658
lines changed

25 files changed

+472
-658
lines changed

agentkit/sdk/knowledge/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from typing import Dict
2121
from agentkit.client import BaseAgentkitClient
22-
from agentkit.sdk.knowledge.types import (
22+
from .types import (
2323
AddKnowledgeBaseRequest,
2424
AddKnowledgeBaseResponse,
2525
DeleteKnowledgeBaseRequest,

agentkit/sdk/memory/__init__.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,6 @@
3535
StrategiesForGetMemoryCollection,
3636
LongTermConfigurationForUpdateMemoryCollection,
3737
LongTermConfigurationForGetMemoryCollection,
38-
CollectionsItem,
39-
FiltersItem,
40-
StrategiesItem,
4138
)
4239

4340
__all__ = [
@@ -63,7 +60,4 @@
6360
"StrategiesForGetMemoryCollection",
6461
"LongTermConfigurationForUpdateMemoryCollection",
6562
"LongTermConfigurationForGetMemoryCollection",
66-
"CollectionsItem",
67-
"FiltersItem",
68-
"StrategiesItem",
6963
]

agentkit/sdk/memory/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from typing import Dict
2121
from agentkit.client import BaseAgentkitClient
22-
from agentkit.sdk.memory.types import (
22+
from .types import (
2323
AddMemoryCollectionRequest,
2424
AddMemoryCollectionResponse,
2525
CreateMemoryCollectionRequest,

agentkit/sdk/memory/types.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,9 @@ class LongTermStrategiesItemForUpdateMemoryCollection(MemoryBaseModel):
170170
class UpdateMemoryCollectionRequest(MemoryBaseModel):
171171
description: Optional[str] = Field(default=None, alias="Description")
172172
memory_id: str = Field(..., alias="MemoryId")
173+
long_term_configuration: Optional[LongTermForUpdateMemoryCollection] = Field(
174+
default=None, alias="LongTermConfiguration"
175+
)
173176
vpc_config: Optional[VpcForUpdateMemoryCollection] = Field(
174177
default=None, alias="VpcConfig"
175178
)
@@ -348,12 +351,15 @@ class CreateMemoryCollectionRequest(MemoryBaseModel):
348351
name: str = Field(..., alias="Name")
349352
project_name: Optional[str] = Field(default=None, alias="ProjectName")
350353
provider_type: Optional[str] = Field(default=None, alias="ProviderType")
351-
tags: Optional[list[TagsItemForCreateMemoryCollection]] = Field(
352-
default=None, alias="Tags"
354+
long_term_configuration: Optional[LongTermForCreateMemoryCollection] = Field(
355+
default=None, alias="LongTermConfiguration"
353356
)
354357
vpc_config: Optional[VpcForCreateMemoryCollection] = Field(
355358
default=None, alias="VpcConfig"
356359
)
360+
tags: Optional[list[TagsItemForCreateMemoryCollection]] = Field(
361+
default=None, alias="Tags"
362+
)
357363

358364

359365
# CreateMemoryCollection - Response

agentkit/sdk/runtime/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from typing import Dict
2121
from agentkit.client import BaseAgentkitClient
22-
from agentkit.sdk.runtime.types import (
22+
from .types import (
2323
CreateRuntimeRequest,
2424
CreateRuntimeResponse,
2525
DeleteRuntimeRequest,

agentkit/sdk/runtime/types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -445,10 +445,10 @@ class CreateRuntimeRequest(RuntimeTypeBaseModel):
445445
authorizer_configuration: Optional[AuthorizerForCreateRuntime] = Field(
446446
default=None, alias="AuthorizerConfiguration"
447447
)
448-
envs: Optional[list[EnvsItemForCreateRuntime]] = Field(default=None, alias="Envs")
449448
network_configuration: Optional[NetworkForCreateRuntime] = Field(
450449
default=None, alias="NetworkConfiguration"
451450
)
451+
envs: Optional[list[EnvsItemForCreateRuntime]] = Field(default=None, alias="Envs")
452452
tags: Optional[list[TagsItemForCreateRuntime]] = Field(default=None, alias="Tags")
453453

454454

agentkit/sdk/tools/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
from typing import Dict
2121
from agentkit.client import BaseAgentkitClient
22-
from agentkit.sdk.tools.types import (
22+
from .types import (
2323
CreateSessionRequest,
2424
CreateSessionResponse,
2525
CreateToolRequest,

agentkit/toolkit/cli/cli_destroy.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ def destroy_command(
3535
typer.confirm("Are you sure you want to destroy?", abort=True)
3636

3737
try:
38-
# Call SDK
39-
result = sdk.destroy(config_file=str(config_file), force=force)
38+
# Call SDK (force only controls CLI confirmation, SDK always performs the same destroy)
39+
result = sdk.destroy(config_file=str(config_file))
4040

4141
if result.success:
4242
console.print("[green]✅ Destruction completed successfully![/green]")

agentkit/toolkit/executors/lifecycle_executor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,10 @@ def launch(
126126
Args:
127127
config_dict: Configuration dictionary (optional)
128128
config_file: Path to configuration file (optional)
129-
platform: Build platform: "auto", "local", or "cloud"
129+
platform: Docker build platform/architecture string
130+
(e.g., "linux/amd64", "linux/arm64", or "auto"). This controls
131+
the Docker build target platform and is independent from the
132+
launch_type (local/cloud/hybrid) configured in agentkit.yaml.
130133
preflight_mode: How to handle missing cloud services (default: PROMPT)
131134
132135
Returns:

agentkit/toolkit/runners/base.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,6 @@ def _http_post_invoke(
140140

141141
# Use longer timeout for streaming calls
142142
actual_timeout = timeout if not stream else max(timeout, 300)
143-
144143
response = requests.post(
145144
url=endpoint,
146145
json=payload,

0 commit comments

Comments
 (0)