From eaadb32bba93aac79a67168bfe35483d03cb8a06 Mon Sep 17 00:00:00 2001
From: "Laisky.Cai" <github@laisky.com>
Date: Fri, 19 Apr 2024 01:49:57 +0000
Subject: [PATCH] fix: Implement AWS handler with usage tracking and error
 handling

- Implemented streaming response handling for AWS handler
- Set response content type to text/event-stream
- Added error handling for failed marshaling/unmarshaling
- Updated return values to include `relaymodel.ErrorWithStatusCode` and `relaymodel.Usage`
- Improved error handling and response formatting for AWS adaptor
---
 relay/adaptor/aws/main.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/relay/adaptor/aws/main.go b/relay/adaptor/aws/main.go
index 43a631e3..6cd21f7f 100644
--- a/relay/adaptor/aws/main.go
+++ b/relay/adaptor/aws/main.go
@@ -183,6 +183,7 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*relaymodel.ErrorWithSt
 	stream := awsResp.GetStream()
 	defer stream.Close()
 
+	c.Writer.Header().Set("Content-Type", "text/event-stream")
 	var usage relaymodel.Usage
 	var id string
 	c.Stream(func(w io.Writer) bool {