def generate_internal()

in evalbench/generators/models/gemini.py [0:0]

• 15 lines of code
• 4 McCabe index (conditional complexity)

    def generate_internal(self, prompt):
        logger = logging.getLogger(__name__)
        try:
            response = self.model.generate_content(
                self.base_prompt + prompt,
                generation_config=self.generation_config,
            )
            if isinstance(response, GenerationResponse):
                r = sanitize_sql(response.text)
            return r
        except ResourceExhausted as e:
            raise ResourceExhaustedError(e)
        except Exception as e:
            logger.exception("Unhandled exception during generate_content")
            raise