ai

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

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type Model struct {
	ID      string `json:"id"`
	Name    string `json:"name"`
	Pricing *struct {
		Prompt     any `json:"prompt"`
		Completion any `json:"completion"`
	} `json:"pricing,omitempty"`
	ContextLength    int `json:"context_length"`
	PerRequestLimits *struct {
		PromptTokens     any `json:"prompt_tokens"`
		CompletionTokens any `json:"completion_tokens"`
	} `json:"per_request_limits"`
}

type ModelsList struct {
	Data []Model `json:"data"`
}

type Prompt struct {
	Model     string
	MaxTokens int
	System    string
	User      string
}

type PromptRequest struct {
	Model     string    `json:"model"`
	MaxTokens int       `json:"max_tokens"`
	Messages  []Message `json:"messages"`
}

type PromptResponse struct {
	ID      string `json:"id"`
	Model   string `json:"model"`
	Choices []struct {
		Message Message `json:"message"`
	} `json:"choices"`
}

type ProviderConfig struct {
	Provider string
	Endpoint string
	APIKey   string
}