mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:13:17 +08:00
oliver
4a23b715a2
本次迁移将网关/通道/工具/技能/脚本与协议资源集中到新结构，统一路径常量与脚本转发机制，减少顶层噪音并保证运行与测试行为一致。 Made-with: Cursor
119 lines
4 KiB
Python
119 lines
4 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from oclaw.runtime.agent_core_run import AgentCoreRunInput, run_agent_core
|
|
from oclaw.runtime.types import StandardMessage
|
|
from oclaw.platform.llm.chat_models import StaticTextChatModel
|
|
from oclaw.platform.persistence.sqlite_store import SqliteStore
|
|
from oclaw.runtime.tools.base import ToolRegistry
|
|
|
|
|
|
def _msg(session_id: str, *, metadata: dict | None = None) -> StandardMessage:
|
|
return StandardMessage(
|
|
session_id=session_id,
|
|
tenant_id="t1",
|
|
user_id="u1",
|
|
role="member",
|
|
channel="admin_chat",
|
|
text="hello",
|
|
attachments=[],
|
|
metadata=metadata if isinstance(metadata, dict) else {"tenant_id": "t1", "user_id": "u1"},
|
|
)
|
|
|
|
|
|
def test_agent_core_run_started_trace_payload_keys(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
store = SqliteStore(str(tmp_path / "trace.sqlite"))
|
|
sess = store.create_session("t")
|
|
events: list[dict] = []
|
|
real_add = store.add_trace_event
|
|
|
|
def _wrap(**kwargs: object) -> None:
|
|
events.append(dict(kwargs))
|
|
return real_add(**kwargs)
|
|
|
|
monkeypatch.setattr(store, "add_trace_event", _wrap)
|
|
|
|
trace_id = "trace-abc"
|
|
run_agent_core(
|
|
store=store,
|
|
data=AgentCoreRunInput(
|
|
msg=_msg(sess.id),
|
|
lang="zh",
|
|
system_prompt="sys",
|
|
model=StaticTextChatModel("done"),
|
|
tools=ToolRegistry([]),
|
|
trace_id=trace_id,
|
|
parent_span_id=None,
|
|
max_attempts=1,
|
|
oclaw_task_id="task-99",
|
|
oclaw_worker_id="worker-zz",
|
|
),
|
|
)
|
|
|
|
started = [e for e in events if e.get("event_type") == "run_started"]
|
|
assert started, events
|
|
pl = started[0].get("payload") or {}
|
|
assert pl.get("pipeline") == "oclaw_agent_core"
|
|
assert pl.get("oc_stage") == "run_start"
|
|
assert pl.get("trace_id") == trace_id
|
|
assert pl.get("lang") == "zh"
|
|
assert "run_id" in pl
|
|
assert pl.get("oclaw_task_id") == "task-99"
|
|
assert pl.get("oclaw_worker_id") == "worker-zz"
|
|
assert pl.get("relay_envelope_present") is False
|
|
assert pl.get("relay_envelope_pointer_count") == 0
|
|
|
|
att = [e for e in events if e.get("event_type") == "attempt_started"]
|
|
assert att
|
|
apl = att[0].get("payload") or {}
|
|
assert apl.get("oc_stage") == "attempt"
|
|
assert apl.get("attempt_no") == 1
|
|
|
|
|
|
def test_agent_core_trace_includes_relay_envelope_stats(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
store = SqliteStore(str(tmp_path / "trace2.sqlite"))
|
|
sess = store.create_session("t")
|
|
events: list[dict] = []
|
|
real_add = store.add_trace_event
|
|
|
|
def _wrap(**kwargs: object) -> None:
|
|
events.append(dict(kwargs))
|
|
return real_add(**kwargs)
|
|
|
|
monkeypatch.setattr(store, "add_trace_event", _wrap)
|
|
run_agent_core(
|
|
store=store,
|
|
data=AgentCoreRunInput(
|
|
msg=_msg(
|
|
sess.id,
|
|
metadata={
|
|
"tenant_id": "t1",
|
|
"user_id": "u1",
|
|
"relay_share_envelope": {
|
|
"schema_version": "v1",
|
|
"attachments": {"pointers": [{"pointer_uri": "relay://attachments/scope_1/abcdef123456"}]},
|
|
},
|
|
},
|
|
),
|
|
lang="zh",
|
|
system_prompt="sys",
|
|
model=StaticTextChatModel("done"),
|
|
tools=ToolRegistry([]),
|
|
trace_id="trace-relay",
|
|
parent_span_id=None,
|
|
max_attempts=1,
|
|
),
|
|
)
|
|
started = [e for e in events if e.get("event_type") == "run_started"]
|
|
assert started
|
|
sp = started[0].get("payload") or {}
|
|
assert sp.get("relay_envelope_present") is True
|
|
assert sp.get("relay_envelope_pointer_count") == 1
|
|
att = [e for e in events if e.get("event_type") == "attempt_started"]
|
|
assert att
|
|
ap = att[0].get("payload") or {}
|
|
assert ap.get("relay_envelope_present") is True
|
|
assert ap.get("relay_envelope_pointer_count") == 1
|