(event anthropic.MessageStreamEventUnion)
| 535 | } |
| 536 | |
| 537 | func (i *StreamingInterception) marshalEvent(event anthropic.MessageStreamEventUnion) ([]byte, error) { |
| 538 | sj, err := sjson.Set(event.RawJSON(), "message.id", i.ID().String()) |
| 539 | if err != nil { |
| 540 | return nil, xerrors.Errorf("marshal event id failed: %w", err) |
| 541 | } |
| 542 | |
| 543 | sj, err = sjson.Set(sj, "usage.output_tokens", event.Usage.OutputTokens) |
| 544 | if err != nil { |
| 545 | return nil, xerrors.Errorf("marshal event usage failed: %w", err) |
| 546 | } |
| 547 | |
| 548 | return i.encodeForStream([]byte(sj), event.Type), nil |
| 549 | } |
| 550 | |
| 551 | func (i *StreamingInterception) marshal(payload any) ([]byte, error) { |
| 552 | data, err := json.Marshal(payload) |
no test coverage detected