client

type ChatClient interface {
	Chat(ctx context.Context, request *ChatRequest) (*ChatResponse, error)
}

type ChatRequest struct {
	Messages        []*Message
	Temperature     float64
	Schema          *jsonschema.Schema
	Model           string
	MaxOutputTokens int64
	PresencePenalty float64
}

type ChatResponse struct {
	Content string `json:"content"`
	Tokens  int64  `json:"tokens"`
}

type Content struct {
	Type ContentType
	Data string
}

type ContentType string

type ImageGenerateRequest struct {
	Prompt string
	Model  string
}

type ImageResponse struct {
	URL     string `json:"url"`
	Content string `json:"content"`
	Tokens  int    `json:"tokens"`
}

type Message struct {
	Role    string
	Content []Content
}

type OpenAIClient struct {
	// contains filtered or unexported fields
}