feat(models): add kind and models_meta snapshot

This commit is contained in:
zenfun
2025-12-17 23:15:12 +08:00
parent 2b0ed3d3d5
commit 96e1fe41a5
6 changed files with 272 additions and 13 deletions

View File

@@ -294,8 +294,25 @@ func (h *Handler) CreateModel(c *gin.Context) {
return
}
name := strings.TrimSpace(req.Name)
if name == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "name required"})
return
}
kind := strings.ToLower(strings.TrimSpace(req.Kind))
if kind == "" {
kind = "chat"
}
switch kind {
case "chat", "embedding", "rerank", "other":
default:
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid kind"})
return
}
modelReq := model.Model{
Name: req.Name,
Name: name,
Kind: kind,
ContextWindow: req.ContextWindow,
CostPerToken: req.CostPerToken,
SupportsVision: req.SupportsVision,
@@ -370,7 +387,27 @@ func (h *Handler) UpdateModel(c *gin.Context) {
return
}
existing.Name = req.Name
name := strings.TrimSpace(req.Name)
if name == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "name required"})
return
}
kind := strings.ToLower(strings.TrimSpace(req.Kind))
if kind == "" {
kind = strings.ToLower(strings.TrimSpace(existing.Kind))
}
if kind == "" {
kind = "chat"
}
switch kind {
case "chat", "embedding", "rerank", "other":
default:
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid kind"})
return
}
existing.Name = name
existing.Kind = kind
existing.ContextWindow = req.ContextWindow
existing.CostPerToken = req.CostPerToken
existing.SupportsVision = req.SupportsVision

View File

@@ -0,0 +1,105 @@
package api
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
"net/http/httptest"
"testing"
"github.com/alicebob/miniredis/v2"
"github.com/ez-api/ez-api/internal/model"
"github.com/ez-api/ez-api/internal/service"
"github.com/gin-gonic/gin"
"github.com/redis/go-redis/v9"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
func newTestHandlerWithRedis(t *testing.T) (*Handler, *gorm.DB, *miniredis.Miniredis) {
t.Helper()
gin.SetMode(gin.TestMode)
dsn := fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())
db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
if err != nil {
t.Fatalf("open sqlite: %v", err)
}
if err := db.AutoMigrate(&model.Provider{}, &model.Binding{}, &model.Model{}); err != nil {
t.Fatalf("migrate: %v", err)
}
mr := miniredis.RunT(t)
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
sync := service.NewSyncService(rdb)
return NewHandler(db, sync, nil), db, mr
}
func TestCreateModel_DefaultsKindChat_AndWritesModelsMeta(t *testing.T) {
h, _, mr := newTestHandlerWithRedis(t)
r := gin.New()
r.POST("/admin/models", h.CreateModel)
reqBody := map[string]any{
"name": "ns.m",
}
b, _ := json.Marshal(reqBody)
req := httptest.NewRequest(http.MethodPost, "/admin/models", bytes.NewReader(b))
req.Header.Set("Content-Type", "application/json")
rr := httptest.NewRecorder()
r.ServeHTTP(rr, req)
if rr.Code != http.StatusCreated {
t.Fatalf("expected 201, got %d body=%s", rr.Code, rr.Body.String())
}
raw := mr.HGet("meta:models", "ns.m")
if raw == "" {
t.Fatalf("expected meta:models[ns.m] to be written")
}
var snap map[string]any
if err := json.Unmarshal([]byte(raw), &snap); err != nil {
t.Fatalf("unmarshal snapshot: %v raw=%s", err, raw)
}
if snap["kind"] != "chat" {
t.Fatalf("expected kind=chat, got %v raw=%s", snap["kind"], raw)
}
if v := mr.HGet("meta:models_meta", "version"); v == "" {
t.Fatalf("expected meta:models_meta.version")
}
if v := mr.HGet("meta:models_meta", "updated_at"); v == "" {
t.Fatalf("expected meta:models_meta.updated_at")
}
if v := mr.HGet("meta:models_meta", "source"); v == "" {
t.Fatalf("expected meta:models_meta.source")
}
if v := mr.HGet("meta:models_meta", "checksum"); v == "" {
t.Fatalf("expected meta:models_meta.checksum")
}
}
func TestCreateModel_InvalidKind_Returns400(t *testing.T) {
h, _, _ := newTestHandlerWithRedis(t)
r := gin.New()
r.POST("/admin/models", h.CreateModel)
reqBody := map[string]any{
"name": "ns.m2",
"kind": "bad",
}
b, _ := json.Marshal(reqBody)
req := httptest.NewRequest(http.MethodPost, "/admin/models", bytes.NewReader(b))
req.Header.Set("Content-Type", "application/json")
rr := httptest.NewRecorder()
r.ServeHTTP(rr, req)
if rr.Code != http.StatusBadRequest {
t.Fatalf("expected 400, got %d body=%s", rr.Code, rr.Body.String())
}
}

View File

@@ -28,7 +28,7 @@ func newTestHandler(t *testing.T) (*Handler, *gorm.DB) {
if err != nil {
t.Fatalf("open sqlite: %v", err)
}
if err := db.AutoMigrate(&model.Provider{}, &model.Binding{}); err != nil {
if err := db.AutoMigrate(&model.Provider{}, &model.Binding{}, &model.Model{}); err != nil {
t.Fatalf("migrate: %v", err)
}