package openai

import (
	"context"
	"encoding/json"

	"github.com/sashabaranov/go-openai"
)

type (
	ChatReq struct {
		Role    string `json:"role"`
		Content string `json:"content"`
		// Images 可选图片 url 列表：非空时该消息以 OpenAI 多模态格式（text + image_url）发送，
		// 用于视觉模型（如通义千问-VL / Azure OpenAI 视觉模型）。为空时按纯文本发送。
		Images []string `json:"images,omitempty"`
	}
	Parameters struct {
		Type       string              `json:"type"`
		Properties map[string]Property `json:"properties"`
		Required   []string            `json:"required"`
	}
	Property struct {
		Type        string `json:"type"`
		Description string `json:"description"`
		// Pattern     string        `json:"pattern"`
		// MaxLength   string        `json:"max_length"`
		// Examples    []interface{} `json:"examples"`
	}
	ChatResponse struct {
		Role    string                 `json:"role"`
		Content string                 `json:"content,omitempty"`
		Meta    map[string]interface{} `json:"meta,omitempty"`
	}
	//回应流式切片
	ChatResponseChoice struct {
		Role    string                 `json:"role"`
		Content string                 `json:"content,omitempty"`
		Meta    map[string]interface{} `json:"meta,omitempty"`
	}
	ITool interface {
		GetToolInfo() openai.Tool
		Execute(args json.RawMessage) (any, string, error)
	}
	ISys interface {
		RegisterTools(tool ITool)
		Chat(ctx context.Context, messages []ChatReq) (resp *ChatResponse, err error)
		ChatForSteams(ctx context.Context, messages []ChatReq, choiceChan chan *ChatResponseChoice) (err error)
		CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (response openai.ChatCompletionResponse, err error)
		CreateChatCompletionStream(ctx context.Context, request openai.ChatCompletionRequest) (stream *openai.ChatCompletionStream, err error)
	}
)

var defsys ISys

func OnInit(config map[string]interface{}, option ...Option) (err error) {
	defsys, err = newSys(newOptions(config, option...))
	return
}

func NewSys(option ...Option) (sys ISys, err error) {
	sys, err = newSys(newOptionsByOption(option...))
	return
}

func RegisterTools(tool ITool) {
	defsys.RegisterTools(tool)
}

func Chat(ctx context.Context, messages []ChatReq) (resp *ChatResponse, err error) {
	return defsys.Chat(ctx, messages)
}

func ChatForSteams(ctx context.Context, messages []ChatReq, choiceChan chan *ChatResponseChoice) (err error) {
	return defsys.ChatForSteams(ctx, messages, choiceChan)
}

func CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (response openai.ChatCompletionResponse, err error) {
	return defsys.CreateChatCompletion(ctx, request)
}

func CreateChatCompletionStream(ctx context.Context, request openai.ChatCompletionRequest) (stream *openai.ChatCompletionStream, err error) {
	return defsys.CreateChatCompletionStream(ctx, request)
}