llm

type APIError struct {
	Message string `json:"message"`
	Type    string `json:"type"`
	Code    int    `json:"code,omitempty"`
}

type APIFunction struct {
	Name        string     `json:"name"`
	Description string     `json:"description"`
	Parameters  Parameters `json:"parameters"`
}

type ChatRequest struct {
	Model     string     `json:"model"`
	Messages  []Message  `json:"messages"`
	Functions []Function `json:"tools,omitempty"`
}

type ChatResponse struct {
	Choices []MessageChoice `json:"choices"`
	Error   *APIError       `json:"error,omitempty"`
}

type Client interface {
	GenerateText(ctx context.Context, systemPrompt, userPrompt string, functions []APIFunction) (string, error)
}

type Function struct {
	Type     string      `json:"type"`
	Function FunctionDef `json:"function"`
}

type FunctionCall struct {
	Function struct {
		Name      string `json:"name"`
		Arguments string `json:"arguments"`
	} `json:"function"`
}

type FunctionCallArguments struct {
	File                  string `json:"file"`
	Status                string `json:"status"`
	Diff                  string `json:"diff"`
	HasSignificantChanges bool   `json:"hasSignificantChanges"`
	Summary               string `json:"summary"`
}

type FunctionChoice struct {
	Type     string `json:"type,omitempty"`
	Function *struct {
		Name string `json:"name,omitempty"`
	} `json:"function,omitempty"`
}

type FunctionDef struct {
	Name        string     `json:"name"`
	Description string     `json:"description"`
	Parameters  Parameters `json:"parameters"`
}

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

type MessageChoice struct {
	Message MessageResponse `json:"message"`
	Index   int             `json:"index"`
	Role    string          `json:"role,omitempty"`
	Content string          `json:"content,omitempty"`
}

type MessageResponse struct {
	Content       string         `json:"content"`
	FunctionCalls []FunctionCall `json:"tool_calls,omitempty"` //nolint:tagliatelle // Following OpenAI API spec
}

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

type Parameters struct {
	Type       string              `json:"type"`
	Properties map[string]Property `json:"properties"`
	Required   []string            `json:"required"`
}

type Property struct {
	Type        string   `json:"type"`
	Description string   `json:"description"`
	Enum        []string `json:"enum,omitempty"`
}

type RateLimitInfo struct {
	RemainingTokens   int
	TokensResetIn     time.Duration
	RemainingRequests int
	RequestsResetIn   time.Duration
	RetryAfter        time.Duration // Only set when receiving 429
}

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

type WaitInfo struct {
	NeedsToWait     bool
	RemainingTokens int
	WaitTime        time.Duration
	ResetAt         time.Time
}