providers

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

type AnthropicProvider struct {
	Config
	// contains filtered or unexported fields
}

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

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

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

type Config struct {
	APIKey  string
	ModelID string
}

type GoogleContent struct {
	Parts []struct {
		Text string `json:"text"`
	} `json:"parts"`
}

type GoogleProvider struct {
	Config
	// contains filtered or unexported fields
}

type GoogleRequest struct {
	Contents []struct {
		Parts []struct {
			Text string `json:"text"`
		} `json:"parts"`
		Role string `json:"role,omitempty"`
	} `json:"contents"`
	GenerationConfig struct {
		MaxOutputTokens int `json:"maxOutputTokens"`
	} `json:"generationConfig"`
}

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

type LLMProvider interface {
	// Summarize takes a text input and returns a condensed summary
	Summarize(ctx context.Context, text string, maxLength int) (string, error)

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

type MockResponseConfig struct {
	StatusCode   int
	ResponseBody interface{}
	Headers      map[string]string
}

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

type OpenAIProvider struct {
	Config
	// contains filtered or unexported fields
}

type OpenAIRequest struct {
	Model     string          `json:"model"`
	Messages  []OpenAIMessage `json:"messages"`
	MaxTokens int             `json:"max_tokens"`
}

type OpenAIResponse struct {
	Choices []struct {
		Message struct {
			Content string `json:"content"`
		} `json:"message"`
	} `json:"choices"`
	Error *struct {
		Message string `json:"message"`
		Type    string `json:"type"`
	} `json:"error,omitempty"`
}

type ProviderFactory struct {
	// ProviderConfigs stores configuration for each provider
	ProviderConfigs map[string]Config
}

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

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

type XAIProvider struct {
	Config
	// contains filtered or unexported fields
}

type XAIRequest struct {
	Model     string       `json:"model"`
	Messages  []XAIMessage `json:"messages"`
	MaxTokens int          `json:"max_tokens"`
}

type XAIResponse struct {
	Choices []struct {
		Message struct {
			Content string `json:"content"`
		} `json:"message"`
	} `json:"choices"`
	Error *struct {
		Message string `json:"message"`
		Type    string `json:"type"`
	} `json:"error,omitempty"`
}