huggingface

type ChatCompletion struct {
	Object string `json:"object"`

	ID string `json:"id"`

	Model   string `json:"model"`
	Created int64  `json:"created"`

	Choices []ChatCompletionChoice `json:"choices"`
}

type ChatCompletionChoice struct {
	Index int `json:"index"`

	Delta   *ChatCompletionMessage `json:"delta,omitempty"`
	Message *ChatCompletionMessage `json:"message,omitempty"`

	FinishReason *CompletionReason `json:"finish_reason"`
}

type ChatCompletionContent struct {
	Type string `json:"type,omitempty"`
	Text string `json:"text,omitempty"`
}

type ChatCompletionMessage struct {
	Role MessageRole `json:"role,omitempty"`

	Content string `json:"content"`
}

type ChatCompletionRequest struct {
	Model string `json:"model"`

	Messages []ChatCompletionMessage `json:"messages"`

	Stop   []string `json:"stop,omitempty"`
	Stream bool     `json:"stream,omitempty"`

	MaxTokens   *int     `json:"max_tokens,omitempty"`
	Temperature *float32 `json:"temperature,omitempty"`
}

type Client struct {
	*Embedder
	*Completer
}

type Completer struct {
	*Config
}

type CompletionReason string

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

type Embedder struct {
	*Config
}

type MessageRole string

type Option func(*Config)