mock

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

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

type MockAudioURL struct {
	URL string `yaml:"url"` // URL to the audio file (can be mock://, http://, https://, data:, or file path)
}

type MockConfig struct {
	// Default response if no specific match is found
	DefaultResponse string `yaml:"defaultResponse"`

	// Scenario-specific responses keyed by scenario ID
	Scenarios map[string]ScenarioMockConfig `yaml:"scenarios,omitempty"`
}

type MockContentPart struct {
	Type     string                 `yaml:"type"`                // "text", "image", "audio", or "video"
	Text     string                 `yaml:"text,omitempty"`      // Text content (for type="text")
	ImageURL *MockImageURL          `yaml:"image_url,omitempty"` // Image URL (for type="image")
	AudioURL *MockAudioURL          `yaml:"audio_url,omitempty"` // Audio URL (for type="audio")
	VideoURL *MockVideoURL          `yaml:"video_url,omitempty"` // Video URL (for type="video")
	Metadata map[string]interface{} `yaml:"metadata,omitempty"`  // Additional metadata
}

type MockImageURL struct {
	URL    string  `yaml:"url"`              // URL to the image (can be mock://, http://, https://, data:, or file path)
	Detail *string `yaml:"detail,omitempty"` // Detail level: "low", "high", "auto"
}

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

type MockResponseParams struct {
	ScenarioID string // Optional: ID of the scenario being executed
	TurnNumber int    // Optional: Turn number in a multi-turn conversation
	ProviderID string // Optional: ID of the provider being mocked
	ModelName  string // Optional: Model name being mocked
}

type MockResponseRepository interface {
	// GetResponse retrieves a mock response for the given context.
	// Parameters can include scenario ID, turn number, provider ID, etc.
	// Returns the response text and any error encountered.
	GetResponse(ctx context.Context, params MockResponseParams) (string, error)

	// GetTurn retrieves a mock turn response that may include tool calls.
	// This extends GetResponse to support structured turn data with tool call simulation.
	GetTurn(ctx context.Context, params MockResponseParams) (*MockTurn, error)
}

type MockToolCall struct {
	Name      string                 `yaml:"name"`      // Name of the tool to call
	Arguments map[string]interface{} `yaml:"arguments"` // Arguments to pass to the tool
}

type MockToolError struct {
	Type    string `yaml:"type"`    // Error type/category
	Message string `yaml:"message"` // Error message
}

type MockToolProvider struct {
	*MockProvider
}

type MockToolResponse struct {
	CallArgs map[string]interface{} `yaml:"call_args"`        // Match these arguments
	Result   interface{}            `yaml:"result,omitempty"` // Return this result
	Error    *MockToolError         `yaml:"error,omitempty"`  // Or return this error
}

type MockTurn struct {
	Type      string            `yaml:"type"`                 // "text", "tool_calls", or "multimodal"
	Content   string            `yaml:"content,omitempty"`    // Text content for the response
	Parts     []MockContentPart `yaml:"parts,omitempty"`      // Multimodal content parts (text, image, audio, video)
	ToolCalls []MockToolCall    `yaml:"tool_calls,omitempty"` // Tool calls to simulate
}

type MockVideoURL struct {
	URL string `yaml:"url"` // URL to the video file (can be mock://, http://, https://, data:, or file path)
}

type ScenarioMockConfig struct {
	// Default response for this scenario (overrides global default)
	DefaultResponse string `yaml:"defaultResponse,omitempty"`

	// Turn-specific responses keyed by turn number (1-indexed)
	// Supports both simple string responses (backward compatibility) and structured MockTurn responses
	Turns map[int]interface{} `yaml:"turns,omitempty"`

	// Tool execution responses for repository-backed tool mocking
	ToolResponses map[string][]MockToolResponse `yaml:"tool_responses,omitempty"`
}