From 0d01d926812f63c23b5f5e47ebca273c773b648a Mon Sep 17 00:00:00 2001 From: huangzhuang_3rd Date: Wed, 12 Aug 2026 13:30:28 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20AI=20Worker=20=E5=B9=B3=E5=8F=B0=20MVP?= =?UTF-8?q?=20v1.0.0=20-=20=E5=A4=9A=E7=A7=9F=E6=88=B7/=E9=A1=B9=E7=9B=AE?= =?UTF-8?q?=E7=AE=A1=E7=90=86/AI=20Worker/=E4=BB=BB=E5=8A=A1=E7=BC=96?= =?UTF-8?q?=E6=8E=92/HITL=E5=AE=A1=E6=A0=B8/=E6=88=90=E6=9C=AC=E6=B2=BB?= =?UTF-8?q?=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/app/api/v1/artifacts.py | 66 +++++++++++++++++++++++++++++++++ 1 file changed, 66 insertions(+) create mode 100644 backend/app/api/v1/artifacts.py diff --git a/backend/app/api/v1/artifacts.py b/backend/app/api/v1/artifacts.py new file mode 100644 index 0000000..97cda7f --- /dev/null +++ b/backend/app/api/v1/artifacts.py @@ -0,0 +1,66 @@ +"""Artifact endpoints.""" +from fastapi import APIRouter, Depends, HTTPException +from sqlalchemy.orm import Session +from typing import List +from app.database import get_db +from app.schemas.artifact import ArtifactCreate, ArtifactResponse +from app.models.artifact import Artifact +from app.api.deps import get_current_user +from app.models.user import User + +router = APIRouter(prefix="/artifacts") + + +@router.get("", response_model=List[ArtifactResponse]) +def list_artifacts( + task_id: int = None, + project_id: int = None, + skip: int = 0, limit: int = 100, + user: User = Depends(get_current_user), + db: Session = Depends(get_db), +): + q = db.query(Artifact).filter(Artifact.tenant_id == user.tenant_id) + if task_id: + q = q.filter(Artifact.task_id == task_id) + if project_id: + q = q.filter(Artifact.project_id == project_id) + return q.order_by(Artifact.created_at.desc()).offset(skip).limit(limit).all() + + +@router.get("/{artifact_id}", response_model=ArtifactResponse) +def get_artifact( + artifact_id: int, + user: User = Depends(get_current_user), + db: Session = Depends(get_db), +): + a = db.query(Artifact).filter( + Artifact.id == artifact_id, + Artifact.tenant_id == user.tenant_id, + ).first() + if not a: + raise HTTPException(status_code=404, detail="Artifact not found") + return a + + +@router.post("", response_model=ArtifactResponse, status_code=201) +def create_artifact( + req: ArtifactCreate, + task_id: int, + project_id: int, + user: User = Depends(get_current_user), + db: Session = Depends(get_db), +): + artifact = Artifact( + tenant_id=user.tenant_id, + task_id=task_id, + project_id=project_id, + title=req.title, + content=req.content, + artifact_type=req.artifact_type, + sources=req.sources, + created_by_user_id=user.id, + ) + db.add(artifact) + db.commit() + db.refresh(artifact) + return artifact