新增header头

juxiaoming · Sep 16, 2024 · f5e0f48 · f5e0f48
1 parent 3b31eb4
commit f5e0f48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/implement_baichuan.go b/implement_baichuan.go
@@ -107,7 +107,7 @@ type BaiChuanChatResponse struct {
 
 func (b *BaiChuanServer) Chat(requestPath string, data []byte) (*Response, error) {
 	ret := &Response{RequestData: data}
-	headers := map[string]string{"Authorization": "Bearer " + b.Conf.Key}
+	headers := map[string]string{"Authorization": "Bearer " + b.Conf.Key, "Content-Type": "application/json"}
 	response, err := postBase(requestPath, string(data), headers)
 	if err != nil {
 		return ret, err
@@ -182,7 +182,7 @@ type BaiChuanErrorInfo struct {
 func (b *BaiChuanServer) ChatStream(requestPath string, data []byte, msgCh chan string, errChan chan error) (*Response, error) {
 	ret := &Response{RequestData: data, ResponseData: make([]byte, 0)}
 
-	headers := map[string]string{"Authorization": "Bearer " + b.Conf.Key}
+	headers := map[string]string{"Authorization": "Bearer " + b.Conf.Key, "Content-Type": "application/json"}
 	response, err := postBase(requestPath, string(data), headers)
 	if err != nil {
 		errChan <- err