agentkit-sdk-python 0.6.5__tar.gz → 0.7.0__tar.gz

{agentkit_sdk_python-0.6.5 → agentkit_sdk_python-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.6.5
+Version: 0.7.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.6.5 → agentkit_sdk_python-0.7.0}/agentkit/toolkit/cli/cli_add.py

@@ -26,6 +26,7 @@ auth block), serialized as a layered JSON document::
       "tools": ["web_search"],
       "skills": [],
       "system_prompt": "You are a helpful assistant.",
+      "description": "A helpful assistant.",
       "runtime": "adk",
       "knowledgebase": {"type": "viking", "project": "...", "region": "..."},
       "long_term_memory": {"type": ""},
@@ -42,7 +43,7 @@ import os
 import re
 from pathlib import Path
 from typing import Any, Optional
-from urllib.parse import parse_qs, urlparse
+from urllib.parse import parse_qs, urlparse, urlunparse
 
 import typer
 from rich.console import Console
@@ -74,6 +75,10 @@ _REGISTRY_QUERY_KEYS = {
 _REGISTRY_INT_KEYS = {"top_k"}
 _REGISTER_NETWORK_TYPES = {"public", "private"}
 _REGISTER_DEFAULT_VERSION = "2025-10-30"
+_DEFAULT_A2A_REGISTRY_URI = (
+    "agentkit://a2a-registry?"
+    "space_name=Default&region=cn-beijing&endpoint=https://open.volcengineapi.com/"
+)
 
 
 class _A2ARegisterError(Exception):
@@ -96,6 +101,17 @@ def _is_blank(value: object) -> bool:
     return value is None or value == "" or value == [] or value == {}
 
 
+def _derive_agent_name(harness_name: str) -> str:
+    """ADK agent name derived from a harness name.
+
+    Must mirror veadk's ``harness_app/utils.py::agent_name_from_harness``.
+    """
+    name = re.sub(r"[^0-9A-Za-z_]", "_", harness_name or "")
+    if not name or name[0].isdigit():
+        name = f"_{name}"
+    return f"{name}_" if name == "user" else name
+
+
 def _prune(data: dict) -> None:
     """Drop unset fields so the file holds only what is configured.
 
@@ -142,9 +158,16 @@ def _parse_registry_int(key: str, value: object) -> object:
         raise ValueError(f"Registry param `{key}` must be an integer, got {value!r}.") from exc
 
 
+def _expand_default_registry_uri(value: str) -> str:
+    raw = value.strip()
+    if raw.lower() == "default":
+        return _DEFAULT_A2A_REGISTRY_URI
+    return raw
+
+
 def _parse_registry_uri(value: str) -> dict:
     """Parse the supported AgentKit A2A registry URI into a spec section."""
-    raw = value.strip()
+    raw = _expand_default_registry_uri(value)
     if raw.lower() == "disabled":
         return {"type": ""}
 
@@ -157,7 +180,7 @@ def _parse_registry_uri(value: str) -> dict:
         raise ValueError(
             "Unsupported registry URI. Currently only "
             "`agentkit://a2a-registry?space_id=xxx&top_k=3` or "
-            "`disabled` is supported."
+            "`default` / `disabled` is supported."
         )
 
     query = {
@@ -211,8 +234,13 @@ def _apply_registry_config(
         section = {}
 
     if registry is not None:
-        section.update(_parse_registry_uri(registry))
+        parsed_registry = _parse_registry_uri(registry)
+        if parsed_registry.get("space_name") and "space_id" not in parsed_registry:
+            section.pop("space_id", None)
+        section.update(parsed_registry)
 
+    if registry_space_name is not None:
+        section.pop("space_id", None)
     _set_registry_value(section, "space_id", registry_space_id)
     _set_registry_value(section, "space_name", registry_space_name)
     _set_registry_value(section, "top_k", registry_top_k)
@@ -253,6 +281,16 @@ def _apply_registry_config(
         )
 
     data["registry"] = section
+    if section.get("type") == "agentkit_a2a":
+        resolved_endpoint, resolved_region = _resolve_agentkit_openapi_target(
+            endpoint=str(section["endpoint"]) if section.get("endpoint") else None,
+            region=str(section["region"]) if section.get("region") else None,
+        )
+        _enable_a2a_space_intent(
+            str(section["space_id"]),
+            endpoint=resolved_endpoint,
+            region=resolved_region,
+        )
 
 
 def _load_spec(path: Path) -> dict:
@@ -294,6 +332,30 @@ def _default_agentkit_endpoint(region: str) -> str:
     return f"https://agentkit.{region}.volcengineapi.com/"
 
 
+def _resolve_agentkit_openapi_target(
+    *,
+    endpoint: Optional[str],
+    region: Optional[str],
+) -> tuple[str, str]:
+    resolved_region = (
+        region
+        or os.getenv("AGENTKIT_REGION")
+        or os.getenv("VOLCENGINE_REGION")
+        or "cn-beijing"
+    )
+    resolved_endpoint = endpoint or _default_agentkit_endpoint(str(resolved_region))
+    parsed = urlparse(resolved_endpoint)
+    if (
+        parsed.scheme
+        and parsed.netloc
+        and (parsed.query or parsed.params or parsed.fragment)
+    ):
+        resolved_endpoint = urlunparse(
+            (parsed.scheme, parsed.netloc, parsed.path or "/", "", "", "")
+        )
+    return resolved_endpoint, str(resolved_region)
+
+
 def _request_id(response: dict[str, Any]) -> str | None:
     return (response.get("ResponseMetadata") or {}).get("RequestId")
 
@@ -374,6 +436,25 @@ def _agentkit_post(
     return data, duration_ms
 
 
+def _enable_a2a_space_intent(
+    space_id: str,
+    *,
+    endpoint: str,
+    region: str,
+) -> dict[str, Any]:
+    response, request_duration_ms = _agentkit_post(
+        endpoint=endpoint,
+        version=_REGISTER_DEFAULT_VERSION,
+        region=region,
+        action="UpdateA2aSpace",
+        body={"Id": space_id, "IntentEnabled": True},
+    )
+    return {
+        "request_id": _request_id(response),
+        "request_duration_ms": request_duration_ms,
+    }
+
+
 def _resolve_a2a_space_id_by_name(
     space_name: str,
     *,
@@ -587,6 +668,12 @@ def harness_command(
     system_prompt: Optional[str] = typer.Option(
         None, "--system-prompt", help="Agent system prompt / instruction."
     ),
+    description: Optional[str] = typer.Option(
+        None,
+        "--description",
+        "--desc",
+        help="Agent description (used at agent init, e.g. for A2A discovery).",
+    ),
     model_name: Optional[str] = typer.Option(
         None, "--model-name", help="Reasoning model name."
     ),
@@ -612,7 +699,10 @@ def harness_command(
     registry: Optional[str] = typer.Option(
         None,
         "--registry",
-        help='AgentKit A2A registry URI, e.g. "agentkit://a2a-registry?space_id=xxx&top_k=3".',
+        help=(
+            'AgentKit A2A registry URI, "default", or "disabled", e.g. '
+            '"agentkit://a2a-registry?space_id=xxx&top_k=3".'
+        ),
     ),
     registry_space_id: Optional[str] = typer.Option(
         None, "--registry-space-id", help="AgentKit A2A SpaceId."
@@ -790,6 +880,13 @@ def harness_command(
         )
         raise typer.Exit(1)
 
+    agent_name = _derive_agent_name(name)
+    if agent_name != name:
+        console.print(
+            f"[yellow]ℹ Once deployed, the agent will be named "
+            f"'{agent_name}' instead of '{name}'.[/yellow]"
+        )
+
     _validate_choice("--runtime", runtime, _RUNTIMES)
     _validate_choice("--knowledgebase-type", knowledgebase_type, _KNOWLEDGEBASE_TYPES)
     _validate_choice(
@@ -820,6 +917,8 @@ def harness_command(
         data["model"]["name"] = model_name
     if system_prompt is not None:
         data["system_prompt"] = system_prompt
+    if description is not None:
+        data["description"] = description
     if runtime is not None:
         data["runtime"] = runtime
     if structured_tool_calls is not None:
@@ -904,7 +1003,9 @@ def harness_command(
             registry_region,
         )
     except _A2ARegisterError as exc:
-        console.print(f"[red]Error: failed to resolve A2A space name: {exc.message}[/red]")
+        console.print(
+            f"[red]Error: failed to configure A2A registry: {exc.message}[/red]"
+        )
         if exc.diagnostics:
             console.print(json.dumps(exc.diagnostics, ensure_ascii=False, indent=2))
         raise typer.Exit(1) from exc

{agentkit_sdk_python-0.6.5 → agentkit_sdk_python-0.7.0}/agentkit/toolkit/cli/cli_invoke.py

@@ -662,6 +662,7 @@ def _parse_harness_registry_override(value: Optional[str]) -> dict[str, Any]:
     """Parse ``--registry`` into one-time harness registry overrides.
 
     Supported forms:
+    - default (Default space in cn-beijing via https://open.volcengineapi.com/)
     - agentkit://a2a-registry?space_id=xxx&top_k=3&region=cn-beijing
     - https://... (treated as registry_endpoint; recognized query params are
       also extracted when present)
@@ -673,6 +674,9 @@ def _parse_harness_registry_override(value: Optional[str]) -> dict[str, Any]:
     if not raw:
         return {}
 
+    from agentkit.toolkit.cli.cli_add import _expand_default_registry_uri
+
+    raw = _expand_default_registry_uri(raw)
     parsed = urlparse(raw)
     overrides: dict[str, Any] = {}
 
@@ -686,14 +690,16 @@ def _parse_harness_registry_override(value: Optional[str]) -> dict[str, Any]:
         if parsed.netloc != "a2a-registry" or parsed.path not in {"", "/"}:
             raise ValueError(
                 "Unsupported registry URI. Use "
-                '`agentkit://a2a-registry?space_id=xxx&top_k=3` or an http(s) URL.'
+                '`default`, `agentkit://a2a-registry?space_id=xxx&top_k=3`, '
+                "or an http(s) URL."
             )
     elif parsed.scheme in {"http", "https"}:
         overrides["registry_endpoint"] = raw
     else:
         raise ValueError(
             "Unsupported registry value. Use "
-            '`agentkit://a2a-registry?space_id=xxx&top_k=3` or an http(s) URL.'
+            '`default`, `agentkit://a2a-registry?space_id=xxx&top_k=3`, '
+            "or an http(s) URL."
         )
 
     unknown = sorted(set(query) - set(_INVOKE_REGISTRY_QUERY_ALIASES))
@@ -786,6 +792,23 @@ def _merge_harness_registry_overrides(
     return overrides
 
 
+def _enable_harness_registry_intent(overrides: dict[str, Any]) -> None:
+    space_id = overrides.get("registry_space_id")
+    if not space_id:
+        return
+
+    from agentkit.toolkit.cli.cli_add import (
+        _enable_a2a_space_intent,
+        _resolve_agentkit_openapi_target,
+    )
+
+    endpoint, region = _resolve_agentkit_openapi_target(
+        endpoint=overrides.get("registry_endpoint"),
+        region=overrides.get("registry_region"),
+    )
+    _enable_a2a_space_intent(str(space_id), endpoint=endpoint, region=region)
+
+
 # Fixed ADK app name for the run_sse path. The harness loader serves its single
 # agent under any app name, so a stable constant keeps the CLI decoupled from the
 # deployed HARNESS_NAME.
@@ -998,7 +1021,8 @@ def harness_command(
         "--registry",
         help=(
             "Override A2A registry for this invocation. Accepts "
-            "`agentkit://a2a-registry?space_id=xxx&top_k=3` or an http(s) URL."
+            "`default`, `agentkit://a2a-registry?space_id=xxx&top_k=3`, "
+            "or an http(s) URL."
         ),
     ),
     registry_top_k: int = typer.Option(
@@ -1066,8 +1090,9 @@ def harness_command(
             registry_endpoint=registry_endpoint,
             registry_region=registry_region,
         )
+        _enable_harness_registry_intent(registry_overrides)
     except _A2ARegisterError as e:
-        console.print(f"[red]Error: failed to resolve A2A space name: {e}[/red]")
+        console.print(f"[red]Error: failed to configure A2A registry: {e}[/red]")
         raise typer.Exit(1)
     except ValueError as e:
         console.print(f"[red]Error: {e}[/red]")

{agentkit_sdk_python-0.6.5 → agentkit_sdk_python-0.7.0}/agentkit/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.6.5"
+VERSION = "0.7.0"

{agentkit_sdk_python-0.6.5 → agentkit_sdk_python-0.7.0}/agentkit_sdk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.6.5
+Version: 0.7.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.6.5 → agentkit_sdk_python-0.7.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentkit-sdk-python"
-version = "0.6.5"
+version = "0.7.0"
 description = "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."
 readme = "README.md"
 requires-python = ">=3.10"