()
| 28 | } |
| 29 | |
| 30 | func (m *mockLLMClient) GetModelName() string { |
| 31 | args := m.Called() |
| 32 | return args.String(0) |
| 33 | } |
| 34 | |
| 35 | func (m *mockLLMClient) SendPrompt(ctx context.Context, prompt string, history []models.Message, maxTokens int) (string, error) { |
| 36 | args := m.Called(ctx, prompt, history, maxTokens) |