mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-11-15 07:41:20 +00:00
9cb2bde557
Bundled examples keep their behavior tests in backend/tests/, so the core backend workflow ran optional extension behavior on every pull request. Mark those suites with pytest.mark.bundled_extension, exclude them from the core shards and from the duration baseline, and run them in a dedicated workflow through make test-bundled-extensions. Tests of host mechanisms stay in the core shards even when an example package is the payload. The marked suites import deerflow.agents, config, runtime, knowledge and app.gateway, so the workflow triggers for backend/packages/**, backend/app/gateway/** and the marked test modules; a backend test pins those paths, the make target and both marker selections, and fails when a marked module is missing from the path list. Host/extension compatibility checks — a host API gate and authorization of contributed routes — live in test_extension_host_contracts.py so they still fail core CI. The committed duration baseline drops the 482 entries for the excluded suites. Refs #8173 Co-authored-by: REI-330 <173389910+REI-330@users.noreply.github.com> Co-authored-by: willem-bd <143703838+willem-bd@users.noreply.github.com> Co-authored-by: Eilen6316 <136898293+Eilen6316@users.noreply.github.com>
87 lines
4.1 KiB
Python
87 lines
4.1 KiB
Python
"""Standalone Pi-inspired plugin: persistence, ownership and real tool dispatch."""
|
|
|
|
import asyncio
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from deerflow_extension_api.auth import ExtensionPrincipal
|
|
from deerflow_extension_api.plugins import ActionContext
|
|
from langchain_core.messages import AIMessage
|
|
from langgraph.graph import END, START, MessagesState, StateGraph
|
|
from langgraph.prebuilt import ToolNode
|
|
|
|
from deerflow.extensions.loader import ExtensionSpec, load_extensions
|
|
from deerflow.extensions.plugin_tools import build_plugin_tools
|
|
|
|
pytestmark = pytest.mark.bundled_extension
|
|
|
|
|
|
@pytest.fixture
|
|
def bookmarks(tmp_path, monkeypatch):
|
|
monkeypatch.syspath_prepend(str(Path(__file__).resolve().parents[2] / "examples/deerflow-extension-bookmarks"))
|
|
|
|
def load():
|
|
loaded, diagnostics = load_extensions([ExtensionSpec(use="deerflow_extension_bookmarks:install", config={"enabled": True, "storage_path": str(tmp_path / "bookmarks.sqlite")})])
|
|
assert not diagnostics
|
|
return loaded
|
|
|
|
loaded = load()
|
|
((_, plugin),) = loaded.plugins
|
|
actions = {action.name: action.handler for action in plugin.backend}
|
|
return loaded, actions, load
|
|
|
|
|
|
def user(name):
|
|
return ActionContext(ExtensionPrincipal(name), {"enabled": True})
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_save_search_rename_delete_are_owner_scoped_and_persisted(bookmarks):
|
|
_, actions, reload_plugin = bookmarks
|
|
payload = {"thread_id": "thread-1", "message_id": "answer-1", "label": "Release notes", "text": "ORCHID launches on Friday."}
|
|
saved = await actions["save"](payload, user("alice"))
|
|
# Retrying save does not create duplicate bookmarks.
|
|
assert (await actions["save"](payload, user("alice")))["id"] == saved["id"]
|
|
assert (await actions["search"]({"query": "orchid"}, user("alice")))["items"][0]["text"] == payload["text"]
|
|
assert (await actions["search"]({"query": ""}, user("bob")))["items"] == []
|
|
for action, args in [("rename", {"id": saved["id"], "label": "Stolen"}), ("delete", {"id": saved["id"]})]:
|
|
with pytest.raises(ValueError):
|
|
await actions[action](args, user("bob"))
|
|
await actions["rename"]({"id": saved["id"], "label": "Launch plan"}, user("alice"))
|
|
((_, plugin),) = reload_plugin().plugins
|
|
search = next(a.handler for a in plugin.backend if a.name == "search")
|
|
assert (await search({"query": "Launch plan"}, user("alice")))["items"][0]["label"] == "Launch plan"
|
|
await actions["delete"]({"id": saved["id"]}, user("alice"))
|
|
assert (await search({"query": ""}, user("alice")))["items"] == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_concurrent_duplicate_save_and_payload_identity_rejection(bookmarks):
|
|
_, actions, _ = bookmarks
|
|
payload = {"thread_id": "t", "message_id": "m", "label": "One", "text": "Hello"}
|
|
values = await asyncio.gather(*(actions["save"](payload, user("alice")) for _ in range(6)))
|
|
assert len({v["id"] for v in values}) == 1
|
|
with pytest.raises(ValueError):
|
|
await actions["save"]({**payload, "user_id": "bob"}, user("alice"))
|
|
with pytest.raises(ValueError):
|
|
await actions["save"]({**payload, "text": "x" * 12001}, user("alice"))
|
|
# Query is literal text, not SQL wildcard syntax.
|
|
assert (await actions["search"]({"query": "%"}, user("alice")))["items"] == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_model_tool_reads_only_authenticated_users_bookmarks(bookmarks):
|
|
loaded, actions, _ = bookmarks
|
|
await actions["save"]({"thread_id": "t", "message_id": "m", "label": "Release", "text": "ORCHID Friday"}, user("alice"))
|
|
(tool,) = build_plugin_tools(loaded)
|
|
assert "user_id" not in tool.tool_call_schema.get("properties", {})
|
|
graph = StateGraph(MessagesState)
|
|
graph.add_node("tools", ToolNode([tool]))
|
|
graph.add_edge(START, "tools")
|
|
graph.add_edge("tools", END)
|
|
runtime = graph.compile()
|
|
for owner, count in [("alice", 1), ("bob", 0)]:
|
|
result = await runtime.ainvoke({"messages": [AIMessage(content="", tool_calls=[{"name": tool.name, "args": {"query": "ORCHID"}, "id": "c"}])]}, context={"user_id": owner})
|
|
assert len(json.loads(result["messages"][-1].content)["items"]) == count
|