youngkingdom/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-05-06 02:07:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refa: follow-up expose agent structured outputs in non-stream completions (#13524)

Browse Source 
### What problem does this PR solve?

Follow-up expose agent structured outputs in non-stream completions
#13389.

### Type of change

- [x] Documentation Update
- [x] Refactoring

---------

Co-authored-by: writinwaters <cai.keith@gmail.com>
This commit is contained in:
Yongteng Lei
2026-03-17 17:11:27 +08:00
committed by GitHub
parent ca182dc188
commit ca6c3218c3
3 changed files with 48 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
test/testcases/test_http_api/test_session_management/test_session_sdk_routes_unit.py
View File

@ -862,7 +862,18 @@ def test_agent_completions_stream_and_nonstream_unit(monkeypatch):
async def _agent_stream(*_args, **_kwargs):
yield "data:not-json"
yield "data:" + json.dumps({"event": "node_finished", "data": {"component_id": "c1"}})
yield "data:" + json.dumps(
{
"event": "node_finished",
"data": {"component_id": "c1", "outputs": {"structured": {"alpha": 1}}},
}
)
yield "data:" + json.dumps(
{
"event": "node_finished",
"data": {"component_id": "c2", "outputs": {"structured": {}}},
}
)
yield "data:" + json.dumps({"event": "other", "data": {}})
yield "data:" + json.dumps({"event": "message", "data": {"content": "hello"}})
@ -878,14 +889,36 @@ def test_agent_completions_stream_and_nonstream_unit(monkeypatch):
async def _agent_nonstream(*_args, **_kwargs):
yield "data:" + json.dumps({"event": "message", "data": {"content": "A", "reference": {"doc": "r"}}})
yield "data:" + json.dumps({"event": "node_finished", "data": {"component_id": "c2"}})
yield "data:" + json.dumps(
{
"event": "node_finished",
"data": {"component_id": "c2", "outputs": {"structured": {"foo": "bar"}}},
}
)
yield "data:" + json.dumps(
{
"event": "node_finished",
"data": {"component_id": "c3", "outputs": {"structured": {"baz": 1}}},
}
)
yield "data:" + json.dumps(
{
"event": "node_finished",
"data": {"component_id": "c4", "outputs": {"structured": {}}},
}
)
monkeypatch.setattr(module, "agent_completion", _agent_nonstream)
monkeypatch.setattr(module, "get_request_json", lambda: _AwaitableValue({"stream": False, "return_trace": True}))
res = _run(inspect.unwrap(module.agent_completions)("tenant-1", "agent-1"))
assert res["data"]["data"]["content"] == "A"
assert res["data"]["data"]["reference"] == {"doc": "r"}
assert res["data"]["data"]["trace"][0]["component_id"] == "c2"
assert res["data"]["data"]["structured"] == {
"c2": {"foo": "bar"},
"c3": {"baz": 1},
"c4": {},
}
assert [item["component_id"] for item in res["data"]["data"]["trace"]] == ["c2", "c3", "c4"]
async def _agent_nonstream_broken(*_args, **_kwargs):
yield "data:{"