agentengine-sdk-python 0.8.5__py3-none-any.whl → 0.8.6__py3-none-any.whl

ksadk/studio/run_submission_receipts.py

@@ -0,0 +1,68 @@
+"""Read original submission facts; never infer delivery from the latest turn."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from ksadk.studio.contracts import OperationKind, OperationStatus, RunStatus, RunSubmissionRef
+from ksadk.studio.errors import StudioError
+
+
+def submission_receipt(studio: Any, ref: RunSubmissionRef) -> dict[str, Any]:
+    studio._require_direct_session(ref.session_id)
+    identity = {
+        "SchemaVersion": 1,
+        "AgentId": ref.agent_id,
+        "SessionId": ref.session_id,
+        "InvocationId": ref.client_invocation_id,
+    }
+    records = studio.event_store.list_runs(session_id=ref.session_id, agent_id=ref.agent_id)
+    matched = [
+        record
+        for record in records
+        if record.submission_ref == ref or record.id == ref.client_invocation_id
+    ]
+    if not matched:
+        # Legacy completed submissions already have a durable correlation key.
+        # Looking it up is not a RunAgent retry and cannot start an operation.
+        operation = studio.operations.find_by_idempotency_key(
+            f"responses:{ref.client_invocation_id}"
+        )
+        if operation is not None and operation.kind == OperationKind.RUN:
+            matched = [record for record in records if record.id == operation.resource_id]
+    if len(matched) > 1:
+        raise StudioError(
+            "RUN_SUBMISSION_AMBIGUOUS", "原请求对应多个运行，不能确定投递结果", status_code=409
+        )
+    if matched:
+        record = matched[0]
+        studio._require_direct_run(record.id)
+        return {
+            **identity,
+            "RunId": record.id,
+            "Status": RunStatus(record.status).value.lower(),
+            "Source": "run",
+        }
+    # A queued request has no RunRecord yet. The original operation stores the
+    # correlation at submission, before starting or waiting for admission.
+    operations = [
+        operation
+        for operation in studio.operations.list(kind=OperationKind.RUN)
+        if operation.metadata.get("submission_ref") == ref.model_dump()
+    ]
+    if len(operations) > 1:
+        raise StudioError(
+            "RUN_SUBMISSION_AMBIGUOUS", "原请求对应多个操作，不能确定投递结果", status_code=409
+        )
+    if operations:
+        operation = operations[0]
+        state = OperationStatus(operation.status).value.lower()
+        if state in {"queued", "running", "failed", "cancelled", "interrupted"}:
+            return {
+                **identity,
+                "RunId": "",
+                "Status": state,
+                "Source": "operation",
+            }
+    # Missing evidence never proves non-delivery, success or permission to retry.
+    return {**identity, "RunId": "", "Status": "unknown", "Source": "unavailable"}

ksadk/studio/runtime_catalog.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from importlib.metadata import PackageNotFoundError
 from importlib.metadata import version as package_version
 
-from ksadk.harness.managed_runtime import ManagedHarnessRuntime
+from ksadk.harness.runtime_capabilities import managed_harness_native_capabilities
 from ksadk.managed_runtime import installed_runtime_version
 from ksadk.runtime import RuntimeExecutor
 
@@ -61,7 +61,7 @@ def _harness_capabilities() -> dict:
     # StudioPluginRuntime always supplies a Workspace state directory to the
     # shipped DSH Harness provider. The catalog describes that product path,
     # not a separately constructed in-memory SDK adapter.
-    native = ManagedHarnessRuntime(durable=True).native_capabilities()
+    native = managed_harness_native_capabilities(durable=True)
     checkpoint = native.get("checkpoint") or {}
     session_continuity = native.get("session_continuity") or {}
     return {
@@ -82,7 +82,12 @@ def _harness_capabilities() -> dict:
     }
 
 
-def inspect_runtime_catalog(executor: RuntimeExecutor) -> list[dict]:
+def inspect_runtime_catalog(
+    executor: RuntimeExecutor,
+    *,
+    native_harness_default: bool = False,
+    native_harness_durable: bool = False,
+) -> list[dict]:
     items: list[dict] = []
     for runtime_type in executor.registered_runtime_types():
         metadata = _RUNTIMES[runtime_type]
@@ -109,7 +114,21 @@ def inspect_runtime_catalog(executor: RuntimeExecutor) -> list[dict]:
             ),
         }
         if runtime_type == "harness":
-            entry["capabilities"] = _harness_capabilities()
+            if native_harness_default:
+                from ksadk.harness.native_runtime import NativeRuntime
+
+                declared = NativeRuntime(durable=native_harness_durable).native_capabilities()
+                entry["adapter"] = "NativeRuntimeAdapter"
+                entry["capabilities"] = {
+                    "session": True,
+                    "cancel": declared["cancel"]["supported"],
+                    "resume": "ledger_run_id" if native_harness_durable else "in_process",
+                    "checkpoint": "not_supported",
+                    "checkpointGranularity": None,
+                    "durableAcrossProcess": declared["session_continuity"]["durable"],
+                }
+            else:
+                entry["capabilities"] = _harness_capabilities()
         items.append(entry)
     return items
 

ksadk/studio/runtime_prewarm.py

@@ -0,0 +1,22 @@
+"""Prewarm immutable host wiring without claiming a future user turn."""
+
+import logging
+
+from ksadk.plugins.host import PluginHostError
+
+logger = logging.getLogger(__name__)
+
+
+async def prewarm_runtime_build(studio, build_id: str) -> None:
+    try:
+        spec = studio.resolve_run_spec(build_id)
+        runtime = studio.run_service.plugin_runtime
+        if runtime is None or spec.plugin_bundle_root is None:
+            return
+        # Session activation claims include selected model, reasoning and
+        # approval settings. Only the actual Run may pin those claims.
+        entry = await runtime._admission.resolve(lambda: runtime._host_for(spec.plugin_bundle_root))
+        if entry.agent_id != spec.agent_id:
+            raise PluginHostError("plugin_bundle_agent_mismatch", "prewarm Build Agent mismatch")
+    except Exception as error:
+        logger.info("runtime prewarm for Build %s skipped: %s", build_id, error)

ksadk/studio/scheduler_assistant.py

@@ -41,7 +41,10 @@ _SCHEDULE_TOPIC = re.compile(
     r"定时|周期|自动化|计划任务|每天|每日|每周|每星期|每月|工作日|"
     r"每(?:隔)?\s*[\d一二两三四五六七八九十百半几多少]+\s*(?:分|小时|天|日|周)|"
     r"(?:明天|后天|今晚|今天).{0,16}(?:点|时|[:：])|"
-    r"\b(?:schedul\w*|automation\w*|cron|every\s|daily|weekly|hourly|tomorrow)\b",
+    r"\b(?:schedul\w*|automation\w*|cron|daily|weekly|hourly|tomorrow)\b|"
+    r"\bevery\s+(?:\d+\s+)?(?:seconds?|minutes?|hours?|days?|weeks?|months?|"
+    r"weekdays?|mondays?|tuesdays?|wednesdays?|thursdays?|fridays?|"
+    r"saturdays?|sundays?)\b",
     re.IGNORECASE,
 )