.1915763867792649:61a8e5205d9edfb5e731e02c2d8f826a_69de652c5b35a0d4139f7f98.69de6d225b35a0d4139f8002.69de6d221f2e35bf02e01ba5:Trae CN.T(2026/4/15 00:36:50)

apps/api/deps.py

@@ -106,6 +106,7 @@ def paper_list_response(papers: list, repo: PaperRepository) -> dict:
     """论文列表统一序列化"""
     paper_ids = [str(p.id) for p in papers]
     topic_map = repo.get_topic_names_for_papers(paper_ids)
+    tag_map = repo.get_tags_for_papers(paper_ids)
     return {
         "items": [
             {
@@ -123,6 +124,7 @@ def paper_list_response(papers: list, repo: PaperRepository) -> dict:
                 "title_zh": (p.metadata_json or {}).get("title_zh", ""),
                 "abstract_zh": (p.metadata_json or {}).get("abstract_zh", ""),
                 "topics": topic_map.get(str(p.id), []),
+                "tags": tag_map.get(str(p.id), []),
             }
             for p in papers
         ]

apps/api/main.py

@@ -158,6 +158,7 @@ async def app_error_handler(_request: Request, exc: AppError):
     papers,
     pipelines,
     system,
+    tags,
     topics,
     writing,
 )
@@ -168,6 +169,7 @@ async def app_error_handler(_request: Request, exc: AppError):
 app.include_router(system.router)
 app.include_router(papers.router)
 app.include_router(topics.router)
+app.include_router(tags.router)
 app.include_router(cs_feeds.router)
 app.include_router(graph.router)
 app.include_router(agent.router)

apps/api/routers/papers.py

@@ -56,6 +56,7 @@ def latest(
     sort_by: str = Query(default="created_at"),
     sort_order: str = Query(default="desc"),
     category: str | None = Query(default=None),
+    tag_ids: list[str] | None = Query(default=None),
 ) -> dict:
     with session_scope() as session:
         repo = PaperRepository(session)
@@ -72,6 +73,7 @@ def latest(
             else "created_at",
             sort_order=sort_order if sort_order in ("asc", "desc") else "desc",
             category=category,
+            tag_ids=tag_ids,
         )
         resp = paper_list_response(papers, repo)
         resp["total"] = total
@@ -218,6 +220,7 @@ def paper_detail(paper_id: UUID) -> dict:
         except ValueError as exc:
             raise HTTPException(status_code=404, detail=str(exc)) from exc
         topic_map = repo.get_topic_names_for_papers([str(p.id)])
+        tag_map = repo.get_tags_for_papers([str(p.id)])
         # 查询已有分析报告
         from sqlalchemy import select as _sel
 
@@ -253,6 +256,7 @@ def paper_detail(paper_id: UUID) -> dict:
             "title_zh": (p.metadata_json or {}).get("title_zh", ""),
             "abstract_zh": (p.metadata_json or {}).get("abstract_zh", ""),
             "topics": topic_map.get(str(p.id), []),
+            "tags": tag_map.get(str(p.id), []),
             "metadata": p.metadata_json,
             "has_embedding": p.embedding is not None,
             "skim_report": skim_data,

apps/api/routers/tags.py

