9a3dc10d93
Redesign chat transcript + fix streaming/persistence lifecycle — v0.50.70 Squash-merges PR #587 by @aronprins (Aron Prins). Full credit to @aronprins for all feature and fix work. Transcript redesign: unified --msg-rail/--msg-max CSS variables, user turns as tinted cards, thinking cards as bordered panels, error card treatment, day-change separators, composer fade. Approval/clarify as composer flyouts: cards slide up from behind composer top, overflow:hidden + translateY clip prevents travel visibility, focus({preventScroll:true}). Streaming lifecycle: DOM order user→thinking→tool cards→response, no mid-stream jump. Live tool cards inserted before [data-live-assistant]. Persistence: reasoning attached before s.save(), _restore_reasoning_metadata on reload, role=tool rows preserved in S.messages, CLI-session tool-result fallback. Workspace panel FOUC fix: [data-workspace-panel] set at parse time. Docs: docs/ui-ux/index.html + two-stage-proposal.html. Maintainer additions (433b867): CHANGELOG v0.50.70, version badge, usage badge loop simplification. Reviewed and approved by @nesquena (independent review). 1361 tests passing.
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
"""Tests for backend tool-call summary extraction used by WebUI session persistence."""
|
|
import pathlib
|
|
import sys
|
|
|
|
REPO_ROOT = pathlib.Path(__file__).parent.parent.resolve()
|
|
sys.path.insert(0, str(REPO_ROOT))
|
|
|
|
from api.streaming import _extract_tool_calls_from_messages
|
|
|
|
|
|
def test_extract_tool_calls_from_openai_message_linkage():
|
|
messages = [
|
|
{"role": "user", "content": "ls"},
|
|
{
|
|
"role": "assistant",
|
|
"content": "",
|
|
"tool_calls": [{
|
|
"id": "call-1",
|
|
"function": {"name": "terminal", "arguments": '{"command":"ls"}'},
|
|
}],
|
|
},
|
|
{
|
|
"role": "tool",
|
|
"tool_call_id": "call-1",
|
|
"content": '{"output":"file.txt","exit_code":0}',
|
|
},
|
|
]
|
|
result = _extract_tool_calls_from_messages(messages)
|
|
assert len(result) == 1
|
|
assert result[0]["name"] == "terminal"
|
|
assert result[0]["assistant_msg_idx"] == 1
|
|
assert result[0]["snippet"] == "file.txt"
|
|
|
|
|
|
def test_extract_tool_calls_falls_back_to_live_progress_when_ids_missing():
|
|
messages = [
|
|
{"role": "user", "content": "write spec"},
|
|
{"role": "assistant", "content": "Starting."},
|
|
{"role": "tool", "content": '{"bytes_written":4955}'},
|
|
{"role": "assistant", "content": ""},
|
|
]
|
|
live_tool_calls = [{"name": "write_file", "args": {"path": "/tmp/SPEC.md"}}]
|
|
result = _extract_tool_calls_from_messages(messages, live_tool_calls=live_tool_calls)
|
|
assert len(result) == 1
|
|
assert result[0]["name"] == "write_file"
|
|
assert result[0]["assistant_msg_idx"] == 1
|
|
assert "bytes_written" in result[0]["snippet"]
|
|
assert result[0]["args"]["path"] == "/tmp/SPEC.md"
|
|
|
|
|
|
def test_extract_tool_calls_preserves_mixed_linked_and_fallback_results():
|
|
messages = [
|
|
{
|
|
"role": "assistant",
|
|
"content": "",
|
|
"tool_calls": [{"id": "call-1", "function": {"name": "terminal", "arguments": '{"command":"pwd"}'}}],
|
|
},
|
|
{"role": "tool", "tool_call_id": "call-1", "content": '{"output":"/tmp"}'},
|
|
{"role": "assistant", "content": "Next"},
|
|
{"role": "tool", "content": '{"result":"saved"}'},
|
|
]
|
|
live_tool_calls = [
|
|
{"name": "terminal", "args": {"command": "pwd"}},
|
|
{"name": "write_file", "args": {"path": "/tmp/out.txt"}},
|
|
]
|
|
result = _extract_tool_calls_from_messages(messages, live_tool_calls=live_tool_calls)
|
|
assert len(result) == 2
|
|
assert result[0]["name"] == "terminal"
|
|
assert result[1]["name"] == "write_file"
|
|
assert result[1]["assistant_msg_idx"] == 2
|
|
assert result[1]["snippet"] == "saved"
|