streamAdapter adapts Bedrock's ConverseStreamEventStream to chat.MessageStream
| 14 | |
| 15 | // streamAdapter adapts Bedrock's ConverseStreamEventStream to chat.MessageStream |
| 16 | type streamAdapter struct { |
| 17 | stream *bedrockruntime.ConverseStreamEventStream |
| 18 | model string |
| 19 | trackUsage bool |
| 20 | |
| 21 | // State for accumulating tool call data |
| 22 | currentToolID string |
| 23 | currentToolName string |
| 24 | |
| 25 | // Buffered state for proper event ordering |
| 26 | // Bedrock sends MessageStop before Metadata, but runtime expects usage before FinishReason |
| 27 | pendingFinishReason chat.FinishReason |
| 28 | pendingUsage *chat.Usage |
| 29 | metadataReceived bool |
| 30 | } |
| 31 | |
| 32 | func newStreamAdapter(stream *bedrockruntime.ConverseStreamEventStream, model string, trackUsage bool) *streamAdapter { |
| 33 | return &streamAdapter{ |
nothing calls this directly
no outgoing calls
no test coverage detected