gitextract_46s0phol/ ├── .github/ │ ├── ISSUE_TEMPLATE/ │ │ ├── bug_report.yml │ │ ├── config.yml │ │ └── feature_request.yml │ ├── PULL_REQUEST_TEMPLATE.md │ └── workflows/ │ ├── cd.yml │ ├── ci-core.yml │ └── ci-extensions.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .sourcery.yaml ├── CITATION.cff ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── docker-compose.yml ├── docs/ │ ├── mint.json │ ├── v2/ │ │ ├── advanced-security-agent.mdx │ │ ├── cache.mdx │ │ ├── connectors.mdx │ │ ├── contributing.mdx │ │ ├── custom-head.mdx │ │ ├── custom-response.mdx │ │ ├── custom-whitelisted-dependencies.mdx │ │ ├── determinism.mdx │ │ ├── examples.mdx │ │ ├── fields-description.mdx │ │ ├── intro.mdx │ │ ├── judge-agent.mdx │ │ ├── library.mdx │ │ ├── license.mdx │ │ ├── llms.mdx │ │ ├── pipelines/ │ │ │ └── pipelines.mdx │ │ ├── platform.mdx │ │ ├── semantic-agent.mdx │ │ ├── skills.mdx │ │ └── train.mdx │ └── v3/ │ ├── agent.mdx │ ├── chat-and-output.mdx │ ├── contributing.mdx │ ├── enterprise-features.mdx │ ├── getting-started.mdx │ ├── introduction.mdx │ ├── large-language-models.mdx │ ├── license.mdx │ ├── migration-backwards-compatibility.mdx │ ├── migration-guide.mdx │ ├── migration-troubleshooting.mdx │ ├── overview-nl.mdx │ ├── privacy-security.mdx │ ├── semantic-layer/ │ │ ├── data-ingestion.mdx │ │ ├── new.mdx │ │ ├── semantic-layer.mdx │ │ ├── transformations.mdx │ │ └── views.mdx │ └── skills.mdx ├── ee/ │ └── LICENSE ├── examples/ │ ├── data/ │ │ ├── heart.csv │ │ └── loans_payments.csv │ ├── docker_sandbox.ipynb │ ├── quickstart.ipynb │ └── semantic_layer_csv.ipynb ├── extensions/ │ ├── connectors/ │ │ ├── sql/ │ │ │ ├── README.md │ │ │ ├── pandasai_sql/ │ │ │ │ └── __init__.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_sql.py │ │ └── yfinance/ │ │ ├── README.md │ │ ├── pandasai_yfinance/ │ │ │ └── __init__.py │ │ ├── pyproject.toml │ │ └── tests/ │ │ └── test_yahoo_finance.py │ ├── ee/ │ │ ├── LICENSE │ │ ├── connectors/ │ │ │ ├── bigquery/ │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── pandasai_bigquery/ │ │ │ │ │ └── __init__.py │ │ │ │ ├── pyproject.toml │ │ │ │ └── tests/ │ │ │ │ └── test_bigquery.py │ │ │ ├── databricks/ │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── pandasai_databricks/ │ │ │ │ │ └── __init__.py │ │ │ │ ├── pyproject.toml │ │ │ │ └── tests/ │ │ │ │ └── test_databricks.py │ │ │ ├── oracle/ │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── pandasai_oracle/ │ │ │ │ │ └── __init__.py │ │ │ │ ├── pyproject.toml │ │ │ │ └── tests/ │ │ │ │ └── test_oracle.py │ │ │ └── snowflake/ │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── pandasai_snowflake/ │ │ │ │ └── __init__.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_snowflake.py │ │ └── vectorstores/ │ │ ├── chromadb/ │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── pandasai_chromadb/ │ │ │ │ ├── __init__.py │ │ │ │ └── chroma.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_chromadb.py │ │ ├── lancedb/ │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── pandasai_lancedb/ │ │ │ │ ├── __init__.py │ │ │ │ └── lancedb.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_lancedb.py │ │ ├── milvus/ │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── pandasai_milvus/ │ │ │ │ ├── __init__.py │ │ │ │ └── milvus.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_milvus.py │ │ ├── pinecone/ │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── pandasai_pinecone/ │ │ │ │ ├── __init__.py │ │ │ │ └── pinecone.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_pinecone.py │ │ └── qdrant/ │ │ ├── LICENSE │ │ ├── README.md │ │ ├── pandasai_qdrant/ │ │ │ ├── __init__.py │ │ │ └── qdrant.py │ │ ├── pyproject.toml │ │ └── tests/ │ │ └── test_qdrant.py │ ├── llms/ │ │ ├── litellm/ │ │ │ ├── README.md │ │ │ ├── pandasai_litellm/ │ │ │ │ ├── __init__.py │ │ │ │ └── litellm.py │ │ │ ├── pyproject.toml │ │ │ └── tests/ │ │ │ └── test_litellm.py │ │ └── openai/ │ │ ├── README.md │ │ ├── pandasai_openai/ │ │ │ ├── __init__.py │ │ │ ├── azure_openai.py │ │ │ ├── base.py │ │ │ └── openai.py │ │ ├── pyproject.toml │ │ └── tests/ │ │ ├── test_azure_openai.py │ │ └── test_openai.py │ └── sandbox/ │ └── docker/ │ ├── README.md │ ├── pandasai_docker/ │ │ ├── Dockerfile │ │ ├── __init__.py │ │ ├── docker_sandbox.py │ │ └── serializer.py │ ├── pyproject.toml │ └── tests/ │ ├── test_sandbox.py │ └── test_serializer.py ├── ignore-words.txt ├── pandasai/ │ ├── __init__.py │ ├── __version__.py │ ├── agent/ │ │ ├── __init__.py │ │ ├── base.py │ │ └── state.py │ ├── cli/ │ │ ├── __init__.py │ │ └── main.py │ ├── config.py │ ├── constants.py │ ├── core/ │ │ ├── code_execution/ │ │ │ ├── __init__.py │ │ │ ├── code_executor.py │ │ │ └── environment.py │ │ ├── code_generation/ │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── code_cleaning.py │ │ │ └── code_validation.py │ │ ├── prompts/ │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── correct_execute_sql_query_usage_error_prompt.py │ │ │ ├── correct_output_type_error_prompt.py │ │ │ ├── generate_python_code_with_sql.py │ │ │ ├── generate_system_message.py │ │ │ └── templates/ │ │ │ ├── correct_execute_sql_query_usage_error_prompt.tmpl │ │ │ ├── correct_output_type_error_prompt.tmpl │ │ │ ├── generate_python_code_with_sql.tmpl │ │ │ ├── generate_system_message.tmpl │ │ │ └── shared/ │ │ │ ├── dataframe.tmpl │ │ │ ├── output_type_template.tmpl │ │ │ ├── sql_functions.tmpl │ │ │ └── vectordb_docs.tmpl │ │ ├── response/ │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── chart.py │ │ │ ├── dataframe.py │ │ │ ├── error.py │ │ │ ├── number.py │ │ │ ├── parser.py │ │ │ └── string.py │ │ └── user_query.py │ ├── data_loader/ │ │ ├── duck_db_connection_manager.py │ │ ├── loader.py │ │ ├── local_loader.py │ │ ├── semantic_layer_schema.py │ │ ├── sql_loader.py │ │ └── view_loader.py │ ├── dataframe/ │ │ ├── __init__.py │ │ ├── base.py │ │ └── virtual_dataframe.py │ ├── ee/ │ │ ├── LICENSE │ │ └── skills/ │ │ ├── __init__.py │ │ └── manager.py │ ├── exceptions.py │ ├── helpers/ │ │ ├── __init__.py │ │ ├── dataframe_serializer.py │ │ ├── env.py │ │ ├── filemanager.py │ │ ├── folder.py │ │ ├── json_encoder.py │ │ ├── logger.py │ │ ├── memory.py │ │ ├── path.py │ │ ├── session.py │ │ ├── sql_sanitizer.py │ │ └── telemetry.py │ ├── llm/ │ │ ├── __init__.py │ │ ├── base.py │ │ └── fake.py │ ├── query_builders/ │ │ ├── __init__.py │ │ ├── base_query_builder.py │ │ ├── local_query_builder.py │ │ ├── paginator.py │ │ ├── sql_parser.py │ │ ├── sql_query_builder.py │ │ ├── sql_transformation_manager.py │ │ └── view_query_builder.py │ ├── sandbox/ │ │ ├── __init__.py │ │ └── sandbox.py │ ├── smart_dataframe/ │ │ └── __init__.py │ ├── smart_datalake/ │ │ └── __init__.py │ └── vectorstores/ │ ├── __init__.py │ └── vectorstore.py ├── poetry.toml ├── pyproject.toml ├── pytest.ini └── tests/ ├── __init__.py ├── examples/ │ └── data/ │ ├── sample_multi_sheet_data.xlsx │ └── sample_single_sheet_data.xlsx ├── integration_tests/ │ ├── __init__.py │ ├── conftest.py │ ├── local_view/ │ │ ├── __init__.py │ │ ├── test_local_view.py │ │ ├── test_local_view_grouped.py │ │ └── test_local_view_transformed.py │ ├── parquet/ │ │ ├── __init__.py │ │ ├── test_parquet.py │ │ ├── test_parquet_grouped.py │ │ └── test_parquet_transformed.py │ ├── sql/ │ │ ├── __init__.py │ │ └── test_sql.py │ └── sql_view/ │ ├── __init__.py │ └── test_sql_view.py └── unit_tests/ ├── __init__.py ├── agent/ │ ├── .ipynb_checkpoints/ │ │ └── test_agent_llm_judge-checkpoint.py │ ├── test_agent.py │ ├── test_agent_chat.py │ └── test_agent_llm_judge.py ├── conftest.py ├── core/ │ ├── code_execution/ │ │ ├── test_code_execution.py │ │ └── test_environment.py │ ├── code_generation/ │ │ ├── test_code_cleaning.py │ │ └── test_code_validation.py │ └── prompts/ │ ├── test_base.py │ ├── test_correct_execute_sql_query_usage_error_prompt.py │ ├── test_correct_output_type_error_prompt.py │ ├── test_generate_python_code_with_sql_prompt.py │ └── test_prompts.py ├── data_loader/ │ ├── test_duckdbmanager.py │ ├── test_loader.py │ ├── test_sql_loader.py │ ├── test_transformation_schema.py │ └── test_view_loader.py ├── dataframe/ │ ├── test_dataframe.py │ ├── test_pull.py │ └── test_semantic_layer_schema.py ├── helpers/ │ ├── __init__.py │ ├── test_dataframe_serializer.py │ ├── test_folder.py │ ├── test_json_encoder.py │ ├── test_logger.py │ ├── test_optional_dependency.py │ ├── test_responses.py │ ├── test_session.py │ └── test_sql_sanitizer.py ├── llms/ │ ├── __init_.py │ └── test_base_llm.py ├── prompts/ │ ├── __init_.py │ └── test_sql_prompt.py ├── query_builders/ │ ├── __init__.py │ ├── test_group_by.py │ ├── test_paginator.py │ ├── test_query_builder.py │ ├── test_sql_parser.py │ ├── test_sql_transformation_manager.py │ └── test_view_query_builder.py ├── response/ │ ├── test_chart_response.py │ ├── test_dataframe_response.py │ ├── test_error_response.py │ ├── test_number_response.py │ └── test_string_response.py ├── sandbox/ │ └── test_sandbox.py ├── skills/ │ ├── __init__.py │ ├── test_shared_template.py │ ├── test_skill.py │ ├── test_skill_decorator.py │ ├── test_skills_integration.py │ └── test_skills_manager.py ├── smart_dataframe/ │ └── test_smart_dataframe.py ├── smart_datalake/ │ └── test_smart_datalake.py ├── test_api_key_manager.py ├── test_cli.py ├── test_config.py ├── test_memory.py ├── test_pandasai_init.py └── test_pandasai_read_excel.py