def CURATED_ZERO_SHOT_PROMPT()

in nl2sql/tasks/sql_generation/core.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

    def CURATED_ZERO_SHOT_PROMPT(self) -> _CoreSqlGeneratorPrompt:
        prompt_template = ZeroShotPrompts.TASK_SQL_GENERATION_CORE_V1.partial(
            format_instructions=self.default_parser.get_format_instructions()
        )
        return _CoreSqlGeneratorPrompt(
            prompt_id="TASK_SQL_GENERATION_CORE_V1",
            dialect_prompt_template_map={"default": prompt_template},
            parser=self.default_parser,
            post_processor=lambda x: x.get("query"),
        )