🐛 fix: cohere stream error (MartialBE#319)

wk920 · Jul 29, 2024 · 28ab819 · 28ab819
1 parent e2f7951
commit 28ab819
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/providers/cohere/chat.go b/providers/cohere/chat.go
@@ -156,11 +156,13 @@ func ConvertToChatOpenai(provider base.ProviderInterface, response *CohereRespon
 // 转换为OpenAI聊天流式请求体
 func (h *CohereStreamHandler) HandlerStream(rawLine *[]byte, dataChan chan string, errChan chan error) {
  // 如果rawLine 前缀不为data:，则直接返回
- if !strings.HasPrefix(string(*rawLine), "{") {
+ if !strings.HasPrefix(string(*rawLine), "data: ") {
  *rawLine = nil
  return
  }
 
+ *rawLine = (*rawLine)[6:]
+
  var cohereResponse CohereStreamResponse
  err := json.Unmarshal(*rawLine, &cohereResponse)
  if err != nil {