Upload routes.py

app/api/routes.py

@@ -6,7 +6,8 @@ from fastapi import APIRouter, Depends, HTTPException, Query
 
 from app.core.config import Settings, get_settings
 from app.core.security import CurrentUser, get_current_user
-from app.models.catalog import NodeDefinition, ProjectSummary, TemplateSummary
+from app.core.errors import ServiceConfigurationError
+from app.models.catalog import DashboardSummary, NodeDefinition, ProjectSummary, TemplateSummary
 from app.models.workflow import (
     ChatRequest,
     ChatResponse,
@@ -62,7 +63,7 @@ from app.models.workflow import (
     WebhookInspectResponse,
 )
 from app.services.adapters import adapters
-from app.services.catalog import NODE_CATALOG, TEMPLATES
+from app.services.catalog import NODE_CATALOG
 from app.services.chat import chat_service
 from app.services.collaboration import CollaborationRepository
 from app.services.deployment import deploy_to_n8n
@@ -104,8 +105,13 @@ async def generate_workflow(
     user: User,
     settings: AppSettings,
 ) -> GenerateWorkflowResponse:
-    provider = providers.get(request.provider or settings.ai_provider)
-    return await provider.generate(request.prompt)
+    try:
+        provider = providers.get(request.provider or settings.ai_provider)
+        return await provider.generate(request.prompt, request.model)
+    except ValueError as exc:
+        raise HTTPException(status_code=422, detail=str(exc)) from exc
+    except (RuntimeError, httpx.HTTPError) as exc:
+        raise HTTPException(status_code=502, detail=f"Workflow generation failed: {exc}") from exc
 
 
 @router.post("/validate", response_model=ValidationResult)
@@ -291,19 +297,35 @@ async def export_workflow(request: ExportRequest, user: User):
 @router.get("/templates")
 async def list_templates(
     user: User,
+    settings: AppSettings,
     q: str = Query(default="", max_length=200),
     category: str | None = Query(default=None, max_length=80),
     limit: int = Query(default=24, ge=1, le=100),
     offset: int = Query(default=0, ge=0),
 ) -> dict[str, list[TemplateSummary] | int]:
-    query = q.lower().strip()
-    items = [
-        template
-        for template in TEMPLATES
-        if (not query or query in f"{template.name} {template.description}".lower())
-        and (not category or template.category.lower() == category.lower())
-    ]
-    return {"items": items[offset : offset + limit], "total": len(items)}
+    try:
+        items, total = await WorkflowRepository(settings).templates(
+            user.id,
+            query=q.strip(),
+            category=category,
+            limit=limit,
+            offset=offset,
+        )
+        return {"items": items, "total": total}
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
+
+
+@router.get("/templates/{template_id}", response_model=WorkflowDocument)
+async def get_template(
+    template_id: str, user: User, settings: AppSettings
+) -> WorkflowDocument:
+    try:
+        return await WorkflowRepository(settings).template(template_id, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
+    except ValueError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
 
 
 @router.get("/nodes")
@@ -329,7 +351,30 @@ async def list_nodes(
 
 @router.get("/projects", response_model=list[ProjectSummary])
 async def list_projects(user: User, settings: AppSettings) -> list[ProjectSummary]:
-    return await WorkflowRepository(settings).projects(user.id)
+    try:
+        return await WorkflowRepository(settings).projects(user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
+
+
+@router.get("/dashboard", response_model=DashboardSummary)
+async def dashboard(user: User, settings: AppSettings) -> DashboardSummary:
+    try:
+        return await WorkflowRepository(settings).dashboard(user.id, user.email)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
+
+
+@router.get("/workflows/{workflow_id}", response_model=WorkflowDocument)
+async def get_workflow(
+    workflow_id: str, user: User, settings: AppSettings
+) -> WorkflowDocument:
+    try:
+        return await WorkflowRepository(settings).workflow(workflow_id, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
+    except ValueError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
 
 
 @router.post("/save", response_model=SaveResponse)
@@ -340,6 +385,8 @@ async def save_workflow(
 ) -> SaveResponse:
     try:
         return await WorkflowRepository(settings).save(request, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=403, detail=str(exc)) from exc
 
@@ -350,6 +397,8 @@ async def create_share(
 ) -> ShareResponse:
     try:
         return await CollaborationRepository(settings).create_share(request, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=403, detail=str(exc)) from exc
 
@@ -360,6 +409,8 @@ async def get_shared_workflow(token: str, settings: AppSettings) -> SharedWorkfl
         raise HTTPException(status_code=404, detail="Share link was not found")
     try:
         return await CollaborationRepository(settings).shared_workflow(token)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=404, detail=str(exc)) from exc
 
@@ -370,6 +421,8 @@ async def list_versions(
 ) -> list[VersionSummary]:
     try:
         return await CollaborationRepository(settings).versions(workflow_id, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=403, detail=str(exc)) from exc
 
@@ -380,6 +433,8 @@ async def restore_version(
 ) -> WorkflowDocument:
     try:
         return await CollaborationRepository(settings).restore(workflow_id, version, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=403, detail=str(exc)) from exc
 
@@ -390,6 +445,8 @@ async def list_comments(
 ) -> list[WorkflowComment]:
     try:
         return await CollaborationRepository(settings).comments(workflow_id, user.id)
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=403, detail=str(exc)) from exc
 
@@ -402,6 +459,8 @@ async def add_comment(
         return await CollaborationRepository(settings).add_comment(
             request.workflow_id, request.body, request.node_id, user.id
         )
+    except ServiceConfigurationError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
     except ValueError as exc:
         raise HTTPException(status_code=403, detail=str(exc)) from exc