feat(*): 生图与视频工作台 - 新增独立生成流程并整合平台配置
This commit is contained in:
ddcat666
@@ -35,6 +35,7 @@ func RegisterCanvasAPI(api *gin.RouterGroup, svc *service.Service) {
|
||||
RegisterSystemProxyRoutes(api, svc)
|
||||
RegisterCustomRelayRoutes(api, svc)
|
||||
RegisterTaskRoutes(api, svc)
|
||||
RegisterWorkbenchRoutes(api, svc)
|
||||
RegisterRunningHubRoutes(api, svc)
|
||||
RegisterSkillRoutes(api, svc)
|
||||
RegisterToolRoutes(api, svc)
|
||||
|
||||
@@ -18,6 +18,9 @@ func TestRegisterCanvasAPIExposesOpenAPIAndProjects(t *testing.T) {
|
||||
|
||||
wanted := map[string]bool{
|
||||
"GET /api/openapi.yaml": false,
|
||||
"GET /api/workbench/tools": false,
|
||||
"GET /api/workbench/tasks": false,
|
||||
"POST /api/workbench/tasks": false,
|
||||
"GET /api/projects": false,
|
||||
"POST /api/tasks": false,
|
||||
"POST /api/tasks/:id/recover-media": false,
|
||||
|
||||
@@ -26,6 +26,7 @@ tags:
|
||||
- name: agent
|
||||
- name: admin
|
||||
- name: skills
|
||||
- name: workbench
|
||||
paths:
|
||||
/admin/login-logs:
|
||||
get:
|
||||
@@ -1432,6 +1433,121 @@ paths:
|
||||
$ref: "#/components/responses/Error"
|
||||
"404":
|
||||
$ref: "#/components/responses/Error"
|
||||
/workbench/tools:
|
||||
get:
|
||||
tags: [workbench]
|
||||
summary: 获取当前用户可用的工作台工具
|
||||
parameters:
|
||||
- in: query
|
||||
name: kind
|
||||
required: true
|
||||
schema: { type: string, enum: [image, video] }
|
||||
responses:
|
||||
"200":
|
||||
$ref: "#/components/responses/Envelope"
|
||||
"401":
|
||||
$ref: "#/components/responses/Error"
|
||||
"403":
|
||||
$ref: "#/components/responses/Error"
|
||||
/workbench/tasks:
|
||||
get:
|
||||
tags: [workbench]
|
||||
summary: 查询当前用户的工作台任务
|
||||
parameters:
|
||||
- in: query
|
||||
name: kind
|
||||
required: true
|
||||
schema: { type: string, enum: [image, video] }
|
||||
- in: query
|
||||
name: activeOnly
|
||||
schema: { type: boolean, default: false }
|
||||
- in: query
|
||||
name: pageSize
|
||||
schema: { type: integer, minimum: 1, maximum: 100, default: 50 }
|
||||
responses:
|
||||
"200":
|
||||
$ref: "#/components/responses/Envelope"
|
||||
"401":
|
||||
$ref: "#/components/responses/Error"
|
||||
"403":
|
||||
$ref: "#/components/responses/Error"
|
||||
post:
|
||||
tags: [workbench]
|
||||
summary: 创建工作台生成任务
|
||||
description: 仅接受模型选择意图、资源 ID、提示词和允许的生成参数；toolId 可选，服务端校验开关、所选工具、资源、模型能力和账单，不接受客户端 providerConfig、密钥或上游 URL。
|
||||
requestBody:
|
||||
required: true
|
||||
content:
|
||||
application/json:
|
||||
schema:
|
||||
type: object
|
||||
additionalProperties: false
|
||||
required: [kind, prompt, model, modelChoice, parameters, references]
|
||||
properties:
|
||||
kind: { type: string, enum: [image, video] }
|
||||
toolId: { type: string }
|
||||
presetId: { type: string }
|
||||
prompt: { type: string, minLength: 1, maxLength: 32000 }
|
||||
model: { type: string }
|
||||
logicalModelId: { type: string }
|
||||
modelChoice:
|
||||
type: object
|
||||
additionalProperties: false
|
||||
properties:
|
||||
logicalModelId: { type: string }
|
||||
modelName: { type: string }
|
||||
workbench:
|
||||
type: object
|
||||
additionalProperties: false
|
||||
properties:
|
||||
kind: { type: string, enum: [image, video] }
|
||||
toolId: { type: string }
|
||||
presetId: { type: string }
|
||||
parameters:
|
||||
type: object
|
||||
additionalProperties: false
|
||||
properties:
|
||||
size: { type: string }
|
||||
quality: { type: string }
|
||||
count: { type: integer, minimum: 1 }
|
||||
videoSeconds: { type: integer, minimum: 1 }
|
||||
ratio: { type: string }
|
||||
resolution: { type: string }
|
||||
generateAudio: { type: boolean }
|
||||
watermark: { type: boolean }
|
||||
references:
|
||||
type: array
|
||||
maxItems: 20
|
||||
items:
|
||||
type: object
|
||||
additionalProperties: false
|
||||
required: [resourceId, kind, role]
|
||||
properties:
|
||||
resourceId: { type: string }
|
||||
kind: { type: string, enum: [image, video, audio] }
|
||||
role: { type: string, enum: [reference, first_frame, last_frame] }
|
||||
responses:
|
||||
"200":
|
||||
$ref: "#/components/responses/Envelope"
|
||||
"400":
|
||||
$ref: "#/components/responses/Error"
|
||||
"401":
|
||||
$ref: "#/components/responses/Error"
|
||||
"403":
|
||||
$ref: "#/components/responses/Error"
|
||||
/admin/workbench/tools:
|
||||
get:
|
||||
tags: [workbench, admin]
|
||||
summary: 管理工作台工具目录
|
||||
parameters:
|
||||
- in: query
|
||||
name: kind
|
||||
schema: { type: string, enum: [image, video] }
|
||||
responses:
|
||||
"200":
|
||||
$ref: "#/components/responses/Envelope"
|
||||
"403":
|
||||
$ref: "#/components/responses/Error"
|
||||
/tasks:
|
||||
post:
|
||||
tags: [tasks]
|
||||
|
||||
204
backend/internal/handler/workbench.go
Normal file
204
backend/internal/handler/workbench.go
Normal file
@@ -0,0 +1,204 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"io"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"yingce/backend/internal/service"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func RegisterWorkbenchRoutes(r *gin.RouterGroup, svc *service.Service) {
|
||||
r.GET("/workbench/tools", func(c *gin.Context) {
|
||||
if _, err := currentUser(c, svc); err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
kind := c.Query("kind")
|
||||
tools, err := svc.PublicWorkbenchTools(kind)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"tools": tools})
|
||||
})
|
||||
r.GET("/workbench/tasks", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
limit, err := parsePositiveQueryInt(c.Query("pageSize"), 50)
|
||||
if err != nil {
|
||||
fail(c, http.StatusBadRequest, err)
|
||||
return
|
||||
}
|
||||
tasks, err := svc.WorkbenchTasks(user.ID, c.Query("kind"), c.Query("activeOnly") == "true", limit)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, tasks)
|
||||
})
|
||||
r.POST("/workbench/tasks", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
policy, available := loadRuntimePolicy(c, svc)
|
||||
if !available || !enforceRateLimit(c, "tasks:"+user.ID, policy.Request.TaskCreatePerMinute, time.Minute) {
|
||||
return
|
||||
}
|
||||
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 16<<20)
|
||||
var req service.WorkbenchTaskInput
|
||||
if err := decodeWorkbenchJSON(c, &req); err != nil {
|
||||
fail(c, http.StatusBadRequest, err)
|
||||
return
|
||||
}
|
||||
task, err := svc.CreateWorkbenchTask(user.ID, req, TraceID(c), RequestID(c))
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, task)
|
||||
})
|
||||
admin := r.Group("/admin/workbench/tools")
|
||||
admin.GET("", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
items, err := svc.AdminWorkbenchTools(user, c.Query("kind"))
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"tools": items})
|
||||
})
|
||||
admin.POST("", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
var req service.WorkbenchToolInput
|
||||
if err := decodeWorkbenchJSON(c, &req); err != nil {
|
||||
fail(c, http.StatusBadRequest, err)
|
||||
return
|
||||
}
|
||||
item, err := svc.CreateWorkbenchTool(user, req)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"tool": item})
|
||||
})
|
||||
admin.PATCH("/:id", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
var req service.WorkbenchToolInput
|
||||
if err := decodeWorkbenchJSON(c, &req); err != nil {
|
||||
fail(c, http.StatusBadRequest, err)
|
||||
return
|
||||
}
|
||||
item, err := svc.UpdateWorkbenchTool(user, c.Param("id"), req)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"tool": item})
|
||||
})
|
||||
admin.POST("/:id/publish", func(c *gin.Context) { setWorkbenchToolState(c, svc, true) })
|
||||
admin.POST("/:id/unpublish", func(c *gin.Context) { setWorkbenchToolState(c, svc, false) })
|
||||
admin.POST("/:id/presets", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 1<<20)
|
||||
var req service.WorkbenchPresetInput
|
||||
if err := decodeWorkbenchJSON(c, &req); err != nil {
|
||||
fail(c, http.StatusBadRequest, err)
|
||||
return
|
||||
}
|
||||
if req.ToolID != c.Param("id") {
|
||||
failService(c, service.BadAuthRequest("预设所属工具与路径不匹配"))
|
||||
return
|
||||
}
|
||||
item, err := svc.CreateWorkbenchPreset(user, req)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"preset": item})
|
||||
})
|
||||
admin.PATCH("/:id/presets/:presetId", func(c *gin.Context) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
var req service.WorkbenchPresetInput
|
||||
if err := decodeWorkbenchJSON(c, &req); err != nil {
|
||||
fail(c, http.StatusBadRequest, err)
|
||||
return
|
||||
}
|
||||
item, err := svc.UpdateWorkbenchPreset(user, c.Param("presetId"), req)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"preset": item})
|
||||
})
|
||||
admin.POST("/:id/presets/:presetId/publish", func(c *gin.Context) { setWorkbenchPresetState(c, svc, true) })
|
||||
admin.POST("/:id/presets/:presetId/unpublish", func(c *gin.Context) { setWorkbenchPresetState(c, svc, false) })
|
||||
}
|
||||
|
||||
func setWorkbenchToolState(c *gin.Context, svc *service.Service, enabled bool) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
if err := svc.SetWorkbenchToolEnabled(user, c.Param("id"), enabled); err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"id": c.Param("id"), "enabled": enabled})
|
||||
}
|
||||
func setWorkbenchPresetState(c *gin.Context, svc *service.Service, enabled bool) {
|
||||
user, err := currentUser(c, svc)
|
||||
if err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
if err := svc.SetWorkbenchPresetEnabled(user, c.Param("presetId"), enabled); err != nil {
|
||||
failService(c, err)
|
||||
return
|
||||
}
|
||||
ok(c, gin.H{"id": c.Param("presetId"), "enabled": enabled})
|
||||
}
|
||||
|
||||
func decodeWorkbenchJSON(c *gin.Context, target any) error {
|
||||
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, 2<<20)
|
||||
decoder := json.NewDecoder(c.Request.Body)
|
||||
decoder.DisallowUnknownFields()
|
||||
if err := decoder.Decode(target); err != nil {
|
||||
return err
|
||||
}
|
||||
var trailing any
|
||||
if err := decoder.Decode(&trailing); err != io.EOF {
|
||||
return errors.New("请求必须只包含一个 JSON 对象")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user