File size: 934 Bytes
c7e9e1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
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
} |