openaiwrap

type Config struct {
	// MaxRetries is the maximum number of retries for API calls. Defaults to 3.
	MaxRetries int `json:"max_retries" default:"3"`
	// InitialBackoff is the initial backoff duration in milliseconds. Defaults to 1000ms (1s).
	InitialBackoff int `json:"initial_backoff" default:"1000"`
	// MaxBackoff is the maximum backoff duration in milliseconds. Defaults to 30000ms (30s).
	MaxBackoff int `json:"max_backoff" default:"30000"`
	// BackoffFactor is the factor by which the backoff increases. Defaults to 2.0
	BackoffFactor float64 `json:"backoff_factor" default:"2.0"`
}

type FlakyEmbeddingsCaller struct {
	CallCount  int
	Embeddings []float32
}

type FlakyOpenAICaller struct {
	CallCount int
}

type MockEmbeddingsCaller struct {
	Embeddings []float32
}

type OpenaiCaller[T any] interface {
	Call(ctx context.Context, input string) (T, error)
}