feat: AI Worker 平台 MVP v1.0.0 - 多租户/项目管理/AI Worker/任务编排/HITL审核/成本治理

backend/app/services/worker_service.py

@@ -0,0 +1,74 @@
+"""AI Worker service: CRUD + status management."""
+from sqlalchemy.orm import Session
+from typing import List, Optional
+from app.models.worker import AIWorker
+from app.schemas.worker import WorkerCreate, WorkerUpdate
+
+
+class WorkerService:
+    @staticmethod
+    def list_workers(
+        db: Session, tenant_id: int, skip: int = 0, limit: int = 100
+    ) -> List[AIWorker]:
+        return (
+            db.query(AIWorker)
+            .filter(AIWorker.tenant_id == tenant_id)
+            .order_by(AIWorker.created_at.desc())
+            .offset(skip)
+            .limit(limit)
+            .all()
+        )
+
+    @staticmethod
+    def get_worker(db: Session, tenant_id: int, worker_id: int) -> Optional[AIWorker]:
+        return (
+            db.query(AIWorker)
+            .filter(AIWorker.id == worker_id, AIWorker.tenant_id == tenant_id)
+            .first()
+        )
+
+    @staticmethod
+    def create_worker(db: Session, tenant_id: int, req: WorkerCreate) -> AIWorker:
+        worker = AIWorker(
+            tenant_id=tenant_id,
+            name=req.name,
+            description=req.description,
+            provider=req.provider,
+            model_name=req.model_name,
+            api_key=req.api_key,
+            base_url=req.base_url,
+            temperature=req.temperature,
+            system_prompt=req.system_prompt,
+            tools=req.tools,
+            max_cost_per_task_cents=req.max_cost_per_task_cents,
+            max_cost_per_month_cents=req.max_cost_per_month_cents,
+            available_hours=req.available_hours,
+            max_concurrent_tasks=req.max_concurrent_tasks,
+            status="active",
+        )
+        db.add(worker)
+        db.commit()
+        db.refresh(worker)
+        return worker
+
+    @staticmethod
+    def update_worker(
+        db: Session, tenant_id: int, worker_id: int, req: WorkerUpdate
+    ) -> AIWorker:
+        worker = WorkerService.get_worker(db, tenant_id, worker_id)
+        if not worker:
+            raise ValueError(f"Worker #{worker_id} not found")
+        for field, value in req.model_dump(exclude_unset=True).items():
+            setattr(worker, field, value)
+        db.commit()
+        db.refresh(worker)
+        return worker
+
+    @staticmethod
+    def delete_worker(db: Session, tenant_id: int, worker_id: int) -> bool:
+        worker = WorkerService.get_worker(db, tenant_id, worker_id)
+        if not worker:
+            return False
+        db.delete(worker)
+        db.commit()
+        return True