diff --git a/internal/entity/models/novita.go b/internal/entity/models/novita.go index 980a61949b..51f006443f 100644 --- a/internal/entity/models/novita.go +++ b/internal/entity/models/novita.go @@ -57,7 +57,15 @@ type NovitaModel struct { // clone http.DefaultTransport, override the connection-pool fields, // no client-level Timeout so SSE streams are not capped. func NewNovitaModel(baseURL map[string]string, urlSuffix URLSuffix) *NovitaModel { - transport := http.DefaultTransport.(*http.Transport).Clone() + defaultTransport, ok := http.DefaultTransport.(*http.Transport) + var transport *http.Transport + if ok { + transport = defaultTransport.Clone() + } else { + transport = &http.Transport{ + Proxy: http.ProxyFromEnvironment, + } + } transport.MaxIdleConns = 100 transport.MaxIdleConnsPerHost = 10 transport.IdleConnTimeout = 90 * time.Second diff --git a/internal/entity/models/novita_test.go b/internal/entity/models/novita_test.go index 177085cad9..43d12b0c18 100644 --- a/internal/entity/models/novita_test.go +++ b/internal/entity/models/novita_test.go @@ -9,6 +9,12 @@ import ( "testing" ) +type roundTripperFunc func(*http.Request) (*http.Response, error) + +func (f roundTripperFunc) RoundTrip(r *http.Request) (*http.Response, error) { + return f(r) +} + func newNovitaServer(t *testing.T, expectedPath string, handler func(t *testing.T, body map[string]interface{}, w http.ResponseWriter)) *httptest.Server { t.Helper() return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { @@ -58,6 +64,20 @@ func newNovitaForTest(baseURL string) *NovitaModel { ) } +func TestNovitaNewModelWithCustomDefaultTransport(t *testing.T) { + original := http.DefaultTransport + http.DefaultTransport = roundTripperFunc(func(*http.Request) (*http.Response, error) { + return nil, nil + }) + t.Cleanup(func() { + http.DefaultTransport = original + }) + + if model := NewNovitaModel(map[string]string{"default": "http://unused"}, URLSuffix{}); model == nil { + t.Fatal("NewNovitaModel returned nil") + } +} + // newNovitaSSEServer asserts the SSE-chat wire contract (POST, path, // Authorization, Content-Type) the same way newNovitaServer does for // the JSON-chat path, then writes the supplied SSE payload. Closes