mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-08-11 15:28:37 +00:00
00161811bd
* feat: add workspace change review * chore: format workspace change files * fix: optimize workspace change summaries * style: refine workspace change badge scale * fix: restore workspace change user context import * fix(frontend): gate workspace change badge to assistant messages Only pass run_id to assistant MessageListItems so the workspace-change badge can never render under a user's prompt, which carries the same run_id. Assert single badge render in the E2E flow. * fix(workspace-changes): address review feedback on diff parsing and badge - Restrict unified-diff header detection to "+++ "/"--- " (trailing space) in both backend _count_diff_lines and frontend getWorkspaceChangeLineClass so content lines beginning with +++/--- are counted/styled correctly - Gate WorkspaceChangeBadge to ai messages so tool messages folded into an assistant group don't render a duplicate badge - Add regression tests for both diff-classification fixes - Apply prettier formatting to workspace-changes files flagged by CI --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from .types import WORKSPACE_CHANGES_EVENT_TYPE, WORKSPACE_CHANGES_METADATA_KEY
|
|
|
|
EMPTY_SUMMARY = {
|
|
"created": 0,
|
|
"modified": 0,
|
|
"deleted": 0,
|
|
"additions": 0,
|
|
"deletions": 0,
|
|
"truncated": False,
|
|
}
|
|
|
|
|
|
async def get_workspace_changes_response(
|
|
event_store: Any,
|
|
thread_id: str,
|
|
run_id: str,
|
|
*,
|
|
include_files: bool = True,
|
|
include_diff: bool = True,
|
|
) -> dict[str, Any]:
|
|
events = await event_store.list_events(
|
|
thread_id,
|
|
run_id,
|
|
event_types=[WORKSPACE_CHANGES_EVENT_TYPE],
|
|
limit=10,
|
|
)
|
|
if not events:
|
|
return _empty_response()
|
|
|
|
payload = _extract_workspace_changes_payload(events[-1])
|
|
if not isinstance(payload, dict):
|
|
return _empty_response()
|
|
|
|
response = dict(payload)
|
|
response["available"] = True
|
|
response.setdefault("summary", dict(EMPTY_SUMMARY))
|
|
if include_files:
|
|
response.setdefault("files", [])
|
|
if not include_diff:
|
|
response["files"] = [_without_diff(file) for file in response["files"]]
|
|
else:
|
|
response["files"] = []
|
|
return response
|
|
|
|
|
|
def _empty_response() -> dict[str, Any]:
|
|
return {
|
|
"available": False,
|
|
"version": 1,
|
|
"summary": dict(EMPTY_SUMMARY),
|
|
"files": [],
|
|
"limits": {},
|
|
}
|
|
|
|
|
|
def _extract_workspace_changes_payload(event: dict[str, Any]) -> Any:
|
|
metadata = event.get("metadata") or {}
|
|
if isinstance(metadata, dict) and WORKSPACE_CHANGES_METADATA_KEY in metadata:
|
|
return metadata[WORKSPACE_CHANGES_METADATA_KEY]
|
|
content = event.get("content")
|
|
if isinstance(content, dict):
|
|
return content
|
|
return None
|
|
|
|
|
|
def _without_diff(file: Any) -> Any:
|
|
if not isinstance(file, dict):
|
|
return file
|
|
sanitized = dict(file)
|
|
sanitized["diff"] = ""
|
|
return sanitized
|