types

type AgentContext struct {
	WorkspaceRoot string            `json:"workspace_root"`
	CurrentFiles  []string          `json:"current_files"`
	GitStatus     string            `json:"git_status,omitempty"`
	Metadata      map[string]string `json:"metadata,omitempty"`
}

type AgentTokenUsage struct {
	PromptTokens     int     `json:"prompt_tokens"`
	CompletionTokens int     `json:"completion_tokens"`
	TotalTokens      int     `json:"total_tokens"`
	EstimatedCost    float64 `json:"estimated_cost"`
	Model            string  `json:"model,omitempty"`
	Provider         string  `json:"provider,omitempty"`
}

type ChatResponse struct {
	ID      string   `json:"id"`
	Object  string   `json:"object"`
	Created int64    `json:"created"`
	Model   string   `json:"model"`
	Choices []Choice `json:"choices"`
	Usage   Usage    `json:"usage"`
}

type Choice struct {
	Index   int `json:"index"`
	Message struct {
		Role             string         `json:"role"`
		Content          string         `json:"content"`
		ReasoningContent string         `json:"reasoning_content,omitempty"`
		Images           []ImageData    `json:"images,omitempty"`
		ToolCalls        []api.ToolCall `json:"tool_calls,omitempty"`
	} `json:"message"`
	FinishReason string `json:"finish_reason"`
}

type CodeReviewResult struct {
	Issues           []string         `json:"issues,omitempty"`
	Suggestions      []string         `json:"suggestions,omitempty"`
	Approved         bool             `json:"approved"`
	Status           string           `json:"status,omitempty"`
	Feedback         string           `json:"feedback,omitempty"`
	DetailedGuidance string           `json:"detailed_guidance,omitempty"`
	NewPrompt        string           `json:"new_prompt,omitempty"`
	PatchResolution  *PatchResolution `json:"patch_resolution,omitempty"`
}

type EditPlan struct {
	Target      string   `json:"target"`
	Changes     []string `json:"changes"`
	Rationale   string   `json:"rationale"`
	Files       []string `json:"files"`
	TestChanges bool     `json:"test_changes"`
}

type ImageData struct {
	URL    string `json:"url,omitempty"`    // URL to image
	Base64 string `json:"base64,omitempty"` // Base64 encoded image data
	Type   string `json:"type,omitempty"`   // MIME type (image/jpeg, image/png, etc.)
}

type IntentAnalysis struct {
	Intent       string            `json:"intent"`
	Actions      []string          `json:"actions"`
	Files        []string          `json:"files"`
	Confidence   float64           `json:"confidence"`
	Dependencies map[string]string `json:"dependencies,omitempty"`
}

type Message struct {
	Role             string      `json:"role"`
	Content          string      `json:"content"`
	ReasoningContent string      `json:"reasoning_content,omitempty"`
	Images           []ImageData `json:"images,omitempty"` // Support for multiple images
}

type ModelInfo struct {
	ID            string  `json:"id"`
	Name          string  `json:"name"`
	Provider      string  `json:"provider"`
	Description   string  `json:"description,omitempty"`
	ContextLength int     `json:"context_length,omitempty"`
	InputCost     float64 `json:"input_cost,omitempty"`
	OutputCost    float64 `json:"output_cost,omitempty"`
	Cost          float64 `json:"cost,omitempty"`
}

type ModelPricing struct {
	InputCost       float64 `json:"input_cost"`
	OutputCost      float64 `json:"output_cost"`
	InputCostPer1K  float64 `json:"input_cost_per_1k"`  // Alias for compatibility
	OutputCostPer1K float64 `json:"output_cost_per_1k"` // Alias for compatibility
}

type PatchResolution struct {
	ApprovedChanges []string          `json:"approved_changes,omitempty"`
	RejectedChanges []string          `json:"rejected_changes,omitempty"`
	Comments        []string          `json:"comments,omitempty"`
	Status          string            `json:"status,omitempty"`
	SingleFile      string            `json:"SingleFile"`
	MultiFile       map[string]string `json:"MultiFile"`
}

type PricingTable struct {
	Models map[string]ModelPricing `json:"models"`
}

type TokenUsage = Usage

type ToolCallFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type Usage struct {
	PromptTokens        int     `json:"prompt_tokens"`
	CompletionTokens    int     `json:"completion_tokens"`
	TotalTokens         int     `json:"total_tokens"`
	EstimatedCost       float64 `json:"estimated_cost"`
	Estimated           bool    `json:"estimated,omitempty"`
	PromptTokensDetails struct {
		CachedTokens     int  `json:"cached_tokens"`
		CacheWriteTokens *int `json:"cache_write_tokens"`
	} `json:"prompt_tokens_details,omitempty"`
}