class_docstr: Code Prompt Generation Stage
code: def _get_chat_prompt(
      self, context: PipelineContext
  ) -> [str, FileBasedPrompt]:
      key = (
          "direct_sql_prompt"
          if context.config.direct_sql
          else "generate_python_code"
      )
      return (
          key,
          (
              DirectSQLPrompt(tables=context.dfs)
              if context.config.direct_sql
              else GeneratePythonCodePrompt()
          ),
      )
  
file_path: pandas-ai/pandasai/pipelines/smart_datalake_chat/prompt_generation.py
imports: from typing import Any
  from ..base_logic_unit import BaseLogicUnit
  from ..pipeline_context import PipelineContext
  from ...prompts.generate_python_code import GeneratePythonCodePrompt
  from ...prompts.direct_sql_prompt import DirectSQLPrompt
  from ...prompts.file_based_prompt import FileBasedPrompt
label_desc: Use image node_id 1 for calling the PromptGeneration obj's underlying member method code with example usage: obj._get_chat_prompt(context) and returns: key,  unknown