Added TestBedrockSimple (streaming)

Author: evgeniy-scherbina

fixtures/bedrock/parse_eventstream.go

@@ -0,0 +1,88 @@
+//go:build ignore
+
+// Usage: go run parse_eventstream.go <file.resp.bin>
+//
+// Decodes an AWS EventStream binary file and prints each frame's
+// decoded JSON body. Use this to inspect captured Bedrock responses
+// and verify fixture contents.
+package main
+
+import (
+	"bytes"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"os"
+
+	"github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream"
+	"github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream/eventstreamapi"
+)
+
+func main() {
+	if len(os.Args) < 2 {
+		fmt.Fprintf(os.Stderr, "usage: go run parse_eventstream.go <file.resp.bin>\n")
+		os.Exit(1)
+	}
+
+	data, err := os.ReadFile(os.Args[1])
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "read file: %v\n", err)
+		os.Exit(1)
+	}
+
+	decoder := eventstream.NewDecoder()
+	reader := bytes.NewReader(data)
+	frameNum := 0
+
+	for {
+		msg, err := decoder.Decode(reader, nil)
+		if err != nil {
+			break
+		}
+		frameNum++
+
+		messageType := msg.Headers.Get(eventstreamapi.MessageTypeHeader)
+		eventType := msg.Headers.Get(eventstreamapi.EventTypeHeader)
+
+		fmt.Printf("=== Frame %d ===\n", frameNum)
+		fmt.Printf("  message-type: %s\n", headerStr(messageType))
+		fmt.Printf("  event-type:   %s\n", headerStr(eventType))
+
+		if headerStr(eventType) != "chunk" {
+			fmt.Printf("  payload: %s\n\n", string(msg.Payload))
+			continue
+		}
+
+		var chunk struct {
+			Bytes string `json:"bytes"`
+		}
+		if err := json.Unmarshal(msg.Payload, &chunk); err != nil {
+			fmt.Printf("  unmarshal error: %v\n\n", err)
+			continue
+		}
+
+		decoded, err := base64.StdEncoding.DecodeString(chunk.Bytes)
+		if err != nil {
+			fmt.Printf("  base64 decode error: %v\n\n", err)
+			continue
+		}
+
+		var pretty json.RawMessage
+		if err := json.Unmarshal(decoded, &pretty); err != nil {
+			fmt.Printf("  json: %s\n\n", string(decoded))
+			continue
+		}
+
+		indented, _ := json.MarshalIndent(pretty, "  ", "  ")
+		fmt.Printf("  body:\n  %s\n\n", string(indented))
+	}
+
+	fmt.Printf("Total frames: %d\n", frameNum)
+}
+
+func headerStr(h eventstream.Value) string {
+	if h == nil {
+		return "<nil>"
+	}
+	return h.String()
+}

fixtures/bedrock/simple.req.json

@@ -0,0 +1,10 @@
+{
+  "anthropic_version": "bedrock-2023-05-31",
+  "max_tokens": 50,
+  "messages": [
+    {
+      "role": "user",
+      "content": "how many angels can dance on the head of a pin"
+    }
+  ]
+}