feat: handle streaming for generted clients

enerj · enerj · commit ddde6f131488 · 2025-03-04T13:04:42.000+01:00
diff --git a/pkg/codegen/templates/client-with-responses.tmpl b/pkg/codegen/templates/client-with-responses.tmpl
@@ -119,6 +119,12 @@ func (c *ClientWithResponses) {{$opid}}{{.Suffix}}WithResponse(ctx context.Conte
 
 // Parse{{genResponseTypeName $opid | ucFirst}} parses an HTTP response from a {{$opid}}WithResponse call
 func Parse{{genResponseTypeName $opid | ucFirst}}(rsp *http.Response) (*{{genResponseTypeName $opid}}, error) {
+    if strings.Contains(rsp.Header.Get("Content-Type"), "text/event-stream") {
+        return &{{genResponseTypeName $opid}}{
+            HTTPResponse: rsp,
+        }, nil
+    }
+
     bodyBytes, err := io.ReadAll(rsp.Body)
     defer func() { _ = rsp.Body.Close() }()
     if err != nil {
