testutil

type MockEmbeddingModel struct {
	DoEmbedFunc   func(ctx context.Context, opts model.EmbedCallOptions) (*model.EmbedResult, error)
	DoEmbedCalls  []model.EmbedCallOptions
	EmbedResponse *model.EmbedResult
	// contains filtered or unexported fields
}

type MockEmbeddingModelOptions struct {
	ID                   string
	Provider             string
	Dimensions           int
	MaxEmbeddingsPerCall int
	DoEmbedFunc          func(ctx context.Context, opts model.EmbedCallOptions) (*model.EmbedResult, error)
	EmbedResponse        *model.EmbedResult
}

type MockImageModel struct {
	DoGenerateFunc   func(ctx context.Context, opts model.ImageCallOptions) (*model.ImageResult, error)
	DoGenerateCalls  []model.ImageCallOptions
	GenerateResponse *model.ImageResult
	// contains filtered or unexported fields
}

type MockImageModelOptions struct {
	ID               string
	Provider         string
	MaxImagesPerCall int
	DoGenerateFunc   func(ctx context.Context, opts model.ImageCallOptions) (*model.ImageResult, error)
	GenerateResponse *model.ImageResult
}

type MockLanguageModel struct {

	// DoStreamFunc is called for each Stream call. If set, takes precedence over StreamResponses.
	DoStreamFunc func(ctx context.Context, options *stream.CallOptions) (<-chan stream.Event, error)

	// StreamResponses is a list of responses to return in sequence.
	// Each call to Stream returns the next response.
	StreamResponses [][]stream.Event

	// DoStreamCalls records all calls to Stream for assertions.
	DoStreamCalls []*stream.CallOptions
	// contains filtered or unexported fields
}

type MockLanguageModelOptions struct {
	// ID is the model identifier. Defaults to "mock-model-id".
	ID string

	// Provider is the provider identifier. Defaults to "mock-provider".
	Provider string

	// DoStreamFunc is called for each Stream call.
	// If set, takes precedence over StreamResponse/StreamResponses.
	DoStreamFunc func(ctx context.Context, options *stream.CallOptions) (<-chan stream.Event, error)

	// StreamResponse is a single response to return for all Stream calls.
	// Use StreamResponses for multiple sequential responses.
	StreamResponse []stream.Event

	// StreamResponses is a list of responses to return in sequence.
	// Takes precedence over StreamResponse.
	StreamResponses [][]stream.Event
}

type MockSpeechModel struct {
	DoGenerateFunc   func(ctx context.Context, opts model.SpeechCallOptions) (*model.SpeechResult, error)
	DoGenerateCalls  []model.SpeechCallOptions
	GenerateResponse *model.SpeechResult
	// contains filtered or unexported fields
}

type MockSpeechModelOptions struct {
	ID               string
	Provider         string
	DoGenerateFunc   func(ctx context.Context, opts model.SpeechCallOptions) (*model.SpeechResult, error)
	GenerateResponse *model.SpeechResult
}

type MockTranscriptionModel struct {
	DoTranscribeFunc   func(ctx context.Context, opts model.TranscribeCallOptions) (*model.TranscriptionResult, error)
	DoTranscribeCalls  []model.TranscribeCallOptions
	TranscribeResponse *model.TranscriptionResult
	// contains filtered or unexported fields
}

type MockTranscriptionModelOptions struct {
	ID                 string
	Provider           string
	DoTranscribeFunc   func(ctx context.Context, opts model.TranscribeCallOptions) (*model.TranscriptionResult, error)
	TranscribeResponse *model.TranscriptionResult
}