providers

type AnthropicClient[R any] struct {
	// contains filtered or unexported fields
}

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

type CerebrasClient[R any] struct {
	// contains filtered or unexported fields
}

type CerebrasJsonSchema struct {
	Name   string `json:"name"`
	Strict bool   `json:"strict"`
	Schema any    `json:"schema"`
}

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

type CerebrasRequest struct {
	Model          string                  `json:"model"`
	Messages       []CerebrasMessage       `json:"messages"`
	Temperature    *float64                `json:"temperature,omitempty"`
	MaxTokens      *int                    `json:"max_completion_tokens,omitempty"`
	ResponseFormat *CerebrasResponseFormat `json:"response_format,omitempty"`
}

type CerebrasResponse struct {
	ID      string           `json:"id"`
	Choices []CerebrasChoice `json:"choices"`
	Usage   CerebrasUsage    `json:"usage"`
	Created int64            `json:"created"`
	Model   string           `json:"model"`
}

type CerebrasResponseFormat struct {
	Type       string             `json:"type"`
	JsonSchema CerebrasJsonSchema `json:"json_schema"`
}

type CerebrasUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type Client[R any] interface {
	// Complete sends a completion request to the AI provider
	Complete(ctx context.Context, req *Request) (*R, Usage, error)

	// Name returns the name of the provider
	Name() string

	// Validate checks if the client configuration is valid
	Validate() error
}

type Config struct {
	Provider    Provider
	Model       string
	APIKey      string
	BaseURL     string
	Temperature float64
	MaxTokens   int
}

type GeminiClient[R any] struct {
	// contains filtered or unexported fields
}

type IssueResponse struct {
	Issues []SemanticIssue
}

type OllamaClient[R any] struct {
	// contains filtered or unexported fields
}

type OpenAIClient[R any] struct {
	// contains filtered or unexported fields
}

type Provider string

type ProviderDefaults struct {
	Model     string
	ApiKeyVar string
}

type Request struct {
	SystemPrompt string
	UserPrompt   string
}

type SemanticIssue struct {
	Reasoning  string `json:"reasoning" jsonschema_description:"Reasoning why the found issue has it's severity level"`
	Level      string `json:"level" jsonschema_description:"Severity level of the issue"`
	Message    string `json:"message" jsonschema_description:"Description of the issue"`
	Suggestion string `json:"suggestion" jsonschema_description:"Suggestion for fixing the issue (optional)"`
	File       string `json:"file" jsonschema_description:"The file that the issue is in"`
}

type Usage struct {
	InputTokens  int
	OutputTokens int
	TotalTokens  int
}