Merge pull request #312 from Gregory-Pereira/228-reasonable-http-time…

…outs

implementing reasonable timeouts in the http clients

worker/cmd/generate.go

@@ -796,6 +796,9 @@ func (w *Worker) fetchModelName(fullName bool) (string, error) {
 	endpoint += "models"
 
 	http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
+	http.DefaultTransport.(*http.Transport).TLSHandshakeTimeout = 10 * time.Second
+	http.DefaultTransport.(*http.Transport).ResponseHeaderTimeout = 10 * time.Second
+	http.DefaultTransport.(*http.Transport).ExpectContinueTimeout = 1 * time.Second
 
 	req, err := http.NewRequestWithContext(w.ctx, "GET", endpoint, nil)
 	if err != nil {
@@ -992,7 +995,12 @@ func (w *Worker) createTLSHttpClient() (*http.Client, error) {
 		InsecureSkipVerify: true,
 	}
 	httpClient := &http.Client{
-		Transport: &http.Transport{TLSClientConfig: tlsConfig},
+		Transport: &http.Transport{
+			TLSClientConfig:       tlsConfig,
+			TLSHandshakeTimeout:   10 * time.Second,
+			ResponseHeaderTimeout: 10 * time.Second,
+			ExpectContinueTimeout: 1 * time.Second,
+		},
 	}
 	return httpClient, nil
 }