fix(CI_Pipeline): resolve dependency issues causing build failures

examples/from_csv.py

@@ -15,8 +15,7 @@
 
 
 import pandasai as pai
-import pandasai_sql
 
 df = pai.load("sipa/users")
 response = df.chat("How many users in total?")
-print(response)
+print(response)

extensions/ee/connectors/bigquery/pyproject.toml

@@ -7,7 +7,7 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.12"
-pandasai = "^3.0.0"
+pandasai = "^3.0.0a0"
 pandasai-sql = "^0.1.0"
 sqlalchemy-bigquery = "^1.8.0"
 google-cloud-bigquery = "^3.27.0"

pandasai/__init__.py

@@ -16,9 +16,9 @@
 from pandasai.helpers.request import get_pandaai_session
 
 from .agent import Agent
+from .core.cache import Cache
 from .data_loader.loader import DatasetLoader
 from .dataframe.base import DataFrame
-from .core.cache import Cache
 from .smart_dataframe import SmartDataframe
 from .smart_datalake import SmartDatalake
 

pandasai/config.py

@@ -3,7 +3,7 @@
 from importlib.util import find_spec
 from typing import Any, Dict, List, Literal, Optional, Union
 
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
 import pandasai.llm as llm
 from pandasai.constants import DEFAULT_CHART_DIRECTORY

pandasai/helpers/request.py

@@ -5,9 +5,9 @@
 from urllib.parse import urljoin
 
 import requests
-from pandasai.helpers import load_dotenv
 
 from pandasai.exceptions import PandasAIApiCallError, PandasAIApiKeyError
+from pandasai.helpers import load_dotenv
 from pandasai.helpers.logger import Logger
 
 load_dotenv()

tests/unit_tests/helpers/test_dataframe_serializer.py

@@ -32,9 +32,6 @@ def test_convert_df_to_yml(self):
     fields:
     - name: name
       type: object
-      samples:
-      - en_name
-      - 中文_名称
 """,
             result,
         )