agentengine-sdk-python 0.8.0__py3-none-any.whl → 0.8.2__py3-none-any.whl

ksadk/runners/utils/__init__.py

@@ -2,18 +2,8 @@
 Runner 公共工具模块
 """
 
-from ksadk.runners.utils.langfuse import (
-    get_langfuse_callback,
-    get_langfuse_callbacks,
-    get_langfuse_metadata,
-    prepare_trace_metadata,
-)
 from ksadk.runners.utils.loader import load_agent_module
 
 __all__ = [
-    "get_langfuse_callback",
-    "get_langfuse_callbacks",
-    "get_langfuse_metadata",
-    "prepare_trace_metadata",
     "load_agent_module",
 ]

ksadk/runners/utils/loader.py

@@ -43,13 +43,29 @@ def load_agent_module(
     # 确定模块名
     if entry_point.endswith(".py"):
         module_name = entry_point[:-3]
+        entry_file = (project_path / entry_point).resolve()
     else:
         module_name = entry_point
+        entry_file = None
 
     # 路径转换为模块路径
     module_name = module_name.replace("/", ".").replace("\\", ".")
 
     try:
+        # Studio 会在同一 Python 进程内运行多个不可变 Bundle。它们通常都以
+        # ``agent.py`` 为入口；仅靠 ``import_module("agent")`` 会复用上一个
+        # Bundle 的 sys.modules 缓存，进而加载错误的 agentVariable。只复用与
+        # 当前 entry 文件一致的缓存，其余情况让 import 从当前 project_path 重载。
+        cached = sys.modules.get(module_name)
+        cached_file = getattr(cached, "__file__", None)
+        cached_matches_entry = bool(
+            cached_file is not None
+            and entry_file is not None
+            and Path(str(cached_file)).resolve() == entry_file
+        )
+        if cached is not None and not cached_matches_entry:
+            sys.modules.pop(module_name, None)
+            importlib.invalidate_caches()
         if force_reload and module_name in sys.modules:
             module = importlib.reload(sys.modules[module_name])
         else:

ksadk/runtime/__init__.py

@@ -1,41 +1,67 @@
-"""RuntimeAdapter 平台接口包 (goal-03 冻结签名 + goal-07 框架实现)。"""
-
-from ksadk.runtime.adapter import (
-    CONVERSATION_PREPROCESSING_METADATA_KEY,
-    BaseRuntime,
-    CancelResult,
-    CheckpointCapability,
-    CheckpointDescriptor,
-    ConversationPreprocessingRequest,
-    ResumePayload,
-    ResumeTarget,
-    RunHandle,
-    RuntimeAdapter,
-    RuntimeRegistry,
-    StartRequest,
-)
-from ksadk.runtime.framework_adapters import (
-    ADKRuntimeAdapter,
-    LangGraphRuntimeAdapter,
-    build_default_registry,
-)
-from ksadk.runtime.runner_adapter import RunnerRuntimeAdapter
-
-__all__ = [
-    "ADKRuntimeAdapter",
-    "BaseRuntime",
-    "CancelResult",
-    "CheckpointCapability",
-    "CheckpointDescriptor",
-    "ConversationPreprocessingRequest",
-    "CONVERSATION_PREPROCESSING_METADATA_KEY",
-    "LangGraphRuntimeAdapter",
-    "ResumePayload",
-    "ResumeTarget",
-    "RunHandle",
-    "RunnerRuntimeAdapter",
-    "RuntimeAdapter",
-    "RuntimeRegistry",
-    "StartRequest",
-    "build_default_registry",
-]
+"""Side-effect-free public exports for the canonical RuntimeAdapter stack."""
+
+from __future__ import annotations
+
+from importlib import import_module
+from typing import Any
+
+_EXPORTS = {
+    "ADKRuntimeAdapter": ("ksadk.runtime.framework_adapters", "ADKRuntimeAdapter"),
+    "BaseRuntime": ("ksadk.runtime.adapter", "BaseRuntime"),
+    "CancelResult": ("ksadk.runtime.adapter", "CancelResult"),
+    "PauseResult": ("ksadk.runtime.adapter", "PauseResult"),
+    "CheckpointCapability": ("ksadk.runtime.adapter", "CheckpointCapability"),
+    "CheckpointDescriptor": ("ksadk.runtime.adapter", "CheckpointDescriptor"),
+    "ConversationPreprocessingRequest": (
+        "ksadk.runtime.adapter",
+        "ConversationPreprocessingRequest",
+    ),
+    "CONVERSATION_PREPROCESSING_METADATA_KEY": (
+        "ksadk.runtime.adapter",
+        "CONVERSATION_PREPROCESSING_METADATA_KEY",
+    ),
+    "LangGraphRuntimeAdapter": (
+        "ksadk.runtime.framework_adapters",
+        "LangGraphRuntimeAdapter",
+    ),
+    "RESUME_START_REQUEST_NATIVE_KEY": (
+        "ksadk.runtime.adapter",
+        "RESUME_START_REQUEST_NATIVE_KEY",
+    ),
+    "ResumePayload": ("ksadk.runtime.adapter", "ResumePayload"),
+    "ResumeTarget": ("ksadk.runtime.adapter", "ResumeTarget"),
+    "RunHandle": ("ksadk.runtime.adapter", "RunHandle"),
+    "RunnerRuntimeAdapter": ("ksadk.runtime.runner_adapter", "RunnerRuntimeAdapter"),
+    "RuntimeAdapter": ("ksadk.runtime.adapter", "RuntimeAdapter"),
+    "RuntimeAdapterFactory": ("ksadk.runtime.adapter", "RuntimeAdapterFactory"),
+    "RuntimeExecutor": ("ksadk.runtime.executor", "RuntimeExecutor"),
+    "RuntimeStartPreparation": (
+        "ksadk.runtime.executor",
+        "RuntimeStartPreparation",
+    ),
+    "RuntimeLaunchContext": ("ksadk.runtime.adapter", "RuntimeLaunchContext"),
+    "RuntimeRegistry": ("ksadk.runtime.adapter", "RuntimeRegistry"),
+    "RuntimeServices": ("ksadk.runtime.adapter", "RuntimeServices"),
+    "StartRequest": ("ksadk.runtime.adapter", "StartRequest"),
+    "build_default_runtime_registry": (
+        "ksadk.runtime.factory",
+        "build_default_runtime_registry",
+    ),
+    "create_runtime_adapter": ("ksadk.runtime.factory", "create_runtime_adapter"),
+}
+
+__all__ = list(_EXPORTS)
+
+
+def __getattr__(name: str) -> Any:
+    try:
+        module_name, attribute = _EXPORTS[name]
+    except KeyError:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}") from None
+    value = getattr(import_module(module_name), attribute)
+    globals()[name] = value
+    return value
+
+
+def __dir__() -> list[str]:
+    return sorted({*globals(), *__all__})

ksadk/runtime/_runner_adapter/__init__.py

@@ -0,0 +1 @@
+"""Internal runner_adapter implementation subpackage."""