MockProvider 模拟 Provider
| 13 | |
| 14 | // MockProvider 模拟 Provider |
| 15 | type MockProvider struct { |
| 16 | name string |
| 17 | shouldFail bool |
| 18 | failCount int |
| 19 | currentFails int |
| 20 | completeFunc func(context.Context, []types.Message, *provider.StreamOptions) (*provider.CompleteResponse, error) |
| 21 | streamFunc func(context.Context, []types.Message, *provider.StreamOptions) (<-chan provider.StreamChunk, error) |
| 22 | capabilities provider.ProviderCapabilities |
| 23 | systemPrompt string |
| 24 | } |
| 25 | |
| 26 | func (m *MockProvider) Complete(ctx context.Context, messages []types.Message, opts *provider.StreamOptions) (*provider.CompleteResponse, error) { |
| 27 | if m.completeFunc != nil { |
nothing calls this directly
no outgoing calls
no test coverage detected