openai

type Config struct {
	RequestsPerSecond float64 `json:"rps"`
	Duration          float64 `json:"duration"`

	APIKey              string `json:"apikey"`
	Endpoint            string `json:"endpoint"`
	Model               string `json:"model"`
	MaxCompletionTokens int    `json:"maxtokens"`
}

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

type OpenAIGenerator struct {
	RPS      float64
	Duration time.Duration

	// OpenAI API key
	APIKey              string
	Method              string
	Endpoint            string
	Model               string
	MaxCompletionTokens int
}

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

type OpenAIRequestBody struct {
	Model               string          `json:"model"`
	Messages            []OpenAIMessage `json:"messages"`
	Stream              bool            `json:"stream"`
	StreamOptions       StreamOptions   `json:"stream_options"`
	MaxCompletionTokens int             `json:"max_completion_tokens"`
}

type StreamOptions struct {
	IncludeUsage bool `json:"include_usage"`
}