class_docstr: Code Execution Stage
code: def _get_error_prompt(self, e: Exception) -> AbstractPrompt:
      if isinstance(e, InvalidLLMOutputType):
          return CorrectOutputTypeErrorPrompt()
      else:
          return CorrectErrorPrompt()
  
file_path: pandas-ai/pandasai/pipelines/smart_datalake_chat/code_execution.py
imports: import logging
  import traceback
  from typing import Any, List
  from pandasai.exceptions import InvalidLLMOutputType
  from pandasai.prompts.base import AbstractPrompt
  from pandasai.prompts.correct_output_type_error_prompt import CorrectOutputTypeErrorPrompt
  from ...helpers.code_manager import CodeExecutionContext
  from ...helpers.logger import Logger
  from ..base_logic_unit import BaseLogicUnit
  from ..pipeline_context import PipelineContext
  from ...prompts.correct_error_prompt import CorrectErrorPrompt
label_desc: Use image node_id 2 for calling the CodeExecution obj's underlying member method code with example usage: obj._get_error_prompt(e) and returns: CorrectOutputTypeErrorPrompt, CorrectErrorPrompt