support_response.go

import (
    "github.com/sashabaranov/go-openai"
    "github.com/sashabaranov/go-openai/jsonschema"
)

type SupportResponse struct {
	Answer      string   `json:"answer"`
	RelatedDocs []string `json:"related_docs"`
}

func GetSupportResponse(messages []openai.ChatCompletionMessage) (*SupportResponse, error) {
	var supportResponse SupportResponse
	schema, err := jsonschema.GenerateSchemaForType(supportResponse)
	if err != nil {
		return nil, err
	}

	resp, err := client.CreateChatCompletion(ctx, openai.ChatCompletionRequest{
		Messages: messages,
		ResponseFormat: &openai.ChatCompletionResponseFormat{
			Type: openai.ChatCompletionResponseFormatTypeJSONSchema,
			JSONSchema: &openai.ChatCompletionResponseFormatJSONSchema{
				Name:   "support_response",
				Schema: schema,
				Strict: true,
			},
		},
	})
	if err != nil {
		return nil, err
	}
	err = schema.Unmarshal(resp.Choices[0].Message.Content, &supportResponse)
	if err != nil {
		return nil, err
	}
	return &supportResponse, nil
}