llm

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

type ChatRequest struct {
	Messages    []ChatMessage
	Stream      bool
	Temperature float32
	MaxTokens   int
}

type ChatResponse struct {
	Content      string
	FinishReason string
	Usage        Usage
}

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

type Column struct {
	Name       string
	Type       string
	Nullable   bool
	PrimaryKey bool
	ForeignKey string
}

type Config struct {
	BaseURL     string
	APIKey      string
	Model       string
	Temperature float32
	MaxTokens   int
}

type GenerateSQLRequest struct {
	Question string
	Schema   string
	History  []ChatMessage
}

type GenerateSQLResponse struct {
	SQL          string
	RawResponse  string
	TokensUsed   int
	FinishReason string
}

type SQLError struct {
	Message string
}

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

type StreamCallback func(chunk string) error

type Table struct {
	Name    string
	Columns []Column
}

type Usage struct {
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
}