add role

cubatic45 · Mar 15, 2024 · 0ef7ed2 · 0ef7ed2
1 parent 07caf1f
commit 0ef7ed2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/proxy.go b/proxy.go
@@ -23,6 +23,7 @@ type streamOutput struct {
 	Choices           []struct {
 		Index int `json:"index"`
 		Delta struct {
+			Role    string `json:"role"`
 			Content string `json:"content"`
 		} `json:"delta"`
 	} `json:"choices"`
@@ -94,7 +95,8 @@ func (p *Proxy) scanStream(w http.ResponseWriter, r *http.Response) {
 			if len(output.Choices) == 0 {
 				continue
 			}
-			for _, c := range output.Choices {
+			for i, c := range output.Choices {
+				output.Choices[i].Delta.Role = "assistant"
 				if c.Delta.Content == "" {
 					continue
 				}