ai

type AnthropicClient struct {
	*BaseClient
}

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

type AnthropicRequest struct {
	Model     string             `json:"model"`
	MaxTokens int                `json:"max_tokens"`
	Messages  []AnthropicMessage `json:"messages"`
	System    string             `json:"system,omitempty"`
}

type AnthropicResponse struct {
	Content []struct {
		Type string `json:"type"`
		Text string `json:"text"`
	} `json:"content"`
	Error *struct {
		Type    string `json:"type"`
		Message string `json:"message"`
	} `json:"error,omitempty"`
}

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

type ChatCompletionRequest struct {
	Model       string        `json:"model"`
	Messages    []ChatMessage `json:"messages"`
	Temperature *float64      `json:"temperature,omitempty"`
	MaxTokens   *int          `json:"max_tokens,omitempty"`
}

type ChatCompletionResponse struct {
	Choices []struct {
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Error *struct {
		Message string `json:"message"`
		Type    string `json:"type"`
		Code    string `json:"code"`
	} `json:"error,omitempty"`
}

type ChatMessage struct {
	Role    string `json:"role"` // "system", "user", "assistant"
	Content string `json:"content"`
}

type Client interface {
	GeneratePRSummary(prTitle, prDescription, diff string) (*PRSummary, error)
	GenerateCodeReview(prTitle, prDescription, diff string) (*ReviewResult, error)
	GenerateCodeReviewWithStyleGuide(prTitle, prDescription, diff, styleGuide string) (*ReviewResult, error)
	GenerateResponse(prompt string) (string, error) // For conversational responses
}

type Comment struct {
	File            string `json:"file"`
	StartLine       int    `json:"start_line"`
	EndLine         int    `json:"end_line"`
	HighlightedCode string `json:"highlighted_code"`
	Header          string `json:"header"`
	Content         string `json:"content"`
	Label           string `json:"label"` // e.g. "bug", "security"
	Critical        bool   `json:"critical"`
	SuggestedCode   string `json:"suggested_code,omitempty"` // GitHub suggestion block content
}

type Config struct {
	Provider string
	APIKey   string
	Model    string
	BaseURL  string
}

type GoogleClient struct {
	*BaseClient
}

type GoogleContent struct {
	Role  string       `json:"role"`
	Parts []GooglePart `json:"parts"`
}

type GoogleGenConfig struct {
	Temperature     *float64 `json:"temperature,omitempty"`
	MaxOutputTokens *int     `json:"maxOutputTokens,omitempty"`
}

type GooglePart struct {
	Text string `json:"text"`
}

type GoogleRequest struct {
	Contents          []GoogleContent  `json:"contents"`
	SystemInstruction *GoogleContent   `json:"systemInstruction,omitempty"`
	GenerationConfig  *GoogleGenConfig `json:"generationConfig,omitempty"`
}

type GoogleResponse struct {
	Candidates []struct {
		Content struct {
			Parts []GooglePart `json:"parts"`
			Role  string       `json:"role"`
		} `json:"content"`
		FinishReason string `json:"finishReason"`
	} `json:"candidates"`
	Error *struct {
		Code    int    `json:"code"`
		Message string `json:"message"`
		Status  string `json:"status"`
	} `json:"error,omitempty"`
}

type OpenAIClient struct {
	*BaseClient
}

type OpenRouterClient struct {
	*BaseClient
}

type PRSummary struct {
	Title       string   `json:"title"` // Max 10 words
	Description string   `json:"description"`
	Type        []string `json:"type"` // e.g., BUG, ENHANCEMENT
	Files       []struct {
		Filename string `json:"filename"`
		Summary  string `json:"summary"` // Max 70 words
		Title    string `json:"title"`   // 5-10 words
	} `json:"files"`
}

type ReviewAction string

type ReviewResult struct {
	Review   ReviewSummary `json:"review"`
	Comments []Comment     `json:"comments"`
}

type ReviewSummary struct {
	EstimatedEffort  int    `json:"estimated_effort_to_review"` // 1-5
	Score            int    `json:"score"`                      // 0-100
	HasRelevantTests bool   `json:"has_relevant_tests"`
	SecurityConcerns string `json:"security_concerns"`
}