def _get_prompt_columns()

in src/package/dataplexutils/metadata/wizard.py [0:0]

• 45 lines of code
• 8 McCabe index (conditional complexity)

    def _get_prompt_columns(self):
        try:
            # System
            column_description_prompt = constants["PROMPTS"]["SYSTEM_PROMPT"]
            # Base
            if self._client_options._top_values_in_description==True:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["COLUMN_DESCRIPTION_PROMPT_BASE_WITH_EXAMPLES"]
                )
            else:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["COLUMN_DESCRIPTION_PROMPT_BASE"]
                )
                
            # Additional metadata information
            if self._client_options._use_profile:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["TABLE_DESCRIPTION_PROMPT_PROFILE"]
                )
            if self._client_options._use_data_quality:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["TABLE_DESCRIPTION_PROMPT_QUALITY"]
                )
            if self._client_options._use_lineage_tables:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["TABLE_DESCRIPTION_PROMPT_LINEAGE_TABLES"]
                )
            if self._client_options._use_lineage_processes:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["TABLE_DESCRIPTION_PROMPT_LINEAGE_PROCESSES"]
                )
            if self._client_options._use_human_comments:
                column_description_prompt = (
                    column_description_prompt
                    + constants["PROMPTS"]["COLUMN_DESCRIPTION_PROMPT_HUMAN_COMMENTS"]
                )
            # Output format
            column_description_prompt = (
                column_description_prompt + constants["PROMPTS"]["OUTPUT_FORMAT_PROMPT"]
            )
            return column_description_prompt
        except Exception as e:
            logger.error(f"Exception: {e}.")
            raise e