Update insight_and_tasks/utils/pandasai_setup.py

insight_and_tasks/utils/pandasai_setup.py

@@ -28,13 +28,13 @@ def configure_pandasai(api_key: str, model_name: str = None):
     os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "False"
     os.environ["GOOGLE_API_KEY"] = api_key
 
-    litellm._turn_on_debug()
+    LiteLLM._turn_on_debug()
 
     selected_model = model_name if model_name else DEFAULT_PANDASAI_MODEL
     
     try:
         llm = LiteLLM(
-            model="gemini-2.5-flash-preview-05-20", # Use the selected model
+            model=selected_model, # Use the selected model
             api_key=api_key
         )