## Summary epic r04 - begin refactor ## Change Type - [x] Cowork feature - [ ] Bug fix - [ ] Core AI contribution - [ ] Test / hardening - [ ] Performance - [ ] Documentation ## Related Work Cowork Task: Core Repo: http://34.143.229.138/gitea-admin/fsg-ai-core-assets Core AI Issue: Core Task: Related PR: ## Scope What is intentionally included? What is intentionally NOT included? ## Validation - [ ] Unit tests - [ ] Integration tests - [ ] Manual verification - [ ] Regression check Commands / evidence: ## Security Impact Permission / credential / network / customer data impact: ## Compatibility - [ ] No breaking change - [ ] Breaking change documented ## Reviewer Notes Anything Cowork reviewers should pay attention to. --------- Co-authored-by: Anh Tran Nguyen Minh <anhtnm1@fpt.com> Co-authored-by: Huong Le Thi Thien <huongltt35@fpt.com> Co-authored-by: Nam Pham Dinh Thanh <nampdt@fpt.com> Co-authored-by: Vu Dam Tuan <vudt15@fpt.com> Co-authored-by: Hiep Ha Van <hiephv3@fpt.com> Co-authored-by: Lam Hoang Van <lamhv7@fpt.com> Reviewed-on: #7 Co-authored-by: Duy Le Huu <duylh19@fpt.com>
This commit was merged in pull request #7.
This commit is contained in:
@@ -0,0 +1,86 @@
|
||||
"""Đọc dữ liệu dán/kéo-thả vào ô soạn tin, và nhận diện lệnh trả lời tại chỗ.
|
||||
|
||||
Tách khỏi ``composer_widget.py`` vì hai bên đều cần: ô nhập
|
||||
(``chat_input_box.py::_Input``) bắt sự kiện dán và thả, còn ``Composer`` bọc
|
||||
ngoài quyết định tệp nào được nhận. Đợt tách widget (R08) để mấy hàm này ở lại
|
||||
trong ``composer_widget.py`` trong khi phần gọi chúng đã sang
|
||||
``chat_input_box.py`` — thành ra dán hay thả tệp vào ô chat đều ném
|
||||
``NameError`` (F-12). Đặt ở một module thứ ba là chỗ duy nhất không lặp lại
|
||||
được lỗi ấy.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import re
|
||||
from datetime import datetime
|
||||
from typing import List
|
||||
|
||||
from PySide6.QtGui import QImage
|
||||
|
||||
from ...config import CONFIG_DIR
|
||||
|
||||
#: ``/skill`` hoặc ``/skill:<tên>`` đứng một mình — không có phần yêu cầu theo sau.
|
||||
_BARE_SKILL = re.compile(r"^/skill:[\w\-.]+$")
|
||||
|
||||
#: ``/agent`` hoặc ``/agent:<tên>`` đứng một mình.
|
||||
_BARE_AGENT = re.compile(r"^/agent:[\w\-.]+$")
|
||||
|
||||
|
||||
def save_pasted_image(image) -> str | None:
|
||||
"""Ghi ảnh vừa dán/kéo vào thư mục cấu hình; trả về đường dẫn, lỗi thì ``None``.
|
||||
|
||||
Tên tệp lấy tới phần nghìn giây vì dán liên tiếp trong cùng một giây là
|
||||
chuyện thường — tính tới giây thôi là ảnh sau đè ảnh trước.
|
||||
"""
|
||||
try:
|
||||
if not isinstance(image, QImage) or image.isNull():
|
||||
return None
|
||||
folder = CONFIG_DIR / "pasted"
|
||||
folder.mkdir(parents=True, exist_ok=True)
|
||||
name = "paste-" + datetime.now().strftime("%Y%m%d-%H%M%S-%f")[:-3] + ".png"
|
||||
path = folder / name
|
||||
if image.save(str(path), "PNG"):
|
||||
return str(path)
|
||||
except Exception: # noqa: BLE001 — dán ảnh hỏng không đáng làm vỡ ô nhập
|
||||
return None
|
||||
return None
|
||||
|
||||
|
||||
def is_local_skill_command(text: str) -> bool:
|
||||
"""``/skill`` (liệt kê) hoặc ``/skill:<tên>`` (chọn) — lệnh trả lời ngay tại chỗ.
|
||||
|
||||
Những lệnh này phải chạy được cả khi đang có lượt chat chạy dở, nên chúng đi
|
||||
thẳng chứ không vào hàng đợi. Khác với ``/skill:<tên> <yêu cầu>``: cái đó là
|
||||
một lượt chat thật và phải xếp hàng như mọi lượt khác.
|
||||
"""
|
||||
t = (text or "").strip()
|
||||
return t == "/skill" or bool(_BARE_SKILL.match(t))
|
||||
|
||||
|
||||
def is_local_agent_command(text: str) -> bool:
|
||||
"""Như :func:`is_local_skill_command` nhưng cho lệnh ``/agent``."""
|
||||
t = (text or "").strip()
|
||||
return t == "/agent" or bool(_BARE_AGENT.match(t))
|
||||
|
||||
|
||||
def paths_from_mime(md) -> List[str]:
|
||||
"""Rút danh sách đường dẫn tệp từ dữ liệu kéo-thả/dán.
|
||||
|
||||
Ưu tiên đường dẫn tệp thật; chỉ khi không có tệp nào mới xét tới ảnh trong
|
||||
bộ nhớ tạm và ghi nó ra tệp. Thứ tự này quan trọng: kéo một tệp ảnh từ
|
||||
Explorer thì dữ liệu mang CẢ đường dẫn lẫn ảnh xem trước, và ta muốn tệp
|
||||
gốc chứ không muốn một bản sao PNG.
|
||||
"""
|
||||
paths: List[str] = []
|
||||
if md.hasUrls():
|
||||
for u in md.urls():
|
||||
if u.isLocalFile():
|
||||
paths.append(u.toLocalFile())
|
||||
if not paths and md.hasImage():
|
||||
p = save_pasted_image(md.imageData())
|
||||
if p:
|
||||
paths.append(p)
|
||||
return paths
|
||||
|
||||
|
||||
__all__ = ["paths_from_mime", "save_pasted_image",
|
||||
"is_local_skill_command", "is_local_agent_command"]
|
||||
Reference in New Issue
Block a user