SYMBOL INDEX (340 symbols across 55 files)

FILE: premsql/agents/base.py
  class WorkerBase (line 24) | class WorkerBase(ABC):
    method run (line 26) | def run(self):
  class AnalysisWorkerBase (line 30) | class AnalysisWorkerBase(ABC):
    method run (line 32) | def run(
  class ChartPlotWorkerBase (line 38) | class ChartPlotWorkerBase(ABC):
    method run (line 40) | def run(
  class RouterWorkerBase (line 46) | class RouterWorkerBase(ABC):
    method run (line 48) | def run(
  class Text2SQLWorkerBase (line 54) | class Text2SQLWorkerBase(ABC):
    method __init__ (line 55) | def __init__(
    method run (line 73) | def run(self, question: str, **kwargs) -> Text2SQLWorkerOutput:
    method initialize_database (line 76) | def initialize_database(
  class AgentBase (line 101) | class AgentBase(ABC):
    method __init__ (line 102) | def __init__(
    method run (line 117) | def run(
    method convert_exit_output_to_agent_output (line 126) | def convert_exit_output_to_agent_output(
    method __call__ (line 156) | def __call__(

FILE: premsql/agents/baseline/main.py
  class BaseLineAgent (line 20) | class BaseLineAgent(AgentBase):
    method __init__ (line 21) | def __init__(
    method run (line 57) | def run(
    method _execute_worker (line 104) | def _execute_worker(
    method _create_exit_worker_output (line 138) | def _create_exit_worker_output(
    method _handle_followup (line 173) | def _handle_followup(self, prev_output: ExitWorkerOutput):
    method _handle_followup_route (line 180) | def _handle_followup_route(self, question: str) -> ExitWorkerOutput:

FILE: premsql/agents/baseline/workers/analyser.py
  class BaseLineAnalyserWorker (line 28) | class BaseLineAnalyserWorker(AnalysisWorkerBase):
    method __init__ (line 29) | def __init__(self, generator: Text2SQLGeneratorBase) -> None:
    method run_chunkwise_analysis (line 32) | def run_chunkwise_analysis(
    method analyse (line 110) | def analyse(
    method run (line 155) | def run(

FILE: premsql/agents/baseline/workers/followup.py
  class BaseLineFollowupWorker (line 14) | class BaseLineFollowupWorker(WorkerBase):
    method __init__ (line 15) | def __init__(self, generator: Text2SQLGeneratorBase) -> None:
    method run (line 18) | def run(

FILE: premsql/agents/baseline/workers/plotter.py
  class BaseLinePlotWorker (line 15) | class BaseLinePlotWorker(ChartPlotWorkerBase):
    method __init__ (line 16) | def __init__(
    method run (line 21) | def run(

FILE: premsql/agents/baseline/workers/text2sql.py
  class BaseLineText2SQLWorker (line 20) | class BaseLineText2SQLWorker(Text2SQLWorkerBase):
    method __init__ (line 21) | def __init__(
    method show_dataframe (line 44) | def show_dataframe(output: Text2SQLWorkerOutput):
    method filer_tables_from_schema (line 55) | def filer_tables_from_schema(
    method _create_prompt (line 76) | def _create_prompt(
    method run (line 142) | def run(
    method do_correction (line 230) | def do_correction(

FILE: premsql/agents/memory.py
  class AgentInteractionMemory (line 15) | class AgentInteractionMemory:
    method __init__ (line 16) | def __init__(self, session_name: str, db_path: Optional[str] = None):
    method list_sessions (line 28) | def list_sessions(self) -> List[str]:
    method create_table_if_not_exists (line 34) | def create_table_if_not_exists(self):
    method get (line 68) | def get(
    method get_latest_message_id (line 86) | def get_latest_message_id(self) -> Optional[int]:
    method generate_messages_from_session (line 93) | def generate_messages_from_session(
    method get_by_message_id (line 105) | def get_by_message_id(self, message_id: int) -> Optional[dict]:
    method push (line 114) | def push(self, output: ExitWorkerOutput):
    method delete_table (line 133) | def delete_table(self):
    method _row_to_exit_worker_output (line 144) | def _row_to_exit_worker_output(self, row) -> ExitWorkerOutput:
    method _exit_worker_output_to_tuple (line 175) | def _exit_worker_output_to_tuple(self, output: ExitWorkerOutput) -> tu...
    method _parse_json (line 201) | def _parse_json(self, json_str):
    method _serialize_json (line 212) | def _serialize_json(self, obj):
    method clear (line 223) | def clear(self):
    method close (line 228) | def close(self):
    method __del__ (line 231) | def __del__(self):
    method delete_table (line 234) | def delete_table(self):
    method get_latest_dataframe (line 245) | def get_latest_dataframe(

FILE: premsql/agents/models.py
  class BaseWorkerOutput (line 12) | class BaseWorkerOutput(BaseModel):
  class Text2SQLWorkerOutput (line 22) | class Text2SQLWorkerOutput(BaseWorkerOutput):
    method show_output_dataframe (line 31) | def show_output_dataframe(self) -> pd.DataFrame:
  class AnalyserWorkerOutput (line 39) | class AnalyserWorkerOutput(BaseWorkerOutput):
  class ChartPlotWorkerOutput (line 47) | class ChartPlotWorkerOutput(BaseWorkerOutput):
  class RouterWorkerOutput (line 57) | class RouterWorkerOutput(BaseWorkerOutput):
  class FollowupWorkerOutput (line 66) | class FollowupWorkerOutput(BaseWorkerOutput):
  class ExitWorkerOutput (line 74) | class ExitWorkerOutput(BaseModel):
    method show_output_dataframe (line 115) | def show_output_dataframe(
  class AgentOutput (line 131) | class AgentOutput(BaseModel):
    method show_output_dataframe (line 150) | def show_output_dataframe(

FILE: premsql/agents/router.py
  class SimpleRouterWorker (line 12) | class SimpleRouterWorker(RouterWorkerBase):
    method run (line 13) | def run(

FILE: premsql/agents/tools/plot/base.py
  class BasePlotTool (line 9) | class BasePlotTool(ABC):
    method run (line 11) | def run(self, data: pd.DataFrame, plot_config: dict):
    method convert_plot_to_image (line 15) | def convert_plot_to_image(self, fig):
    method convert_image_to_base64 (line 18) | def convert_image_to_base64(self, image: Image.Image) -> str:
    method save_image (line 23) | def save_image(self, image: Image.Image, file_path: str, format: str =...
    method plot_from_base64 (line 26) | def plot_from_base64(self, output_base64: str):

FILE: premsql/agents/tools/plot/matplotlib_tool.py
  class SimpleMatplotlibTool (line 16) | class SimpleMatplotlibTool(BasePlotTool):
    method __init__ (line 17) | def __init__(self):
    method run (line 28) | def run(self, data: pd.DataFrame, plot_config: Dict[str, str]) -> Figure:
    method _validate_config (line 49) | def _validate_config(self, df: pd.DataFrame, plot_config: Dict[str, st...
    method _area_plot (line 66) | def _area_plot(self, df: pd.DataFrame, x: str, y: str, ax: Axes) -> None:
    method _bar_plot (line 69) | def _bar_plot(self, df: pd.DataFrame, x: str, y: str, ax: Axes) -> None:
    method _scatter_plot (line 72) | def _scatter_plot(self, df: pd.DataFrame, x: str, y: str, ax: Axes) ->...
    method _histogram_plot (line 75) | def _histogram_plot(self, df: pd.DataFrame, x: str, y: str, ax: Axes) ...
    method _line_plot (line 78) | def _line_plot(self, df: pd.DataFrame, x: str, y: str, ax: Axes) -> None:
    method convert_plot_to_image (line 81) | def convert_plot_to_image(self, fig: Figure) -> Image.Image:

FILE: premsql/agents/utils.py
  function convert_df_to_dict (line 12) | def convert_df_to_dict(df: pd.DataFrame):
  function execute_and_render_result (line 16) | def execute_and_render_result(
  function _render_error (line 26) | def _render_error(error: str, sql: str, using: str) -> Dict[str, Any]:
  function _render_data (line 36) | def _render_data(result, sql: str, using: str) -> Dict[str, Any]:
  function convert_exit_output_to_agent_output (line 57) | def convert_exit_output_to_agent_output(exit_output: ExitWorkerOutput) -...

FILE: premsql/cli.py
  function cli (line 10) | def cli():
  function launch (line 15) | def launch():
  function launch_all (line 21) | def launch_all():
  function launch_api (line 60) | def launch_api():
  function stop (line 88) | def stop():

FILE: premsql/datasets/__init__.py
  class Text2SQLDataset (line 12) | class Text2SQLDataset:
    method __init__ (line 13) | def __init__(
    method raw_dataset (line 40) | def raw_dataset(self):
    method filter_availables (line 44) | def filter_availables(self):
    method setup_dataset (line 47) | def setup_dataset(

FILE: premsql/datasets/base.py
  class Text2SQLBaseInstance (line 32) | class Text2SQLBaseInstance:
    method __init__ (line 33) | def __init__(self, dataset: list[dict]) -> None:
    method __repr__ (line 42) | def __repr__(self) -> str:
    method __len__ (line 45) | def __len__(self) -> int:
    method __getitem__ (line 48) | def __getitem__(self, idx: int) -> dict:
    method schema_prompt (line 51) | def schema_prompt(self, db_path: str) -> str:
    method additional_prompt (line 76) | def additional_prompt(self, prompt: Optional[str] = None):
    method add_few_shot_examples (line 79) | def add_few_shot_examples(self, db_id: str, k: int = 3) -> str:
    method apply_prompt (line 86) | def apply_prompt(
  class SupervisedDatasetForTraining (line 114) | class SupervisedDatasetForTraining(torch.utils.data.Dataset):
    method load_from_pth (line 116) | def load_from_pth(cls, dataset_path: Union[str, Path]):
    method __init__ (line 130) | def __init__(
    method preprocess (line 190) | def preprocess(self, sources: Sequence[str], targets: Sequence[str]):
    method __len__ (line 203) | def __len__(self):
    method __getitem__ (line 206) | def __getitem__(self, idx: int):
    method save_tokenized_dataset (line 216) | def save_tokenized_dataset(self, path_to_save: Union[str, Path]):
  class Text2SQLBaseDataset (line 221) | class Text2SQLBaseDataset(ABC):
    method __init__ (line 222) | def __init__(
    method raw_dataset (line 240) | def raw_dataset(self):
    method filter_availables (line 244) | def filter_availables(self):
    method setup_dataset (line 248) | def setup_dataset(
    method __len__ (line 281) | def __len__(self):
    method __getitem__ (line 284) | def __getitem__(self, idx):
  class StandardDataset (line 288) | class StandardDataset(Text2SQLBaseDataset):
    method __init__ (line 289) | def __init__(
    method setup_dataset (line 305) | def setup_dataset(

FILE: premsql/datasets/collator.py
  class DataCollatorForSupervisedDataset (line 15) | class DataCollatorForSupervisedDataset:
    method __call__ (line 18) | def __call__(self, instances: Sequence[dict]) -> dict[str, torch.Tensor]:

FILE: premsql/datasets/error_dataset.py
  class ErrorDatasetInstance (line 20) | class ErrorDatasetInstance(Text2SQLBaseInstance):
    method __init__ (line 22) | def __init__(self, dataset: list[dict]) -> None:
    method apply_prompt (line 25) | def apply_prompt(self, prompt_template: Optional[str] = ERROR_HANDLING...
  class ErrorDatasetGenerator (line 51) | class ErrorDatasetGenerator:
    method from_existing (line 53) | def from_existing(
    method __init__ (line 77) | def __init__(
    method generate_and_save (line 87) | def generate_and_save(

FILE: premsql/datasets/real/bird.py
  class BirdDataset (line 12) | class BirdDataset(Text2SQLBaseDataset):
    method __init__ (line 13) | def __init__(
    method setup_dataset (line 52) | def setup_dataset(

FILE: premsql/datasets/real/domains.py
  class DomainsDataset (line 12) | class DomainsDataset(Text2SQLBaseDataset):
    method __init__ (line 13) | def __init__(
    method setup_dataset (line 52) | def setup_dataset(

FILE: premsql/datasets/real/spider.py
  class SpiderUnifiedDataset (line 12) | class SpiderUnifiedDataset(Text2SQLBaseDataset):
    method __init__ (line 13) | def __init__(
    method setup_dataset (line 52) | def setup_dataset(

FILE: premsql/datasets/synthetic/gretel.py
  class GretelAIInstance (line 19) | class GretelAIInstance(Text2SQLBaseInstance):
    method __init__ (line 20) | def __init__(self, dataset: list[dict]) -> None:
    method apply_prompt (line 23) | def apply_prompt(
  class GretelAIDataset (line 47) | class GretelAIDataset(Text2SQLBaseDataset):
    method __init__ (line 48) | def __init__(
    method setup_dataset (line 86) | def setup_dataset(

FILE: premsql/evaluator/base.py
  class Text2SQLEvaluator (line 13) | class Text2SQLEvaluator:
    method __init__ (line 14) | def __init__(
    method _execute_model (line 20) | def _execute_model(
    method execute (line 66) | def execute(
    method compute_metric (line 130) | def compute_metric(self, results: list[dict], metric_name: str) -> float:

FILE: premsql/executors/base.py
  class BaseExecutor (line 6) | class BaseExecutor(ABC):
    method execute_sql (line 8) | def execute_sql(self, sql: str, dsn_or_db_path: str) -> dict:
    method match_sqls (line 11) | def match_sqls(
    method clean_abnormal (line 28) | def clean_abnormal(self, input: list[float]) -> list[float]:
    method iterated_execution (line 34) | def iterated_execution(

FILE: premsql/executors/from_langchain.py
  class ExecutorUsingLangChain (line 10) | class ExecutorUsingLangChain(BaseExecutor):
    method execute_sql (line 12) | def execute_sql(self, sql: str, dsn_or_db_path: Union[str, SQLDatabase...

FILE: premsql/executors/from_sqlite.py
  class OptimizedSQLiteExecutor (line 10) | class OptimizedSQLiteExecutor(BaseExecutor):
    method __init__ (line 11) | def __init__(self, timeout: float = 1000.0) -> None:
    method get_connection (line 16) | def get_connection(self, db_path: str) -> Generator[sqlite3.Connection...
    method execute_sql (line 31) | def execute_sql(self, sql: str, dsn_or_db_path: str) -> Dict[str, Any]:
    method match_sqls (line 57) | def match_sqls(self, predicted_sql: str, gold_sql: str, dsn_or_db_path...
    method iterated_execution (line 71) | def iterated_execution(self, predicted_sql: str, gold_sql: str, dsn_or...
  class SQLiteExecutor (line 91) | class SQLiteExecutor(BaseExecutor):
    method execute_sql (line 92) | def execute_sql(self, sql: str, dsn_or_db_path: str) -> dict:

FILE: premsql/generators/base.py
  class Text2SQLGeneratorBase (line 18) | class Text2SQLGeneratorBase(ABC):
    method __init__ (line 19) | def __init__(
    method load_client (line 40) | def load_client(self):
    method load_tokenizer (line 45) | def load_tokenizer(self):
    method model_name_or_path (line 50) | def model_name_or_path(self):
    method generate (line 54) | def generate(
    method execution_guided_decoding (line 64) | def execution_guided_decoding(
    method postprocess (line 98) | def postprocess(self, output_string: str):
    method load_results_from_folder (line 117) | def load_results_from_folder(self):
    method generate_and_save_results (line 124) | def generate_and_save_results(

FILE: premsql/generators/huggingface.py
  class Text2SQLGeneratorHF (line 16) | class Text2SQLGeneratorHF(Text2SQLGeneratorBase):
    method __init__ (line 17) | def __init__(
    method load_client (line 42) | def load_client(self) -> "transformers.PreTrainedModel":
    method load_tokenizer (line 56) | def load_tokenizer(self) -> "transformers.PreTrainedTokenizer":
    method model_name_or_path (line 66) | def model_name_or_path(self):
    method generate (line 69) | def generate(

FILE: premsql/generators/mlx.py
  class Text2SQLGeneratorMLX (line 18) | class Text2SQLGeneratorMLX(Text2SQLGeneratorBase):
    method __init__ (line 19) | def __init__(
    method load_client (line 38) | def load_client(self):
    method load_tokenizer (line 44) | def load_tokenizer(self):
    method model_name_or_path (line 49) | def model_name_or_path(self):
    method generate (line 52) | def generate(

FILE: premsql/generators/ollama_model.py
  class Text2SQLGeneratorOllama (line 16) | class Text2SQLGeneratorOllama(Text2SQLGeneratorBase):
    method __init__ (line 17) | def __init__(
    method load_client (line 34) | def load_client(self):
    method load_tokenizer (line 38) | def load_tokenizer(self):
    method model_name_or_path (line 42) | def model_name_or_path(self):
    method generate (line 45) | def generate(

FILE: premsql/generators/openai.py
  class Text2SQLGeneratorOpenAI (line 12) | class Text2SQLGeneratorOpenAI(Text2SQLGeneratorBase):
    method __init__ (line 13) | def __init__(
    method load_client (line 30) | def load_client(self):
    method load_tokenizer (line 35) | def load_tokenizer(self):
    method model_name_or_path (line 39) | def model_name_or_path(self):
    method generate (line 42) | def generate(

FILE: premsql/generators/premai.py
  class Text2SQLGeneratorPremAI (line 12) | class Text2SQLGeneratorPremAI(Text2SQLGeneratorBase):
    method __init__ (line 13) | def __init__(
    method load_client (line 35) | def load_client(self) -> Prem:
    method load_tokenizer (line 39) | def load_tokenizer(self) -> None:
    method model_name_or_path (line 43) | def model_name_or_path(self) -> str:
    method generate (line 46) | def generate(

FILE: premsql/logger.py
  function setup_console_logger (line 4) | def setup_console_logger(name, level=logging.INFO):

FILE: premsql/playground/backend/api/apps.py
  class ApiConfig (line 4) | class ApiConfig(AppConfig):

FILE: premsql/playground/backend/api/migrations/0001_initial.py
  class Migration (line 7) | class Migration(migrations.Migration):

FILE: premsql/playground/backend/api/models.py
  class Session (line 4) | class Session(models.Model):
    class Meta (line 12) | class Meta:
  class Completions (line 16) | class Completions(models.Model):
    class Meta (line 26) | class Meta:

FILE: premsql/playground/backend/api/pydantic_models.py
  class SessionCreationRequest (line 11) | class SessionCreationRequest(BaseModel):
  class SessionCreationResponse (line 16) | class SessionCreationResponse(BaseModel):
  class SessionSummary (line 28) | class SessionSummary(BaseModel):
  class SessionListResponse (line 39) | class SessionListResponse(BaseModel):
  class SessionDeleteResponse (line 49) | class SessionDeleteResponse(BaseModel):
  class CompletionCreationRequest (line 59) | class CompletionCreationRequest(BaseModel):
  class CompletionCreationResponse (line 64) | class CompletionCreationResponse(BaseModel):
  class CompletionSummary (line 75) | class CompletionSummary(BaseModel):
  class CompletionListResponse (line 85) | class CompletionListResponse(BaseModel):

FILE: premsql/playground/backend/api/serializers.py
  class AgentOutputSerializer (line 4) | class AgentOutputSerializer(serializers.Serializer):
  class SessionCreationRequestSerializer (line 26) | class SessionCreationRequestSerializer(serializers.Serializer):
  class SessionCreationResponseSerializer (line 30) | class SessionCreationResponseSerializer(serializers.Serializer):
  class SessionSummarySerializer (line 42) | class SessionSummarySerializer(serializers.Serializer):
  class SessionListResponseSerializer (line 51) | class SessionListResponseSerializer(serializers.Serializer):
  class SessionDeletionResponse (line 61) | class SessionDeletionResponse(serializers.Serializer):
  class CompletionCreationRequestSerializer (line 69) | class CompletionCreationRequestSerializer(serializers.Serializer):
  class CompletionCreationResponseSerializer (line 74) | class CompletionCreationResponseSerializer(serializers.Serializer):
  class CompletionSummarySerializer (line 85) | class CompletionSummarySerializer(serializers.Serializer):
  class CompletionListResponseSerializer (line 93) | class CompletionListResponseSerializer(serializers.Serializer):
  function create_model_serializer (line 102) | def create_model_serializer(model_class):

FILE: premsql/playground/backend/api/services.py
  class SessionManageService (line 34) | class SessionManageService:
    method __init__ (line 35) | def __init__(self) -> None:
    method create_session (line 38) | def create_session(
    method get_session (line 75) | def get_session(self, session_name: str) -> Optional[Session]:
    method list_session (line 81) | def list_session(self, page: int, page_size: int = 20) -> SessionListR...
    method delete_session (line 116) | def delete_session(self, session_name: str):
  class CompletionService (line 160) | class CompletionService:
    method __init__ (line 161) | def __init__(self) -> None:
    method completion (line 164) | def completion(
    method chat_history (line 225) | def chat_history(

FILE: premsql/playground/backend/api/utils.py
  function stop_server_on_port (line 11) | def stop_server_on_port(port: int):

FILE: premsql/playground/backend/api/views.py
  function create_session (line 42) | def create_session(request):
  function get_session (line 76) | def get_session(request, session_name):
  function list_sessions (line 125) | def list_sessions(request):
  function delete_session (line 161) | def delete_session(request, session_name):
  function create_completion (line 186) | def create_completion(request):
  function get_chat_history (line 243) | def get_chat_history(request, session_name):

FILE: premsql/playground/backend/backend_client.py
  class BackendAPIClient (line 18) | class BackendAPIClient:
    method __init__ (line 19) | def __init__(self):
    method create_session (line 26) | def create_session(self, request: SessionCreationRequest) -> SessionCr...
    method list_sessions (line 70) | def list_sessions(self, page: int = 1, page_size: int = 20) -> Session...
    method get_session (line 100) | def get_session(self, session_name: str) -> SessionListResponse:
    method delete_session (line 131) | def delete_session(self, session_name: str) -> SessionDeleteResponse:
    method create_completion (line 157) | def create_completion(self, request: CompletionCreationRequest) -> Com...
    method get_chat_history (line 185) | def get_chat_history(self, session_name: str, page: int = 1, page_size...

FILE: premsql/playground/backend/manage.py
  function main (line 7) | def main():

FILE: premsql/playground/frontend/components/chat.py
  class ChatComponent (line 14) | class ChatComponent:
    method __init__ (line 15) | def __init__(self) -> None:
    method _streamlit_chat_output (line 20) | def _streamlit_chat_output(self, message: AgentOutput | ExitWorkerOutp...
    method render_chat_env (line 55) | def render_chat_env(self, session_name: str) -> None:

FILE: premsql/playground/frontend/components/session.py
  class SessionComponent (line 13) | class SessionComponent:
    method __init__ (line 14) | def __init__(self) -> None:
    method render_list_sessions (line 17) | def render_list_sessions(self):
    method render_register_session (line 30) | def render_register_session(self):
    method render_additional_links (line 54) | def render_additional_links(self):
    method render_delete_session_view (line 60) | def render_delete_session_view(self):

FILE: premsql/playground/frontend/components/streamlit_plot.py
  class StreamlitPlotTool (line 10) | class StreamlitPlotTool(BasePlotTool):
    method __init__ (line 11) | def __init__(self):
    method run (line 20) | def run(self, data: pd.DataFrame, plot_config: Dict[str, str]) -> Any:
    method _validate_config (line 38) | def _validate_config(self, df: pd.DataFrame, plot_config: Dict[str, st...
    method _area_plot (line 65) | def _area_plot(self, df: pd.DataFrame, x: str, y: str) -> Any:
    method _bar_plot (line 69) | def _bar_plot(self, df: pd.DataFrame, x: str, y: str) -> Any:
    method _scatter_plot (line 73) | def _scatter_plot(self, df: pd.DataFrame, x: str, y: str) -> Any:
    method _histogram_plot (line 77) | def _histogram_plot(self, df: pd.DataFrame, x: str, y: str) -> Any:
    method _line_plot (line 83) | def _line_plot(self, df: pd.DataFrame, x: str, y: str) -> Any:
    method convert_plot_to_image (line 87) | def convert_plot_to_image(self, fig):

FILE: premsql/playground/frontend/components/uploader.py
  function render_starter_code (line 127) | def render_starter_code(session_name, db_path):
  class UploadComponent (line 179) | class UploadComponent:
    method render_kaggle_view (line 181) | def render_kaggle_view() -> Tuple[Optional[str], Optional[Path]]:
    method render_csv_upload_view (line 217) | def render_csv_upload_view() -> Tuple[Optional[str], Optional[Path]]:

FILE: premsql/playground/frontend/main.py
  function render_main_view (line 8) | def render_main_view():
  function main (line 25) | def main():

FILE: premsql/playground/frontend/utils.py
  function _is_valid_kaggle_id (line 12) | def _is_valid_kaggle_id(kaggle_id: str) -> bool:
  function download_from_kaggle (line 16) | def download_from_kaggle(kaggle_dataset_id: str):
  function _migrate_to_sqlite (line 20) | def _migrate_to_sqlite(csv_folder: Path, sqlite_db_path: Path) -> Path:
  function migrate_from_csv_to_sqlite (line 38) | def migrate_from_csv_to_sqlite(
  function migrate_local_csvs_to_sqlite (line 47) | def migrate_local_csvs_to_sqlite(

FILE: premsql/playground/inference_server/api_client.py
  class InferenceServerAPIError (line 7) | class InferenceServerAPIError(Exception):
  class InferenceServerAPIClient (line 11) | class InferenceServerAPIClient:
    method __init__ (line 12) | def __init__(self, timeout: int = 600) -> None:
    method _make_request (line 19) | def _make_request(
    method is_online (line 36) | def is_online(self, base_url: str) -> bool:
    method post_completion (line 44) | def post_completion(self, base_url: str, question: str) -> Dict[str, A...
    method get_session_info (line 51) | def get_session_info(self, base_url: str) -> Dict[str, Any]:
    method get_chat_history (line 55) | def get_chat_history(self, base_url: str, message_id: int) -> Dict[str...
    method delete_session (line 61) | def delete_session(self, base_url: str) -> Dict[str, Any]:

FILE: premsql/playground/inference_server/service.py
  class QuestionInput (line 17) | class QuestionInput(BaseModel):
  class SessionInfoResponse (line 21) | class SessionInfoResponse(BaseModel):
  class ChatHistoryResponse (line 30) | class ChatHistoryResponse(BaseModel):
  class CompletionResponse (line 35) | class CompletionResponse(BaseModel):
  class AgentServer (line 40) | class AgentServer:
    method __init__ (line 41) | def __init__(
    method lifespan (line 53) | async def lifespan(self, app: FastAPI):
    method create_app (line 62) | def create_app(self):
    method launch (line 158) | def launch(self):

FILE: premsql/tuner/callback.py
  class Text2SQLEvaluationCallback (line 24) | class Text2SQLEvaluationCallback(TrainerCallback):
    method __init__ (line 25) | def __init__(
    method on_step_end (line 53) | def on_step_end(
    method on_train_end (line 111) | def on_train_end(

FILE: premsql/tuner/config.py
  class DefaultTrainingArguments (line 15) | class DefaultTrainingArguments(TrainingArguments):
  class DefaultPeftArguments (line 50) | class DefaultPeftArguments(TrainingArguments):
  class DefaultLoraConfig (line 78) | class DefaultLoraConfig(LoraConfig):

FILE: premsql/tuner/full.py
  class Text2SQLFullFinetuner (line 15) | class Text2SQLFullFinetuner:
    method __init__ (line 16) | def __init__(
    method train (line 37) | def train(

FILE: premsql/tuner/peft.py
  class Text2SQLPeftTuner (line 23) | class Text2SQLPeftTuner:
    method __init__ (line 24) | def __init__(
    method train (line 54) | def train(

FILE: premsql/utils.py
  function convert_sqlite_path_to_dsn (line 22) | def convert_sqlite_path_to_dsn(path: str):
  function convert_sqlite_dsn_to_path (line 29) | def convert_sqlite_dsn_to_path(dsn: str) -> str:
  function print_data (line 37) | def print_data(data: dict):
  function save_to_json (line 52) | def save_to_json(save_path: Union[str, Path], json_object: dict):
  function load_from_json (line 62) | def load_from_json(result_json_path: str) -> dict:
  function sqlite_schema_prompt (line 70) | def sqlite_schema_prompt(db_path: str) -> str:
  function get_random_few_shot_prompts (line 96) | def get_random_few_shot_prompts(dataset: list[dict], num_few_shot: int):
  function get_accepted_filters (line 125) | def get_accepted_filters(data: list[dict]) -> Sequence[str]:
  function filter_options (line 137) | def filter_options(
  function tokenize_fn (line 159) | def tokenize_fn(strings: Sequence[str], tokenizer: "PreTrainedTokenizer"...