def CURATED_ZERO_SHOT_PROMPT()

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

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

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