Function parseAIResponse

module/service/parse.go:45–95

ParseAIResponse parses both streaming and non-streaming AI responses and returns the concatenated content

(input string)

// ParseAIResponse parses both streaming and non-streaming AI responses and returns the concatenated content
func parseAIResponse(input string) (string, error) {
	// First, try to parse as a non-streaming response
	var nonStreaming ChatCompletion
	if err := json.Unmarshal([]byte(input), &nonStreaming); err == nil && nonStreaming.Object == "chat.completion" {
		var result strings.Builder
		for _, choice := range nonStreaming.Choices {
			result.WriteString(choice.Message.Content)
		}
		return result.String(), nil
	}

	// If not non-streaming, parse as streaming response
	var result strings.Builder
	scanner := bufio.NewScanner(strings.NewReader(input))

	for scanner.Scan() {
		line := scanner.Text()
		// Skip empty lines or [DONE]
		if line == "" || line == "data: [DONE]" {
			continue
		}

		// Check if line starts with "data: "
		if !strings.HasPrefix(line, "data: ") {
			continue
		}

		// Extract JSON data
		jsonData := strings.TrimPrefix(line, "data: ")
		var chunk ChatCompletionChunk
		if err := json.Unmarshal([]byte(jsonData), &chunk); err != nil {
			return "", err
		}

		// Process each choice
		for _, choice := range chunk.Choices {
			// Append content from delta
			result.WriteString(choice.Delta.Content)
			// Check if this is the final chunk
			if choice.FinishReason != nil && *choice.FinishReason == "stop" {
				return result.String(), nil
			}
		}
	}

	if err := scanner.Err(); err != nil {
		return "", err
	}

	return result.String(), nil
}

func parseAIRequest(ori string) string {
	type aiRequest struct {