mirror of
https://github.com/CJackHwang/ds2api.git
synced 2026-05-15 21:55:09 +08:00
131ca7d398
- replace legacy DeepSeek ids with the new deepseek-v4 model family\n- move thinking control to request parameters and preserve assistant reasoning content\n- switch history split to IGNORE transcript injection and map upload auth failures to 401\n- update admin defaults, API docs, samples, and tests for the new model scheme
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package openai
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
func TestGetModelRouteDirectAndAlias(t *testing.T) {
|
|
h := &Handler{}
|
|
r := chi.NewRouter()
|
|
RegisterRoutes(r, h)
|
|
|
|
t.Run("direct", func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, "/v1/models/deepseek-v4-flash", nil)
|
|
rec := httptest.NewRecorder()
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
|
|
}
|
|
})
|
|
|
|
t.Run("direct_expert", func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, "/v1/models/deepseek-v4-pro", nil)
|
|
rec := httptest.NewRecorder()
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
|
|
}
|
|
})
|
|
|
|
t.Run("direct_vision", func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, "/v1/models/deepseek-v4-vision", nil)
|
|
rec := httptest.NewRecorder()
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
|
|
}
|
|
})
|
|
|
|
t.Run("alias", func(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, "/v1/models/gpt-4.1", nil)
|
|
rec := httptest.NewRecorder()
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("expected 200 for alias, got %d body=%s", rec.Code, rec.Body.String())
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestGetModelRouteNotFound(t *testing.T) {
|
|
h := &Handler{}
|
|
r := chi.NewRouter()
|
|
RegisterRoutes(r, h)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/v1/models/not-exists", nil)
|
|
rec := httptest.NewRecorder()
|
|
r.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusNotFound {
|
|
t.Fatalf("expected 404, got %d body=%s", rec.Code, rec.Body.String())
|
|
}
|
|
}
|