d5cf872311
- Replace a2a.utils.new_agent_text_message → a2a.helpers.new_text_message - Replace Part(root=TextPart(...)) → Part(text=...) (flat Part API) - Replace A2AStarletteApplication → Starlette route factories (create_agent_card_routes, create_jsonrpc_routes) - Update conftest stubs: remove a2a.server.apps/a2a.utils, add a2a.server.routes/a2a.helpers/AgentInterface - Add AgentInterface to AgentCard supported_interfaces - Rename snake_case AgentCard fields per 1.x schema Co-authored-by: Molecule AI Infra-Runtime-BE <infra-runtime-be@agents.moleculesai.app> Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
99 lines
3.9 KiB
Python
99 lines
3.9 KiB
Python
"""Test-env stubs for workspace-container deps.
|
|
|
|
`claude_agent_sdk` and `a2a` are installed inside the claude-code workspace
|
|
image, not in this package's test environment. Rather than pulling them as
|
|
dev deps (they're heavy + change frequently), we stub them before any test
|
|
module imports `molecule_runtime.*` — the stubs just need to be importable
|
|
so the package source doesn't crash at collect time.
|
|
|
|
Every known submodule + symbol used anywhere in `molecule_runtime/` is
|
|
stubbed here, so test_imports.py (walking every module) + test_session_
|
|
resume_gate.py (ClaudeSDKExecutor methods) + future tests all share one
|
|
consistent stub set.
|
|
|
|
NOTE: a2a-sdk 1.x (KI-009 migration) changed the module layout:
|
|
- REMOVED: a2a.server.apps (A2AStarletteApplication)
|
|
- REMOVED: a2a.utils
|
|
- ADDED: a2a.server.routes (create_agent_card_routes, create_jsonrpc_routes)
|
|
- ADDED: AgentInterface in a2a.types
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
|
|
|
|
def _ensure_package(dotted: str) -> types.ModuleType:
|
|
"""Create a package-shaped module at `dotted` and all its parents.
|
|
|
|
Key detail: the __path__ attribute makes Python treat the stub as a
|
|
PACKAGE, so `from parent.child import X` resolves correctly. Without
|
|
__path__ you get `'parent' is not a package` which is the exact bug
|
|
that broke our first-attempt stubs.
|
|
"""
|
|
parts = dotted.split(".")
|
|
for i in range(1, len(parts) + 1):
|
|
name = ".".join(parts[:i])
|
|
if name not in sys.modules:
|
|
mod = types.ModuleType(name)
|
|
mod.__path__ = [] # type: ignore[attr-defined]
|
|
sys.modules[name] = mod
|
|
if i > 1:
|
|
parent = sys.modules[".".join(parts[:i-1])]
|
|
child_name = parts[i-1]
|
|
if not hasattr(parent, child_name):
|
|
setattr(parent, child_name, sys.modules[name])
|
|
return sys.modules[dotted]
|
|
|
|
|
|
def _set_attrs(mod: types.ModuleType, attrs: list[str]) -> None:
|
|
"""Attach placeholder classes/functions. AgentExecutor is a class
|
|
so subclasses can inherit; everything else is a no-op callable."""
|
|
for attr in attrs:
|
|
if hasattr(mod, attr):
|
|
continue
|
|
if attr in ("AgentExecutor",):
|
|
setattr(mod, attr, type(attr, (), {}))
|
|
else:
|
|
setattr(mod, attr, lambda *a, **kw: None)
|
|
|
|
|
|
# claude_agent_sdk — used by claude_sdk_executor.py
|
|
if "claude_agent_sdk" not in sys.modules:
|
|
sdk_stub = types.ModuleType("claude_agent_sdk")
|
|
sdk_stub.ClaudeAgentOptions = lambda **kwargs: types.SimpleNamespace(**kwargs)
|
|
sdk_stub.query = lambda **kwargs: iter([])
|
|
sys.modules["claude_agent_sdk"] = sdk_stub
|
|
|
|
# a2a + submodules — every import path in the package gets stubbed.
|
|
# NOTE: a2a-sdk 1.x (KI-009 migration) removed a2a.server.apps and
|
|
# a2a.utils; added a2a.server.routes. Update stubs accordingly.
|
|
_A2A_MODULES: dict[str, list[str]] = {
|
|
"a2a": [],
|
|
"a2a.server": [],
|
|
"a2a.server.agent_execution": ["AgentExecutor", "RequestContext", "RequestContextBuilder"],
|
|
"a2a.server.events": ["EventQueue"],
|
|
"a2a.server.tasks": ["TaskUpdater", "InMemoryTaskStore"],
|
|
"a2a.server.routes": [
|
|
"create_agent_card_routes",
|
|
"create_jsonrpc_routes",
|
|
"create_rest_routes",
|
|
"DefaultServerCallContextBuilder",
|
|
],
|
|
"a2a.server.request_handlers": ["DefaultRequestHandler"],
|
|
"a2a.types": [
|
|
# TextPart removed in 1.x; Part is now flat (Part(text="...") instead of Part(root=TextPart(...)))
|
|
"Part", "AgentCard", "AgentCapabilities", "AgentSkill",
|
|
"AgentInterface", # added in a2a-sdk 1.x
|
|
"TaskStatus", "TaskState", "TaskStatusUpdateEvent",
|
|
],
|
|
"a2a.helpers": [
|
|
# Added in 1.x; replaces a2a.utils
|
|
"new_text_message", "new_task", "new_artifact",
|
|
"get_message_text", "get_text_parts",
|
|
],
|
|
}
|
|
for dotted, attrs in _A2A_MODULES.items():
|
|
mod = _ensure_package(dotted)
|
|
_set_attrs(mod, attrs)
|