Jelajahi Sumber

Merge pull request #531 from jmorganca/mxyng/content-length

set request.ContentLength
Michael Yang 1 tahun lalu
induk
melakukan
ccc3e9ac6d
1 mengubah file dengan 9 tambahan dan 0 penghapusan
  1. 9 0
      server/images.go

+ 9 - 0
server/images.go

@@ -1442,6 +1442,15 @@ func makeRequest(ctx context.Context, method string, requestURL *url.URL, header
 
 	req.Header.Set("User-Agent", fmt.Sprintf("ollama/%s (%s %s) Go/%s", version.Version, runtime.GOARCH, runtime.GOOS, runtime.Version()))
 
+	if s := req.Header.Get("Content-Length"); s != "" {
+		contentLength, err := strconv.ParseInt(s, 10, 64)
+		if err != nil {
+			return nil, err
+		}
+
+		req.ContentLength = contentLength
+	}
+
 	client := &http.Client{
 		CheckRedirect: func(req *http.Request, via []*http.Request) error {
 			if len(via) >= 10 {