(event anthropic.MessageStreamEventUnion)
| 632 | } |
| 633 | |
| 634 | func (i *StreamingInterception) marshalEvent(event anthropic.MessageStreamEventUnion) ([]byte, error) { |
| 635 | sj, err := sjson.Set(event.RawJSON(), "message.id", i.ID().String()) |
| 636 | if err != nil { |
| 637 | return nil, xerrors.Errorf("marshal event id failed: %w", err) |
| 638 | } |
| 639 | |
| 640 | sj, err = sjson.Set(sj, "usage.output_tokens", event.Usage.OutputTokens) |
| 641 | if err != nil { |
| 642 | return nil, xerrors.Errorf("marshal event usage failed: %w", err) |
| 643 | } |
| 644 | |
| 645 | return i.encodeForStream([]byte(sj), event.Type), nil |
| 646 | } |
| 647 | |
| 648 | func (i *StreamingInterception) marshal(payload any) ([]byte, error) { |
| 649 | data, err := json.Marshal(payload) |
no test coverage detected