MCPcopy Create free account

hub / github.com/dbt-labs/ade-bench / functions

Functions626 in github.com/dbt-labs/ade-bench

↓ 1 callersMethodconvert_duckdb_to_snowflake
Convert a DuckDB file to Snowflake database.
scripts_python/migrate_duckdb_to_snowflake.py:230
↓ 1 callersFunctioncopy_database_file
Copy the database file to sandbox.
scripts_python/create_sandbox.py:146
↓ 1 callersFunctioncopy_migration_files
Copy migration files to sandbox if migration_directory is specified.
scripts_python/create_sandbox.py:186
↓ 1 callersFunctioncopy_project_contents
Copy shared project contents to sandbox.
scripts_python/create_sandbox.py:105
↓ 1 callersFunctioncopy_shared_scripts
Copy shared scripts to sandbox.
scripts_python/create_sandbox.py:216
↓ 1 callersFunctioncopy_task_files
Copy various task files to sandbox.
scripts_python/create_sandbox.py:163
↓ 1 callersFunctioncopy_to_clipboard
Copy text to clipboard using pbcopy (macOS) or xclip (Linux).
scripts_python/extract_task_details.py:142
↓ 1 callersFunctioncreate_database
(db_path: str)
scripts_python/profiling_statistics.py:26
↓ 1 callersFunctioncreate_database
(db_path: str)
scripts_python/generate_db_from_profiling_summaries.py:20
↓ 1 callersMethodcreate_snowflake_database
Create or replace a Snowflake database.
scripts_python/migrate_duckdb_to_snowflake.py:116
↓ 1 callersFunctiondeep_merge
Recursively merge two dictionaries.
shared/scripts/merge_yaml.py:9
↓ 1 callersMethoddiscover_duckdb_files
Discover DuckDB files in the configured source directory.
scripts_python/migrate_duckdb_to_snowflake.py:73
↓ 1 callersFunctiondisplay_detailed_results
Display a detailed summary table of benchmark results.
scripts_python/summarize_results.py:300
↓ 1 callersFunctiondump_tables_duckdb
Dump tables from a DuckDB database to parquet + CSV. Args: db_path: Path to the .duckdb file relations: List of relation names to
shared/scripts/dump_tables.py:10
↓ 1 callersFunctiondump_tables_snowflake
Dump tables from Snowflake to parquet + CSV. Uses fetch_pandas_all() and DataFrame.to_parquet()/to_csv() directly. Args: relations:
shared/scripts/dump_tables.py:79
↓ 1 callersFunctionexecute_sql
Execute SQL query against the database.
shared/scripts/run_sql.py:203
↓ 1 callersMethodexport_all_tables_to_parquet
Export all tables from a DuckDB file to Parquet format. Args: duckdb_path: Path to the DuckDB file output_di
scripts_python/duckdb_utils.py:166
↓ 1 callersMethodexport_database_to_parquet
Export entire DuckDB database to Parquet using DuckDB's EXPORT DATABASE command. This is more efficient than exporting tables individ
scripts_python/duckdb_utils.py:233
↓ 1 callersMethodexport_table_to_parquet
Export a single table from DuckDB to Parquet format using direct DuckDB export.
scripts_python/duckdb_utils.py:137
↓ 1 callersFunctionextract_and_save_tasks
Extract task details from task.yaml files and save to TSV. Calls the extract_task_details script to do the work. Args: output_fi
scripts_python/analyze.py:255
↓ 1 callersMethodextract_jsonl_content
Extract only the JSON lines from a log file that may contain mixed content. The log file may contain terminal output before the JSON
ade_bench/agents/installed_agents/claude_code/log_formatter.py:33
↓ 1 callersFunctionextract_task_details
Extract details from all task.yaml files.
scripts_python/extract_task_details.py:35
↓ 1 callersMethodextract_tools_used
Extract deduplicated list of tool names from the agent's log file. This method can be overridden by subclasses to provide agent-spec
ade_bench/agents/base_agent.py:78
↓ 1 callersMethodfilter_files
Filter files based on include/exclude logic.
scripts_python/duckdb_utils.py:288
↓ 1 callersFunctionfind_failing_equality_tests
Find failing equality tests from run_results.json.
shared/scripts/detect_failing_equality_tests.py:8
↓ 1 callersFunctionfind_matching_variant
Find the matching variant for the given db_type and project_type.
scripts_python/create_sandbox.py:122
↓ 1 callersMethodformat_agent_log
Format the agent's log file into a human-readable string. This method can be overridden by subclasses to provide agent-specific
ade_bench/agents/base_agent.py:62
↓ 1 callersMethodformat_log
Parse and format a log file in one step. Args: log_path: Path to the log file to parse Returns: For
ade_bench/agents/log_formatter.py:43
↓ 1 callersFunctionformat_log_message
Format message with truncation to 97 characters.
ade_bench/utils/logger.py:147
↓ 1 callersFunctionformat_log_stage
Format the stage name for display.
ade_bench/utils/logger.py:154
↓ 1 callersFunctionformat_log_timestamp
Format timestamp for display.
ade_bench/utils/logger.py:171
↓ 1 callersMethodformat_readable_log
Format the parsed turns into a readable text string. Args: turns: List of turn dictionaries from parse_log_file
ade_bench/agents/log_formatter.py:31
↓ 1 callersFunctionformat_summary_table
Format the summary data into a table format for display.
scripts_python/summarize_results.py:164
↓ 1 callersMethodformat_tool_input
Format tool input parameters nicely.
ade_bench/agents/installed_agents/claude_code/log_formatter.py:92
↓ 1 callersMethodfrom_extension
Determine database type from file extension.
ade_bench/database/pool_manager.py:18
↓ 1 callersMethodfrom_yaml
(cls, path: Path)
ade_bench/handlers/trial_handler.py:139
↓ 1 callersMethodfrom_yaml_list
Load a list of terminal commands from a YAML file.
ade_bench/harness_models.py:233
↓ 1 callersFunctiongenerate_all_charts
(db_path: str)
scripts_python/profiling_statistics.py:578
↓ 1 callersFunctiongenerate_base_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/generate_function_charts_from_db.py:92
↓ 1 callersFunctiongenerate_base_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/profiling_statistics.py:395
↓ 1 callersFunctiongenerate_charts_from_database
(db_path: str)
scripts_python/generate_function_charts_from_db.py:275
↓ 1 callersFunctiongenerate_db_path
(base_dir: str = "profiling")
scripts_python/profiling_statistics.py:19
↓ 1 callersFunctiongenerate_db_path
(base_dir: str = "profiling")
scripts_python/generate_db_from_profiling_summaries.py:13
↓ 1 callersFunctiongenerate_equality_test
Generate an equality test for a solution seed table. Args: table_name: Name of the table to test config: Optional configuration f
ade_bench/utils/test_generator.py:125
↓ 1 callersFunctiongenerate_existence_test
Generate an existence test for a solution seed table. Args: table_name: Name of the table to test Returns: Generated SQL tes
ade_bench/utils/test_generator.py:97
↓ 1 callersFunctiongenerate_function_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/generate_function_charts_from_db.py:160
↓ 1 callersFunctiongenerate_function_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/profiling_statistics.py:463
↓ 1 callersFunctiongenerate_html_table
Generate an HTML table of benchmark results with action links.
scripts_python/summarize_results.py:212
↓ 1 callersMethodgenerate_html_transcript
Generate an HTML transcript using claude-code-transcripts. This method extracts JSON lines from the log file (which may contain
ade_bench/agents/installed_agents/claude_code/log_formatter.py:290
↓ 1 callersFunctiongenerate_solution_tests
Generate both equality and existence tests for a solution seed table. Args: table_name: Name of the table to generate tests for t
ade_bench/utils/test_generator.py:188
↓ 1 callersFunctionget_canonical_column_order
Define the canonical column order for the merged TSV.
scripts_python/analyze.py:167
↓ 1 callersMethodget_dbt_project_path
Get the path to a specific dbt project directory. Args: project_name: Name of the project directory project_type: Typ
ade_bench/handlers/trial_handler.py:486
↓ 1 callersFunctionget_duckdb_connection
Get DuckDB connection from profiles.yml.
shared/scripts/run_sql.py:128
↓ 1 callersMethodget_duckdb_file_path
Get the path to a specific DuckDB database file. Args: db_name: Name of the database file (without .duckdb extension)
ade_bench/handlers/trial_handler.py:470
↓ 1 callersFunctionget_equality_macro_content
Return the Jinja macro that implements equality test logic.
ade_bench/utils/test_generator.py:92
↓ 1 callersFunctionget_latest_summary_folder
(summaries_dir: Path)
scripts_python/profiling_statistics.py:614
↓ 1 callersMethodget_migration_script_path
Get the path to a specific migration script.
ade_bench/handlers/trial_handler.py:510
↓ 1 callersFunctionget_ordered_host_ids
Get all host IDs ordered by ID where IS_SUPERHOST is not null.
tasks/airbnb011/setup/deterministic_mutate.py:15
↓ 1 callersFunctionget_ordered_host_ids
Get all host IDs ordered by ID where IS_SUPERHOST is not null.
tasks/airbnb010/setup/deterministic_mutate.py:12
↓ 1 callersFunctionget_profile_name
Get the profile name from project type and db type.
shared/scripts/run_sql.py:85
↓ 1 callersFunctionget_snowflake_connection
Get Snowflake connection from profiles.yml.
shared/scripts/run_sql.py:150
↓ 1 callersMethodget_solution_seed_configs
Parse solution_seeds into SolutionSeedConfig objects.
ade_bench/handlers/trial_handler.py:124
↓ 1 callersFunctionget_task_config
Get the task configuration from task.yaml.
scripts_python/create_sandbox.py:56
↓ 1 callersMethodget_unified_diff
Get unified diff for a specific file.
ade_bench/handlers/file_diff_handler.py:122
↓ 1 callersFunctionimport_cprofile_data
(cursor: sqlite3.Cursor, run_id: int, prof_path: Path)
scripts_python/profiling_statistics.py:185
↓ 1 callersFunctionimport_cprofile_data
(cursor: sqlite3.Cursor, run_id: int, prof_path: Path)
scripts_python/generate_db_from_profiling_summaries.py:179
↓ 1 callersFunctionimport_metadata
(summaries_dir: str = "profiling/summaries", base_dir: str = "profiling")
scripts_python/generate_db_from_profiling_summaries.py:257
↓ 1 callersFunctionimport_single_folder
(conn: sqlite3.Connection, folder: Path)
scripts_python/profiling_statistics.py:263
↓ 1 callersFunctioninitialize_dynamic_logging
Initialize the Rich logging system with the list of tasks.
ade_bench/utils/logger.py:255
↓ 1 callersMethodinitialize_tasks
Initialize the Rich table with task rows.
ade_bench/utils/logger.py:25
↓ 1 callersFunctioninsert_new_hosts
Insert 50 new hosts: 25 superhosts and 25 non-superhosts. Names are copied from existing hosts to test unique_key handling.
tasks/airbnb011/setup/deterministic_mutate.py:41
↓ 1 callersFunctionis_docker_available
Check if python can connect to an actively running docker daemon.
ade_bench/cli/ab/check.py:174
↓ 1 callersFunctionis_dotenv_available
Check if .env exists and is loadable.
ade_bench/cli/ab/check.py:110
↓ 1 callersMethodis_error
Check if this failure mode is an infrastructure/harness error vs a task failure.
ade_bench/harness_models.py:34
↓ 1 callersFunctionis_error_result
Check if a result is an error (infrastructure/harness issue) vs a task failure.
ade_bench/utils/results_writer.py:10
↓ 1 callersFunctionis_podman_available
Check if python can connect to an actively running podman daemon.
ade_bench/cli/ab/check.py:190
↓ 1 callersFunctionload_diff_json
Load the JSON diff data from the corresponding file_diffs.json file.
scripts_python/render_file_diffs.py:19
↓ 1 callersFunctionload_migration_config
Load migration configuration from a YAML file. Args: migration_file: Path to the migration YAML file (relative to migrations dir), o
scripts_python/analyze.py:24
↓ 1 callersMethodload_parquet_to_snowflake
Load a Parquet file into Snowflake using internal stage and COPY INTO.
scripts_python/migrate_duckdb_to_snowflake.py:138
↓ 1 callersFunctionload_profiles
Load profiles.yml file.
shared/scripts/run_sql.py:74
↓ 1 callersFunctionload_yaml_file
Load and parse a YAML file.
scripts_python/extract_task_details.py:25
↓ 1 callersFunctionlog
Centralized info logging function for the harness. Args: logger: The logger instance to use task: The task being run
ade_bench/utils/logger.py:180
↓ 1 callersFunctionmain
()
shared/scripts/merge_yaml.py:20
↓ 1 callersFunctionmain
Main entry point.
shared/scripts/run_sql.py:254
↓ 1 callersFunctionmain
()
shared/scripts/dump_tables.py:164
↓ 1 callersFunctionmain
()
shared/scripts/detect_failing_equality_tests.py:60
↓ 1 callersFunctionmain
()
shared/scripts/compare_tables.py:731
↓ 1 callersFunctionmain
()
tasks/airbnb011/setup/deterministic_mutate.py:73
↓ 1 callersFunctionmain
()
tasks/airbnb010/setup/deterministic_mutate.py:39
↓ 1 callersFunctionmain
Generate HTML dashboard for the latest experiment and open it.
scripts_python/view_results.py:15
↓ 1 callersFunctionmain
()
scripts_python/create_sandbox.py:311
↓ 1 callersFunctionmain
Main function to generate HTML results.
scripts_python/generate_results_html.py:594
↓ 1 callersFunctionmain
()
scripts_python/run_harness.py:11
↓ 1 callersFunctionmain
Main function to extract task details and generate CSV. Args: output_file: Optional path to save the TSV file to quiet: If True,
scripts_python/extract_task_details.py:163
↓ 1 callersFunctionmain
Main entry point.
scripts_python/analyze.py:305
↓ 1 callersFunctionmain
(include: List[str] = None, exclude: List[str] = None, use_database_export: bool = False)
scripts_python/migrate_duckdb_to_snowflake.py:364
↓ 1 callersFunctionmain
Interactive wizard for creating new ADE-Bench tasks.
ade_bench/cli/wizard.py:12
↓ 1 callersFunctionmake_missing_relation_result
Create a comparison result when one relation is missing.
shared/scripts/compare_tables.py:419
↓ 1 callersFunctionmerge_experiment_results
Merge all results.tsv files from saved experiments into a single TSV. Args: experiments_dir: Path to the experiments_saved directory
scripts_python/analyze.py:193
↓ 1 callersMethodmerge_markers
Merge asciinema markers into a recording. Inserts marker events into an asciinema recording file at specified timestamps. Ma
ade_bench/handlers/asciinema_handler.py:22
← previousnext →201–300 of 626, ranked by callers