| import pandas as pd | |
| from pandasai.llm.openai import OpenAI | |
| from pandasai.llm.starcoder import Starcoder | |
| from pandasai.llm.google_palm import GooglePalm | |
| file_format = { | |
| "csv": pd.read_csv, | |
| "xls": pd.read_excel, | |
| "xlsx": pd.read_excel, | |
| "xlsm": pd.read_excel, | |
| "xlsb": pd.read_excel, | |
| "json": pd.read_json, | |
| "html": pd.read_html, | |
| "sql": pd.read_sql, | |
| "feather": pd.read_feather, | |
| "parquet": pd.read_parquet, | |
| "dta": pd.read_stata, | |
| "sas7bdat": pd.read_sas, | |
| "h5": pd.read_hdf, | |
| "hdf5": pd.read_hdf, | |
| "pkl": pd.read_pickle, | |
| "pickle": pd.read_pickle, | |
| "gbq": pd.read_gbq, | |
| "orc": pd.read_orc, | |
| "xpt": pd.read_sas, | |
| "sav": pd.read_spss, | |
| "gz": pd.read_csv, | |
| "zip": pd.read_csv, | |
| "bz2": pd.read_csv, | |
| "xz": pd.read_csv, | |
| "txt": pd.read_csv, | |
| "xml": pd.read_xml, | |
| } | |
| models = { | |
| "OpenAI": OpenAI, | |
| "Starcoder": Starcoder, | |
| "Google Palm": GooglePalm | |
| } |