ai

type ChatRequest struct {
	Provider    string
	Model       string
	Messages    []Message
	MaxTokens   int
	Temperature float64
	Endpoint    string // Optional custom endpoint URL (overrides env)
}

type ChatResponse struct {
	Content string
	Model   string
	Usage   Usage
	// Tracking info for logging/debugging
	Endpoint     string // Full URL used for the request
	RequestJSON  string // Raw request body JSON
	ResponseJSON string // Raw response body JSON
}

type Client interface {
	Chat(functionID string, req ChatRequest) (*ChatResponse, error)
}

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

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

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

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

type TrackRequest struct {
	Provider     string
	Model        string
	Endpoint     string
	RequestJSON  string
	ResponseJSON *string
	Status       store.AIRequestStatus
	ErrorMessage *string
	InputTokens  *int
	OutputTokens *int
	DurationMs   int64
}

type Tracker interface {
	Track(executionID string, req TrackRequest)
	Requests(executionID string) []store.AIRequest
	RequestsPaginated(executionID string, limit, offset int) ([]store.AIRequest, int64)
}

type Usage struct {
	InputTokens  int
	OutputTokens int
}