GuglielmoTor commited on
Commit
12bbfb7
·
verified ·
1 Parent(s): 8bb118c

Update insight_and_tasks/utils/pandasai_setup.py

Browse files
insight_and_tasks/utils/pandasai_setup.py CHANGED
@@ -33,7 +33,7 @@ def configure_pandasai(api_key: str, model_name: str = None):
33
 
34
  try:
35
  llm = LiteLLM(
36
- model=selected_model, # Use the selected model
37
  api_key=api_key
38
  )
39
 
 
33
 
34
  try:
35
  llm = LiteLLM(
36
+ model=DEFAULT_PANDASAI_MODEL , # Use the selected model
37
  api_key=api_key
38
  )
39