warp

type AIQueryRow struct {
	ID               int
	ExchangeID       string
	ConversationID   string
	Input            string // JSON array
	OutputStatus     string
	ModelID          string
	PlanningModelID  string
	CodingModelID    string
	WorkingDirectory string
	StartTS          time.Time
}

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

type AgentConversationRow struct {
	ID               int
	ConversationID   string
	ConversationData string // JSON
	LastModifiedAt   time.Time
}

type BlockAIMetadata struct {
	ActionID          string          `json:"action_id"`
	ConversationID    string          `json:"conversation_id"`
	ConversationPhase json.RawMessage `json:"conversation_phase"`
}

type BlockRow struct {
	ID              int
	PaneLeafUUID    []byte
	StylizedCommand []byte
	StylizedOutput  []byte
	PWD             string
	ExitCode        int
	StartTS         time.Time
	CompletedTS     time.Time
	AIMetadata      string // JSON, nullable
}

type CodebaseContext struct {
	Path string `json:"path"`
	Name string `json:"name"`
}

type ConversationData struct {
	ServerConversationToken string                     `json:"server_conversation_token"`
	UsageMetadata           *ConversationUsageMetadata `json:"conversation_usage_metadata"`
}

type ConversationUsageMetadata struct {
	WasSummarized         bool               `json:"was_summarized"`
	ContextWindowUsage    float64            `json:"context_window_usage"`
	CreditsSpent          float64            `json:"credits_spent"`
	CreditsSpentLastBlock float64            `json:"credits_spent_for_last_block"`
	TokenUsage            []TokenUsageItem   `json:"token_usage"`
	ToolUsageMetadata     *ToolUsageMetadata `json:"tool_usage_metadata"`
}

type CurrentTimeContext struct {
	CurrentTime string `json:"current_time"`
}

type DiffStats struct {
	Count        int `json:"count"`
	LinesAdded   int `json:"lines_added"`
	LinesRemoved int `json:"lines_removed"`
}

type DirectoryContext struct {
	PWD     string `json:"pwd"`
	HomeDir string `json:"home_dir"`
}

type ExecutionEnvironmentContext struct {
	OS        *OSInfo `json:"os"`
	ShellName string  `json:"shell_name"`
}

type GitContext struct {
	Head string `json:"head"`
}

type OSInfo struct {
	Category string `json:"category"`
}

type ProjectRulesContext struct {
	RootPath    string          `json:"root_path"`
	ActiveRules json.RawMessage `json:"active_rules"` // Complex nested structure, not parsed
}

type QueryContext struct {
	Directory            *DirectoryContext            `json:",omitempty"`
	Git                  *GitContext                  `json:",omitempty"`
	ProjectRules         *ProjectRulesContext         `json:",omitempty"`
	CurrentTime          *CurrentTimeContext          `json:",omitempty"`
	ExecutionEnvironment *ExecutionEnvironmentContext `json:",omitempty"`
	Codebase             *CodebaseContext             `json:",omitempty"`
}

type QueryInput struct {
	Query struct {
		Text                  string          `json:"text"`
		Context               []QueryContext  `json:"context"`
		ReferencedAttachments json.RawMessage `json:"referenced_attachments"`
	} `json:"Query"`
}

type TokenUsageItem struct {
	ModelID              string         `json:"model_id"`
	WarpTokens           int            `json:"warp_tokens"`
	BYOKTokens           int            `json:"byok_tokens"`
	WarpTokensByCategory map[string]int `json:"warp_token_usage_by_category"`
}

type ToolStats struct {
	Count            int `json:"count"`
	CommandsExecuted int `json:"commands_executed,omitempty"`
}

type ToolUsageMetadata struct {
	RunCommand      *ToolStats `json:"run_command_stats"`
	ReadFiles       *ToolStats `json:"read_files_stats"`
	Grep            *ToolStats `json:"grep_stats"`
	FileGlob        *ToolStats `json:"file_glob_stats"`
	ApplyFileDiff   *DiffStats `json:"apply_file_diff_stats"`
	ReadShellOutput *ToolStats `json:"read_shell_command_output_stats"`
}