| 34 | should_timeout_ = timeout; |
| 35 | } |
| 36 | |
| 37 | void ControllableOpenAIClient::reset() { |
| 38 | predefined_response_ = ResponseBuilder::buildSuccessResponse(); |
| 39 | predefined_status_code_ = 200; |
| 40 | should_fail_ = false; |
| 41 | should_timeout_ = false; |
| 42 | call_count_ = 0; |
| 43 | } |
| 44 | |
| 45 | GenerateResult ControllableOpenAIClient::generate_text( |