def CURATED_FEW_SHOT_COT_PROMPT()

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

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

    def CURATED_FEW_SHOT_COT_PROMPT(self) -> _CoreSqlGeneratorPrompt:
        prompt_template = FewShotPrompts.TASK_SQL_GENERATION_CORE_V1_SPIDER_V1.partial(
            format_instructions=self.default_parser.get_format_instructions()
        )
        prompt_template.example_prompt = prompt_template.example_prompt.partial(  # type: ignore
            format_instructions=self.default_parser.get_format_instructions()
        )
        return _CoreSqlGeneratorPrompt(
            prompt_id="TASK_SQL_GENERATION_CORE_V1_SPIDER_V1",
            dialect_prompt_template_map={"default": prompt_template},
            parser=self.default_parser,
            post_processor=lambda x: x.get("query"),
        )