diff --git a/usecase/modelkit.go b/usecase/modelkit.go index 59d004e..c00c49c 100644 --- a/usecase/modelkit.go +++ b/usecase/modelkit.go @@ -69,6 +69,7 @@ func ModelList(ctx context.Context, req *domain.ModelListReq) (*domain.ModelList MaxIdleConnsPerHost: 100, MaxConnsPerHost: 100, IdleConnTimeout: time.Second * 30, + Proxy: http.ProxyFromEnvironment, }, } provider := consts.ParseModelProvider(req.Provider) @@ -198,7 +199,9 @@ func CheckModel(ctx context.Context, req *domain.CheckModelReq) (*domain.CheckMo } request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", req.APIKey)) request.Header.Set("Content-Type", "application/json") - resp, err := http.DefaultClient.Do(request) + client := http.DefaultClient + client.Transport = http.DefaultTransport + resp, err := client.Do(request) if err != nil { checkResp.Error = fmt.Sprintf("send request failed: %s", err.Error()) return checkResp, nil