testutil

type CapturingMockProvider struct {
	MockProvider

	LastPrompt string
	// contains filtered or unexported fields
}

type MockProvider struct {
	ProviderName string // provider identifier, e.g. "openai"
	Content      string // canned response; empty = "mock response from " + ProviderName
	Err          error  // if set, Generate returns this error
}

type OpenAICompatibleResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Model   string `json:"model"`
	Choices []struct {
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type ToolCallMockProvider struct {
	Responses           []*llm.Response // sequence of responses; call N gets Responses[N] or last if N >= len
	CallCount           int             // incremented on each Generate call
	ReceivedMessages    [][]llm.Message
	EstimateCostPerCall float64 // cost returned by EstimateCost (default 0.001)
	ErrOnCall           int     // 1-based; when CallCount == ErrOnCall, Generate returns (nil, Err). 0 = never
	Err                 error   // error to return when ErrOnCall is hit
	// contains filtered or unexported fields
}