Function GetResponseContent

api/controller/response_handler.go:15–77

GetResponseContent 获取响应内容

(c *gin.Context, isStream bool, resp *http.Response)

// GetResponseContent 获取响应内容
func GetResponseContent(c *gin.Context, isStream bool, resp *http.Response) (string, string) {
	if resp == nil {
		return "", ""
	}

	if !isStream {
		respBody, err := io.ReadAll(resp.Body)
		if err != nil {
			logger.Errorf(c.Request.Context(), "read response body failed: %s", err.Error())
			return "", ""
		}
		// 重置响应体，以便后续处理
		resp.Body = io.NopCloser(bytes.NewBuffer(respBody))

		// 尝试解析不同格式的响应内容
		var openaiResp struct {
			Choices []struct {
				Message struct {
					Content string `json:"content"`
				} `json:"message"`
				Text             string `json:"text"`
				ReasoningContent string `json:"reasoning_content"`
			} `json:"choices"`
			Text             string `json:"text"`
			ReasoningContent string `json:"reasoning_content"`
		}

		if err := json.Unmarshal(respBody, &openaiResp); err != nil {
			logger.Errorf(c.Request.Context(), "unmarshal response failed: %s", err.Error())
			return string(respBody), ""
		}

		// 优先检查 message.content (chat completions)
		if len(openaiResp.Choices) > 0 {
			if openaiResp.Choices[0].Message.Content != "" {
				return openaiResp.Choices[0].Message.Content, openaiResp.Choices[0].ReasoningContent
			}
			if openaiResp.Choices[0].Text != "" {
				return openaiResp.Choices[0].Text, openaiResp.Choices[0].ReasoningContent
			}
			if openaiResp.Choices[0].ReasoningContent != "" {
				return "", openaiResp.Choices[0].ReasoningContent
			}
		}
		if openaiResp.Text != "" {
			return openaiResp.Text, openaiResp.ReasoningContent
		}
		if openaiResp.ReasoningContent != "" {
			return "", openaiResp.ReasoningContent
		}
		return string(respBody), ""
	}

	// 对于流式响应，从上下文中获取收集器
	collector, exists := c.Get("stream_response_collector")
	if exists {
		if streamCollector, ok := collector.(*StreamResponseCollector); ok {
			return streamCollector.GetContent()