renjue
321683f863
CI / docker (push) Successful in 3m38s
Add model input defaults (including prompt), workflow model editing, ingress API docs, and fix img2img image upload filenames for signed URLs. Co-authored-by: Cursor <cursoragent@cursor.com>
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package v1handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/rose_cat707/luminary/internal/gateway"
|
|
"github.com/rose_cat707/luminary/internal/model"
|
|
"github.com/rose_cat707/luminary/internal/prediction"
|
|
)
|
|
|
|
type ModelsHandler struct {
|
|
gateway *gateway.Service
|
|
predictions *prediction.Service
|
|
baseURL func() string
|
|
}
|
|
|
|
func NewModelsHandler(gw *gateway.Service, pred *prediction.Service, baseURL func() string) *ModelsHandler {
|
|
return &ModelsHandler{gateway: gw, predictions: pred, baseURL: baseURL}
|
|
}
|
|
|
|
func (h *ModelsHandler) List(c *gin.Context) {
|
|
ingress, ok := ingressKey(c)
|
|
if !ok {
|
|
c.JSON(http.StatusUnauthorized, gin.H{"error": "unauthorized"})
|
|
return
|
|
}
|
|
switch ingress.EffectiveProtocol() {
|
|
case model.IngressProtocolImage:
|
|
results, err := h.predictions.ListPublicModels(c.Request.Context(), ingress, h.baseURL())
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"detail": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"previous": nil,
|
|
"next": nil,
|
|
"results": results,
|
|
})
|
|
default:
|
|
models, err := h.gateway.ListModels(c.Request.Context(), ingress)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"object": "list", "data": models})
|
|
}
|
|
}
|
|
|
|
func ingressKey(c *gin.Context) (*model.IngressKey, bool) {
|
|
v, ok := c.Get("ingress_key")
|
|
if !ok {
|
|
return nil, false
|
|
}
|
|
k, ok := v.(*model.IngressKey)
|
|
return k, ok
|
|
}
|