class_docstr: Unit test for Smart Data Lake Prompt Generation
class_docstr_tok: ["Unit","test","for","Smart","Data","Lake","Prompt","Generation"]
code: def config(self, llm):
      return {"llm": llm, "enable_cache": True}
  
file_path: pandas-ai/tests/pipelines/smart_datalake/test_prompt_generation.py
imports: import pandas
  from typing import Optional
  import pytest
  from pandasai.llm.fake import FakeLLM
  from pandasai.prompts.direct_sql_prompt import DirectSQLPrompt
  from pandasai.prompts.generate_python_code import GeneratePythonCodePrompt
  from pandasai.pipelines.smart_datalake_chat.prompt_generation import PromptGeneration
  from pandasai.pipelines.pipeline_context import PipelineContext
label_desc: Use image node_id 3 for calling the TestPromptGeneration obj's underlying member method code with example usage: obj.config(llm) and returns: dict