package ai

import "time"

type ModelCatalogEntry struct {
	ModelID string `json:"model_id"`
	Label   string `json:"label"`
	Enabled bool   `json:"enabled"`
}

type AssistantPolicy struct {
	Enabled               bool                `json:"enabled"`
	OpenWebUIInternalURL  string              `json:"openwebui_internal_url"`
	PublicPath            string              `json:"public_path"`
	EmbedDefaultTemporary bool                `json:"embed_default_temporary"`
	DefaultModel          string              `json:"default_model"`
	EnabledTools          []string            `json:"enabled_tools"`
	ChatSyncEnabled       bool                `json:"chat_sync_enabled"`
	ChatNCPath            string              `json:"chat_nc_path"`
	Models                []ModelCatalogEntry `json:"models,omitempty"`
}

type QuotaLimits struct {
	RequestsPerDay int   `json:"llm_requests_per_day"`
	TokensPerMonth int64 `json:"llm_tokens_per_month"`
}

type QuotaStatus struct {
	RequestsUsedToday int   `json:"requests_used_today"`
	RequestsLimit     int   `json:"requests_limit"`
	TokensUsedMonth   int64 `json:"tokens_used_month"`
	TokensLimit       int64 `json:"tokens_limit"`
	RequestsRemaining int   `json:"requests_remaining"`
	TokensRemaining   int64 `json:"tokens_remaining"`
}

type ChatMessage struct {
	Role       string         `json:"role"`
	Content    string         `json:"content,omitempty"`
	ToolCalls  []any          `json:"tool_calls,omitempty"`
	Attachments []any         `json:"attachments,omitempty"`
}

type ChatMeta struct {
	Model      string `json:"model,omitempty"`
	TokensUsed int64  `json:"tokens_used,omitempty"`
	Context    string `json:"context,omitempty"`
}

type ChatRecord struct {
	ID               string        `json:"id"`
	Title            string        `json:"title"`
	CreatedAt        time.Time     `json:"created_at"`
	UpdatedAt        time.Time     `json:"updated_at"`
	Source           string        `json:"source"`
	OpenWebUIChatID  string        `json:"openwebui_chat_id,omitempty"`
	Messages         []ChatMessage `json:"messages"`
	Meta             ChatMeta      `json:"meta"`
}

type ChatListItem struct {
	ID        string    `json:"id"`
	Title     string    `json:"title"`
	UpdatedAt time.Time `json:"updated_at"`
	Source    string    `json:"source"`
}

type SessionContext struct {
	App         string `json:"app"`
	Temporary   bool   `json:"temporary"`
	MessageID   string `json:"message_id,omitempty"`
	AccountID   string `json:"account_id,omitempty"`
	DrivePath   string `json:"drive_path,omitempty"`
	FileID      string `json:"file_id,omitempty"`
	ContactID   string `json:"contact_id,omitempty"`
	Subject     string `json:"subject,omitempty"`
	Snippet     string `json:"snippet,omitempty"`
}

type SessionResponse struct {
	SessionID   string `json:"session_id"`
	EmbedURL    string `json:"embed_url"`
	TokenSecret string `json:"token_secret,omitempty"`
	Temporary   bool   `json:"temporary"`
}