(c *gin.Context, resp dto.ClaudeResponse)
| 30 | } |
| 31 | |
| 32 | func ClaudeData(c *gin.Context, resp dto.ClaudeResponse) error { |
| 33 | jsonData, err := json.Marshal(resp) |
| 34 | if err != nil { |
| 35 | common.SysError("error marshalling stream response: " + err.Error()) |
| 36 | } else { |
| 37 | c.Render(-1, common.CustomEvent{Data: fmt.Sprintf("event: %s\n", resp.Type)}) |
| 38 | c.Render(-1, common.CustomEvent{Data: "data: " + string(jsonData)}) |
| 39 | } |
| 40 | if flusher, ok := c.Writer.(http.Flusher); ok { |
| 41 | flusher.Flush() |
| 42 | } else { |
| 43 | return errors.New("streaming error: flusher not found") |
| 44 | } |
| 45 | return nil |
| 46 | } |
| 47 | |
| 48 | func ClaudeChunkData(c *gin.Context, resp dto.ClaudeResponse, data string) { |
| 49 | c.Render(-1, common.CustomEvent{Data: fmt.Sprintf("event: %s\n", resp.Type)}) |
no test coverage detected