@@ -0,0 +1,188 @@
+"""标签管理路由
+@author Color2333
+"""
+
+from uuid import UUID
+
+from fastapi import APIRouter, HTTPException, Query
+
+from packages.storage.db import session_scope
+from packages.storage.repositories import PaperRepository, TagRepository
+
+router = APIRouter()
+
+
+@router.get("/tags")
+def list_tags() -> dict:
+    """获取所有标签列表"""
+    with session_scope() as session:
+        repo = TagRepository(session)
+        tags = repo.list_all()
+        return {
+            "items": [
+                {
+                    "id": tag.id,
+                    "name": tag.name,
+                    "color": tag.color,
+                    "paper_count": getattr(tag, "paper_count", 0),
+                    "created_at": tag.created_at.isoformat() if tag.created_at else None,
+                    "updated_at": tag.updated_at.isoformat() if tag.updated_at else None,
+                }
+                for tag in tags
+            ]
+        }
+
+
+@router.post("/tags")
+def create_tag(name: str, color: str = Query(default="#3b82f6")) -> dict:
+    """创建新标签"""
+    if not name or not name.strip():
+        raise HTTPException(status_code=400, detail="标签名称不能为空")
+    with session_scope() as session:
+        repo = TagRepository(session)
+        try:
+            tag = repo.create(name.strip(), color)
+            return {
+                "id": tag.id,
+                "name": tag.name,
+                "color": tag.color,
+                "paper_count": 0,
+            }
+        except ValueError as e:
+            raise HTTPException(status_code=400, detail=str(e)) from e
+
+
+@router.patch("/tags/{tag_id}")
+def update_tag(
+    tag_id: UUID,
+    name: str | None = Query(default=None),
+    color: str | None = Query(default=None),
+) -> dict:
+    """更新标签"""
+    with session_scope() as session:
+        repo = TagRepository(session)
+        try:
+            tag = repo.update(str(tag_id), name=name, color=color)
+            paper_count = repo.get_paper_count(str(tag_id))
+            return {
+                "id": tag.id,
+                "name": tag.name,
+                "color": tag.color,
+                "paper_count": paper_count,
+            }
+        except ValueError as e:
+            raise HTTPException(status_code=404, detail=str(e)) from e
+
+
+@router.delete("/tags/{tag_id}")
+def delete_tag(tag_id: UUID) -> dict:
+    """删除标签"""
+    with session_scope() as session:
+        repo = TagRepository(session)
+        tag = repo.get_by_id(str(tag_id))
+        if tag is None:
+            raise HTTPException(status_code=404, detail="标签不存在")
+        repo.delete(str(tag_id))
+        return {"deleted": str(tag_id), "name": tag.name}
+
+
+@router.get("/papers/{paper_id}/tags")
+def get_paper_tags(paper_id: UUID) -> dict:
+    """获取论文的标签"""
+    with session_scope() as session:
+        paper_repo = PaperRepository(session)
+        try:
+            paper_repo.get_by_id(paper_id)
+        except ValueError as e:
+            raise HTTPException(status_code=404, detail=str(e)) from e
+
+        tags_map = paper_repo.get_tags_for_papers([str(paper_id)])
+        return {"items": tags_map.get(str(paper_id), [])}
+
+
+@router.post("/papers/{paper_id}/tags")
+def add_paper_tag(paper_id: UUID, tag_id: UUID) -> dict:
+    """为论文添加标签"""
+    with session_scope() as session:
+        paper_repo = PaperRepository(session)
+        tag_repo = TagRepository(session)
+
+        try:
+            paper_repo.get_by_id(paper_id)
+        except ValueError as e:
+            raise HTTPException(status_code=404, detail="论文不存在") from e
+
+        tag = tag_repo.get_by_id(str(tag_id))
+        if tag is None:
+            raise HTTPException(status_code=404, detail="标签不存在")
+
+        paper_repo.link_to_tag(str(paper_id), str(tag_id))
+        session.commit()
+
+        return {
+            "paper_id": str(paper_id),
+            "tag": {
+                "id": tag.id,
+                "name": tag.name,
+                "color": tag.color,
+            },
+        }
+
+
+@router.delete("/papers/{paper_id}/tags/{tag_id}")
+def remove_paper_tag(paper_id: UUID, tag_id: UUID) -> dict:
+    """移除论文的标签"""
+    with session_scope() as session:
+        paper_repo = PaperRepository(session)
+        tag_repo = TagRepository(session)
+
+        try:
+            paper_repo.get_by_id(paper_id)
+        except ValueError as e:
+            raise HTTPException(status_code=404, detail="论文不存在") from e
+
+        tag = tag_repo.get_by_id(str(tag_id))
+        if tag is None:
+            raise HTTPException(status_code=404, detail="标签不存在")
+
+        paper_repo.unlink_from_tag(str(paper_id), str(tag_id))
+        session.commit()
+
+        return {
+            "paper_id": str(paper_id),
+            "tag_id": str(tag_id),
+            "removed": True,
+        }
+
+
+@router.post("/papers/{paper_id}/tags/batch")
+def batch_update_paper_tags(paper_id: UUID, tag_ids: list[UUID]) -> dict:
+    """批量更新论文的标签（替换所有标签）"""
+    with session_scope() as session:
+        paper_repo = PaperRepository(session)
+        tag_repo = TagRepository(session)
+
+        try:
+            paper_repo.get_by_id(paper_id)
+        except ValueError as e:
+            raise HTTPException(status_code=404, detail="论文不存在") from e
+
+        current_tags = paper_repo.get_tags_for_papers([str(paper_id)]).get(str(paper_id), [])
+        current_tag_ids = {t["id"] for t in current_tags}
+        new_tag_ids = {str(tid) for tid in tag_ids}
+
+        for tid in current_tag_ids - new_tag_ids:
+            paper_repo.unlink_from_tag(str(paper_id), tid)
+
+        for tid in new_tag_ids - current_tag_ids:
+            tag = tag_repo.get_by_id(tid)
+            if tag:
+                paper_repo.link_to_tag(str(paper_id), tid)
+
+        session.commit()
+
+        updated_tags = paper_repo.get_tags_for_papers([str(paper_id)]).get(str(paper_id), [])
+        return {
+            "paper_id": str(paper_id),
+            "items": updated_tags,
+        }