1b8429e33a
CI / test (push) Canceled after 0s
fix các bug theo yêu cầu https://fptsoftware362-my.sharepoint.com/❌/g/personal/nampdt_fpt_com/IQAHBJ4A9xqDTLgvt2bhukJEAdRB5LRz2hbJpTivvIiBSYM?wdExp=TEAMS-TREATMENT&web=1&isSPOFile=1&ovuser=f01e930a-b52e-42b1-b70f-a8882b5d043b%2CAnhTNM1%40fpt.com&clickparams=eyJBcHBOYW1lIjoiVGVhbXMtRGVza3RvcCIsIkFwcFZlcnNpb24iOiI0OS8yNjA4MTMxOTMxNyIsIkhhc0ZlZGVyYXRlZFVzZXIiOmZhbHNlfQ%3D%3D --------- Co-authored-by: Duy Le Huu <duylh19@fpt.com> Reviewed-on: #10 Co-authored-by: Anh Tran Nguyen Minh <anhtnm1@fpt.com>
309 lines
16 KiB
Python
309 lines
16 KiB
Python
"""Cột trái: thư viện workflow, agent, skill — R08-T09.
|
|
|
|
Bốn mục gập được (WORKFLOWS / AGENTS / SKILLS / FLOW STATUS). Trạng thái gập
|
|
của từng mục là thứ người dùng đặt rồi mong nó giữ nguyên, nên nó nằm trong
|
|
cấu hình chứ không phải trong widget.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from typing import List
|
|
from PySide6.QtCore import QSize, Qt
|
|
from PySide6.QtWidgets import QHBoxLayout, QListWidget, QListWidgetItem, QPushButton, QSplitter, QVBoxLayout, QWidget
|
|
from ...core import co4e, skills as skills_mod
|
|
from ...i18n import bind_dynamic, bind_text, bind_tip, tr
|
|
from ...ui.icons import icon
|
|
from ...presentation.co4e.agent_list_panel import AgentListPanel
|
|
from ...presentation.co4e.palette_list import _PaletteList
|
|
from ...presentation.co4e.skills_list_panel import SkillsListPanel
|
|
|
|
|
|
def _skill_prefix_lookup(all_skills):
|
|
"""Answer ``skills.skill_prefix_for`` from an ALREADY-LOADED skill list.
|
|
|
|
``skill_prefix_for`` re-reads the whole skill folder on every call, so
|
|
asking it once per skill made a sidebar reload cost one full disk scan per
|
|
skill — measured at ~3.8s of frozen GUI thread on a 121-skill library, and
|
|
that reload runs on every language switch.
|
|
|
|
The scan order and the blank-instructions rule are copied from
|
|
``skill_prefix_for`` deliberately: a namesake with no instructions must NOT
|
|
end the search, or a skill's text silently becomes empty in an agent prompt.
|
|
"""
|
|
cache: dict = {}
|
|
|
|
def lookup(name: str) -> str:
|
|
"""The ``## Skill: <name>\\n<instructions>`` block for one name, or ''."""
|
|
if not name:
|
|
return ""
|
|
low = name.strip().lower()
|
|
if low not in cache:
|
|
cache[low] = next(
|
|
(f"## Skill: {s.name}\n{s.instructions.strip()}" for s in all_skills
|
|
if (s.slug == low or s.name.lower() == low) and s.instructions.strip()),
|
|
"")
|
|
return cache[low]
|
|
|
|
return lookup
|
|
|
|
|
|
class Co4ESidebarMixin:
|
|
"""Cột trái của Co4E Studio: Workflows, Agents, Skills và Flow Status."""
|
|
def _build_sidebar(self) -> QWidget:
|
|
# ONE COLUMN, four named sections — no icon tabs. Every list is on screen
|
|
# at once, so "what can I drag onto the canvas" is answered by looking
|
|
# rather than by clicking through three unlabeled tabs.
|
|
# A vertical splitter, not a fixed stack: on a short window four stacked
|
|
# lists otherwise squeeze down to one visible row each. The splitter
|
|
# hands out the available height by weight and lets the user re-balance
|
|
# it by dragging; each list keeps a small minimum so none disappears.
|
|
"""Dựng cột trái: MỘT cột, bốn mục có tên, không dùng tab icon.
|
|
|
|
Mọi danh sách hiện cùng lúc nên câu hỏi "kéo được gì lên khung vẽ" trả lời
|
|
bằng cách NHÌN, không phải bấm qua ba tab không nhãn. Dùng splitter dọc chứ
|
|
không xếp cứng: cửa sổ thấp thì bốn danh sách chồng nhau sẽ bị bóp còn đúng
|
|
một dòng mỗi cái; splitter chia chiều cao theo trọng số và cho người dùng
|
|
tự cân lại.
|
|
"""
|
|
self._sections: dict = {}
|
|
self.sidebar = QWidget()
|
|
outer_col = QVBoxLayout(self.sidebar)
|
|
outer_col.setContentsMargins(6, 6, 6, 6)
|
|
outer_col.setSpacing(0)
|
|
self.side_split = QSplitter(Qt.Vertical)
|
|
self.side_split.setChildrenCollapsible(False)
|
|
self.side_split.setHandleWidth(8)
|
|
outer_col.addWidget(self.side_split, 1)
|
|
|
|
class _Col:
|
|
"""Adapter so the section builders below read the same as before."""
|
|
|
|
def __init__(self, split):
|
|
"""Bọc một ``QSplitter`` để phần còn lại thao tác với nó như một cột."""
|
|
self._split = split
|
|
|
|
def addWidget(self, w, stretch=1):
|
|
"""Thêm một mục vào splitter dọc kèm trọng số chia chiều cao.
|
|
|
|
Lớp bọc nhỏ này cho phép dùng splitter ở chỗ mã cũ đang gọi theo API của
|
|
layout, nên phần dựng mục không phải sửa.
|
|
"""
|
|
self._split.addWidget(w)
|
|
self._split.setStretchFactor(self._split.count() - 1, stretch)
|
|
|
|
col = _Col(self.side_split)
|
|
|
|
# --- WORKFLOWS ---------------------------------------------------
|
|
# Bound, not set once: Co4ETab._retranslate reloads the sidebar's LIST
|
|
# CONTENTS, but these headings, buttons and tooltips are built here and
|
|
# nothing re-applied them — they stayed in the language of app start-up.
|
|
self.wf_new_btn = bind_text(QPushButton(), "co4e.new")
|
|
self.wf_new_btn.setIcon(icon("plus"))
|
|
bind_tip(self.wf_new_btn, "co4e.tt_new_wf")
|
|
self.wf_new_btn.setObjectName("co4eSectionAction")
|
|
self.wf_new_btn.setFlat(True)
|
|
self.wf_new_btn.setCursor(Qt.PointingHandCursor)
|
|
self.wf_new_btn.clicked.connect(self._new_workflow)
|
|
wf_body = QWidget(); wl = QVBoxLayout(wf_body)
|
|
wl.setContentsMargins(0, 0, 0, 0); wl.setSpacing(4)
|
|
# Draggable: drag a flow onto the canvas to merge it in (Nova-style);
|
|
# double-click loads it onto the canvas.
|
|
self.wf_list = _PaletteList(payload_role=Qt.UserRole + 2)
|
|
bind_tip(self.wf_list, "co4e.drag_hint")
|
|
self.wf_list.itemDoubleClicked.connect(self._load_selected_workflow)
|
|
self.wf_list.setContextMenuPolicy(Qt.CustomContextMenu)
|
|
self.wf_list.customContextMenuRequested.connect(self._wf_context_menu)
|
|
wl.addWidget(self.wf_list, 1)
|
|
wf_btns = QHBoxLayout(); wf_btns.setSpacing(4)
|
|
self.wf_edit_btn = self._icon_btn("edit", "co4e.tt_edit_wf", self._edit_selected_workflow)
|
|
self.wf_dup_btn = self._icon_btn("branch", "co4e.tt_dup_wf", self._duplicate_selected_workflow)
|
|
self.wf_del_btn = self._icon_btn("trash", "co4e.tt_del_wf", self._delete_selected_workflow)
|
|
for b in (self.wf_edit_btn, self.wf_dup_btn, self.wf_del_btn):
|
|
wf_btns.addWidget(b)
|
|
wf_btns.addStretch(1)
|
|
wl.addLayout(wf_btns)
|
|
# Its own row: sharing one line with the three icon buttons cut "Chạy
|
|
# nền" down to "Chạ" as soon as the sidebar hit its narrow width.
|
|
self.wf_runbg_btn = bind_text(QPushButton(), "co4e.run_bg")
|
|
self.wf_runbg_btn.setIcon(icon("play"))
|
|
bind_tip(self.wf_runbg_btn, "co4e.tt_run_bg")
|
|
self.wf_runbg_btn.clicked.connect(self._run_selected_in_background)
|
|
wl.addWidget(self.wf_runbg_btn)
|
|
col.addWidget(self._section("co4e.tab_workflows", wf_body, self.wf_new_btn), 3)
|
|
|
|
# --- AGENTS ------------------------------------------------------
|
|
# Widget cua khu vuc nay da doi sang AgentListPanel (xem
|
|
# presentation/co4e/agent_list_panel.py); o day chi con giu
|
|
# ag_new_btn/agent_list/ag_edit_btn/ag_del_btn nhu 4 ten thuoc tinh cu
|
|
# va tu noi signal - dung nguyen tac panel khong tu wire, Co4ETab moi
|
|
# biet _new_agent/_edit_agent/_delete_agent.
|
|
self._agent_panel = AgentListPanel()
|
|
self.ag_new_btn = self._agent_panel.new_btn
|
|
self.ag_new_btn.clicked.connect(self._new_agent)
|
|
self.agent_list = self._agent_panel.list_widget
|
|
self.ag_edit_btn = self._agent_panel.edit_btn
|
|
self.ag_edit_btn.clicked.connect(self._edit_agent)
|
|
self.ag_del_btn = self._agent_panel.del_btn
|
|
self.ag_del_btn.clicked.connect(self._delete_agent)
|
|
col.addWidget(self._section("co4e.tab_agents", self._agent_panel, self.ag_new_btn), 3)
|
|
|
|
# --- SKILLS ------------------------------------------------------
|
|
# Widget cua khu vuc nay da doi sang SkillsListPanel (xem
|
|
# presentation/co4e/skills_list_panel.py); o day chi con giu
|
|
# sk_manage_btn/skill_list nhu 2 ten thuoc tinh cu va tu noi signal -
|
|
# dung nguyen tac panel khong tu wire, Co4ETab moi biet _manage_skills.
|
|
self._skills_panel = SkillsListPanel()
|
|
self.sk_manage_btn = self._skills_panel.manage_btn
|
|
self.sk_manage_btn.clicked.connect(self._manage_skills)
|
|
self.skill_list = self._skills_panel.list_widget
|
|
col.addWidget(self._section("co4e.tab_skills", self._skills_panel, self.sk_manage_btn), 2)
|
|
|
|
# --- RUNS --------------------------------------------------------
|
|
# A short, always-visible view of the same runs the Flow Status page
|
|
# tables in full. Clicking one opens that page with the run selected.
|
|
# Icon only: the heading beside it already reads FLOW STATUS, and the
|
|
# label was long enough to be cut in half in a narrow sidebar.
|
|
self.runs_more_btn = QPushButton()
|
|
self.runs_more_btn.setIcon(icon("chevron-right"))
|
|
self.runs_more_btn.setFixedWidth(30)
|
|
self.runs_more_btn.setFlat(True)
|
|
bind_tip(self.runs_more_btn, "co4e.tt_runs_tab")
|
|
self.runs_more_btn.clicked.connect(lambda: self._show_runs(True))
|
|
runs_body = QWidget(); rl = QVBoxLayout(runs_body)
|
|
rl.setContentsMargins(0, 0, 0, 0); rl.setSpacing(4)
|
|
self.runs_side_list = QListWidget()
|
|
bind_tip(self.runs_side_list, "co4e.tt_runs_tab")
|
|
self.runs_side_list.itemClicked.connect(self._on_side_run_clicked)
|
|
rl.addWidget(self.runs_side_list, 1)
|
|
col.addWidget(self._section("co4e.runs_tab", runs_body, self.runs_more_btn), 2)
|
|
# Small enough that all four still fit on a laptop screen, large enough
|
|
# that each shows more than a single row.
|
|
for lst in (self.wf_list, self.agent_list, self.skill_list, self.runs_side_list):
|
|
lst.setMinimumHeight(56)
|
|
return self.sidebar
|
|
def _refresh_side_runs(self) -> None:
|
|
"""Mirror the newest runs into the sidebar's short list."""
|
|
lst = getattr(self, "runs_side_list", None)
|
|
if lst is None:
|
|
return
|
|
dots = {"running": "▶", "done": "✓", "error": "✕", "stopped": "■"}
|
|
lst.clear()
|
|
for h in list(reversed(self.manager.runs()))[:self._SIDE_RUNS]:
|
|
it = QListWidgetItem(f"{dots.get(h.status, '•')} {h.name}"
|
|
f" {h.progress_text()}")
|
|
it.setData(Qt.UserRole, h.id)
|
|
it.setToolTip(f"{h.name} · {tr('co4e.status.' + h.status)} · {h.created_at or '-'}")
|
|
lst.addItem(it)
|
|
def _on_side_run_clicked(self, item) -> None:
|
|
"""Open the full Flow Status page with this run selected."""
|
|
run_id = item.data(Qt.UserRole)
|
|
self._show_runs(True)
|
|
for r in range(self.runs_table.rowCount()):
|
|
cell = self.runs_table.item(r, 0)
|
|
if cell is not None and cell.data(Qt.UserRole) == run_id:
|
|
self.runs_table.setCurrentCell(r, 0)
|
|
break
|
|
def _section(self, key: str, body: QWidget, action: QPushButton | None = None,
|
|
stretch: int = 1) -> QWidget:
|
|
"""One named, foldable section of the sidebar column.
|
|
|
|
Replaces the three icon-only tabs: all the lists are visible at once
|
|
(WORKFLOWS · AGENTS · SKILLS · runs), each under its own heading with the
|
|
action that belongs to it. Clicking the heading folds the section, so a
|
|
narrow window can still get to everything.
|
|
"""
|
|
box = QWidget()
|
|
v = QVBoxLayout(box)
|
|
v.setContentsMargins(0, 0, 0, 0)
|
|
v.setSpacing(2)
|
|
|
|
row = QHBoxLayout()
|
|
row.setContentsMargins(0, 0, 0, 0)
|
|
row.setSpacing(4)
|
|
head = QPushButton()
|
|
head.setObjectName("co4eSectionHdr")
|
|
head.setCheckable(True)
|
|
head.setChecked(True)
|
|
head.setCursor(Qt.PointingHandCursor)
|
|
head.setFlat(True)
|
|
head.toggled.connect(lambda on, w=body, b=box: self._fold_section(key, w, b, on))
|
|
row.addWidget(head, 1)
|
|
if action is not None:
|
|
row.addWidget(action, 0)
|
|
v.addLayout(row)
|
|
v.addWidget(body, 1)
|
|
|
|
self._sections[key] = (head, body, stretch)
|
|
# bind_dynamic, not bind_text: the heading is the fold arrow plus the
|
|
# translated name in caps, so re-applying it means re-running the whole
|
|
# line rather than pushing one key into setText.
|
|
bind_dynamic(head, lambda k=key: self._sync_section_arrow(k))
|
|
return box
|
|
def _fold_section(self, key: str, body: QWidget, box: QWidget, on: bool) -> None:
|
|
"""Fold/unfold a section AND give its height back to the others.
|
|
|
|
Inside a splitter, hiding the body is not enough — the pane keeps its
|
|
share of the height, so folding would free nothing. Clamping the whole
|
|
section to its header height makes the splitter re-deal the space.
|
|
"""
|
|
body.setVisible(on)
|
|
if on:
|
|
box.setMaximumHeight(16777215)
|
|
else:
|
|
box.setMaximumHeight(box.layout().itemAt(0).sizeHint().height() + 4)
|
|
self._sync_section_arrow(key)
|
|
def _sync_section_arrow(self, key: str) -> None:
|
|
"""Cập nhật mũi tên gập/mở và nhãn viết hoa của một mục ở cột trái."""
|
|
head, _body, _s = self._sections[key]
|
|
head.setText(("▾ " if head.isChecked() else "▸ ") + tr(key).upper())
|
|
def _icon_btn(self, icon_name: str, tip_key: str, slot) -> QPushButton:
|
|
"""Dựng một nút icon nhỏ (rộng 34px) kèm tooltip cho hàng công cụ của mục."""
|
|
b = QPushButton(); b.setIcon(icon(icon_name)); bind_tip(b, tip_key)
|
|
b.setFixedWidth(34)
|
|
b.clicked.connect(slot)
|
|
return b
|
|
def _reload_sidebar(self) -> None:
|
|
"""Nạp lại cả bốn danh sách ở cột trái: luồng, agent, skill và Flow Status."""
|
|
self.wf_list.clear()
|
|
for wf in co4e.list_workflows():
|
|
tag = tr("co4e.template") if wf.is_template else tr("co4e.saved")
|
|
it = QListWidgetItem(icon("workspaces"), f"{wf.name} · {tag}")
|
|
it.setData(Qt.UserRole, ("saved", wf.id))
|
|
it.setData(Qt.UserRole + 2, {"kind": "workflow", "workflow": co4e.workflow_to_dict(wf)})
|
|
self.wf_list.addItem(it)
|
|
# Agents: only the Parallel fan-out node + the user's own custom agents
|
|
# (create your own with "+ New agent"; drag onto the canvas). The blank
|
|
# "New Step" palette entry was removed — use the toolbar "+ Add" instead.
|
|
self.agent_list.clear()
|
|
self.agent_list.addItem(self._palette_item(
|
|
tr("co4e.parallel_node"), "server",
|
|
{"variant": "parallel", "label": "Parallel", "role": "PARALLEL", "icon": "server",
|
|
"sub_agents": []}))
|
|
for ca in co4e.list_custom_agents():
|
|
step = co4e.Step(label=ca.name, agent_slug=co4e.slugify(ca.name), role=ca.role or "AGENT",
|
|
icon=ca.icon, instructions=ca.instructions,
|
|
context=getattr(ca, "context", ""), model=ca.model,
|
|
permission_preset=ca.permission_preset, skills=list(ca.skills),
|
|
attachments=list(getattr(ca, "attachments", []) or []))
|
|
it = self._palette_item(f"{ca.name} · {ca.role} · {tr('co4e.custom')}", ca.icon or "robot",
|
|
co4e._step_dict(step))
|
|
it.setData(Qt.UserRole + 1, ca.id)
|
|
self.agent_list.addItem(it)
|
|
# Skills — the library is read ONCE here and both the names and the
|
|
# instructions come out of that one read (see _skill_prefix_lookup).
|
|
self.skill_list.clear()
|
|
all_skills = skills_mod.list_skills() + skills_mod.builtin_skills()
|
|
skill_prefix = _skill_prefix_lookup(all_skills)
|
|
for skill in all_skills:
|
|
name = skill.name
|
|
payload = co4e._step_dict(co4e.Step(
|
|
label=name, agent_slug=co4e.slugify(name), role="SKILL", icon="sparkle",
|
|
instructions=skill_prefix(name), skills=[name]))
|
|
self.skill_list.addItem(self._palette_item(name, "sparkle", payload))
|
|
@staticmethod
|
|
def _palette_item(text: str, icon_name: str, payload: dict) -> QListWidgetItem:
|
|
"""Dựng một mục trong bảng nguyên liệu, mang sẵn payload để kéo-thả lên khung vẽ."""
|
|
it = QListWidgetItem(icon(icon_name), text)
|
|
it.setData(Qt.UserRole, payload)
|
|
return it
|