def CURATED_ZERO_SHOT_PROMPT()

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

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

    def CURATED_ZERO_SHOT_PROMPT(self) -> _CoreColumnSelectorPrompt:
        prompt_template = ZeroShotPrompts.TASK_COLUMN_SELECTION_CORE_V1.partial(
            format_instructions=self.default_parser.get_format_instructions()
        )
        return _CoreColumnSelectorPrompt(
            prompt_id="TASK_COLUMN_SELECTION_CORE_V1",
            prompt_template=prompt_template,
            parser=self.default_parser,
            post_processor=lambda x: [i.strip() for i in x["columns"].split(",")]
            if ((x) and (x.get("columns")))
            else [],
        )