X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/d8f3785bb69288663f4cc46bf4d609c6770452d5..6c7859cfea123479aba046614ee81028651ddaf5:/model.go

diff --git a/model.go b/model.go
index c443a12..78836fe 100644
--- a/model.go
+++ b/model.go
@@ -344,7 +344,12 @@ func (c *ModelClient) send(httpRequest *http.Request, record *ModelUsageRecord)
 		return ModelResponse{}, errors.Join(err, readErr)
 	}
 
-	return modelReadStream(httpResponse.Body)
+	response, err = modelReadStream(httpResponse.Body)
+	if err != nil {
+		return response, fmt.Errorf("read model response stream: %w", err)
+	}
+
+	return response, nil
 }
 
 // Write appends record to the file with one write.
@@ -463,8 +468,11 @@ func modelReadEvent(response *ModelResponse, data []byte) (bool, error) {
 		}
 
 		response.Text, err = modelOutputText(response.Output)
+		if err != nil {
+			return true, fmt.Errorf("read completed response text: %w", err)
+		}
 
-		return true, err
+		return true, nil
 	case "response.incomplete":
 		reason := "unknown"
 		if event.Response != nil && event.Response.IncompleteDetails != nil {