Spaces:
Sleeping
Sleeping
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=
|
| 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 |
|