apitest

type APITester interface {
	TestChannel(context.Context, *TestConfig) TestResult
	TestAllChannels(context.Context, []*TestConfig) []TestResult
	TestAllApis([]*Channel) []TestResult
	PrintResults([]TestResult) error
}

type Channel struct {
	Key       string      `json:"key"`
	TestModel []string    `json:"test_model"`
	URL       string      `json:"url"`
	Type      ChannelType `json:"type"`
}

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

type ChannelTestConfig struct {
	MaxConcurrency int
	Timeout        time.Duration
	ResultBuffer   int
}

type ChannelTestOption func(*ChannelTest)

type ChannelType int

type DefaultRequestBuilder struct{}

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

type GeminiError struct {
	Error struct {
		Code    int    `json:"code"`
		Message string `json:"message"`
		Status  string `json:"status"`
		Details []struct {
			Type     string            `json:"@type"`
			Reason   string            `json:"reason,omitempty"`
			Domain   string            `json:"domain,omitempty"`
			Metadata map[string]string `json:"metadata,omitempty"`
			Message  string            `json:"message,omitempty"`
			Locale   string            `json:"locale,omitempty"`
		} `json:"details"`
	} `json:"error"`
}

type HTTPClient interface {
	Do(*http.Request) (*http.Response, error)
}

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

type OpenAIError struct {
	Error struct {
		Message string `json:"message"`
		Type    string `json:"type"`
		Code    string `json:"code"`
	} `json:"error"`
}

type OpenAIRequest struct {
	Model               string    `json:"model"`
	Messages            []Message `json:"messages"`
	Stream              bool      `json:"stream"`
	MaxTokens           int       `json:"max_tokens,omitempty"`
	MaxCompletionTokens int       `json:"max_completion_tokens,omitempty"`
}

type OpenAIResponse struct {
	Usage *Usage `json:"usage"`
}

type RequestBuilder interface {
	BuildRequest(context.Context, *TestConfig) (*http.Request, error)
}

type RequestOptions struct {
	MaxTokens   int
	Temperature float64
	TopP        float64
	TopK        int
	Stream      bool
}

type ResultProcessor interface {
	ProcessResponse(*http.Response) TestResult
}

type TestConfig struct {
	Channel     *Channel
	Model       string
	RequestOpts RequestOptions
	IsGemini    bool
}

type TestResult struct {
	Channel  *Channel
	Model    string
	Success  bool
	Latency  float64
	Error    error
	Response interface{}
}

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