Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dbt-labs/ade-bench
/ functions
Functions
626 in github.com/dbt-labs/ade-bench
⨍
Functions
626
◇
Types & classes
90
↳
Endpoints
6
↓ 1 callers
Method
convert_duckdb_to_snowflake
Convert a DuckDB file to Snowflake database.
scripts_python/migrate_duckdb_to_snowflake.py:230
↓ 1 callers
Function
copy_database_file
Copy the database file to sandbox.
scripts_python/create_sandbox.py:146
↓ 1 callers
Function
copy_migration_files
Copy migration files to sandbox if migration_directory is specified.
scripts_python/create_sandbox.py:186
↓ 1 callers
Function
copy_project_contents
Copy shared project contents to sandbox.
scripts_python/create_sandbox.py:105
↓ 1 callers
Function
copy_shared_scripts
Copy shared scripts to sandbox.
scripts_python/create_sandbox.py:216
↓ 1 callers
Function
copy_task_files
Copy various task files to sandbox.
scripts_python/create_sandbox.py:163
↓ 1 callers
Function
copy_to_clipboard
Copy text to clipboard using pbcopy (macOS) or xclip (Linux).
scripts_python/extract_task_details.py:142
↓ 1 callers
Function
create_database
(db_path: str)
scripts_python/profiling_statistics.py:26
↓ 1 callers
Function
create_database
(db_path: str)
scripts_python/generate_db_from_profiling_summaries.py:20
↓ 1 callers
Method
create_snowflake_database
Create or replace a Snowflake database.
scripts_python/migrate_duckdb_to_snowflake.py:116
↓ 1 callers
Function
deep_merge
Recursively merge two dictionaries.
shared/scripts/merge_yaml.py:9
↓ 1 callers
Method
discover_duckdb_files
Discover DuckDB files in the configured source directory.
scripts_python/migrate_duckdb_to_snowflake.py:73
↓ 1 callers
Function
display_detailed_results
Display a detailed summary table of benchmark results.
scripts_python/summarize_results.py:300
↓ 1 callers
Function
dump_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 callers
Function
dump_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 callers
Function
execute_sql
Execute SQL query against the database.
shared/scripts/run_sql.py:203
↓ 1 callers
Method
export_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 callers
Method
export_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 callers
Method
export_table_to_parquet
Export a single table from DuckDB to Parquet format using direct DuckDB export.
scripts_python/duckdb_utils.py:137
↓ 1 callers
Function
extract_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 callers
Method
extract_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 callers
Function
extract_task_details
Extract details from all task.yaml files.
scripts_python/extract_task_details.py:35
↓ 1 callers
Method
extract_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 callers
Method
filter_files
Filter files based on include/exclude logic.
scripts_python/duckdb_utils.py:288
↓ 1 callers
Function
find_failing_equality_tests
Find failing equality tests from run_results.json.
shared/scripts/detect_failing_equality_tests.py:8
↓ 1 callers
Function
find_matching_variant
Find the matching variant for the given db_type and project_type.
scripts_python/create_sandbox.py:122
↓ 1 callers
Method
format_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 callers
Method
format_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 callers
Function
format_log_message
Format message with truncation to 97 characters.
ade_bench/utils/logger.py:147
↓ 1 callers
Function
format_log_stage
Format the stage name for display.
ade_bench/utils/logger.py:154
↓ 1 callers
Function
format_log_timestamp
Format timestamp for display.
ade_bench/utils/logger.py:171
↓ 1 callers
Method
format_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 callers
Function
format_summary_table
Format the summary data into a table format for display.
scripts_python/summarize_results.py:164
↓ 1 callers
Method
format_tool_input
Format tool input parameters nicely.
ade_bench/agents/installed_agents/claude_code/log_formatter.py:92
↓ 1 callers
Method
from_extension
Determine database type from file extension.
ade_bench/database/pool_manager.py:18
↓ 1 callers
Method
from_yaml
(cls, path: Path)
ade_bench/handlers/trial_handler.py:139
↓ 1 callers
Method
from_yaml_list
Load a list of terminal commands from a YAML file.
ade_bench/harness_models.py:233
↓ 1 callers
Function
generate_all_charts
(db_path: str)
scripts_python/profiling_statistics.py:578
↓ 1 callers
Function
generate_base_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/generate_function_charts_from_db.py:92
↓ 1 callers
Function
generate_base_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/profiling_statistics.py:395
↓ 1 callers
Function
generate_charts_from_database
(db_path: str)
scripts_python/generate_function_charts_from_db.py:275
↓ 1 callers
Function
generate_db_path
(base_dir: str = "profiling")
scripts_python/profiling_statistics.py:19
↓ 1 callers
Function
generate_db_path
(base_dir: str = "profiling")
scripts_python/generate_db_from_profiling_summaries.py:13
↓ 1 callers
Function
generate_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 callers
Function
generate_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 callers
Function
generate_function_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/generate_function_charts_from_db.py:160
↓ 1 callers
Function
generate_function_charts
(conn: sqlite3.Connection, output_dir: Path, pbar: tqdm)
scripts_python/profiling_statistics.py:463
↓ 1 callers
Function
generate_html_table
Generate an HTML table of benchmark results with action links.
scripts_python/summarize_results.py:212
↓ 1 callers
Method
generate_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 callers
Function
generate_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 callers
Function
get_canonical_column_order
Define the canonical column order for the merged TSV.
scripts_python/analyze.py:167
↓ 1 callers
Method
get_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 callers
Function
get_duckdb_connection
Get DuckDB connection from profiles.yml.
shared/scripts/run_sql.py:128
↓ 1 callers
Method
get_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 callers
Function
get_equality_macro_content
Return the Jinja macro that implements equality test logic.
ade_bench/utils/test_generator.py:92
↓ 1 callers
Function
get_latest_summary_folder
(summaries_dir: Path)
scripts_python/profiling_statistics.py:614
↓ 1 callers
Method
get_migration_script_path
Get the path to a specific migration script.
ade_bench/handlers/trial_handler.py:510
↓ 1 callers
Function
get_ordered_host_ids
Get all host IDs ordered by ID where IS_SUPERHOST is not null.
tasks/airbnb011/setup/deterministic_mutate.py:15
↓ 1 callers
Function
get_ordered_host_ids
Get all host IDs ordered by ID where IS_SUPERHOST is not null.
tasks/airbnb010/setup/deterministic_mutate.py:12
↓ 1 callers
Function
get_profile_name
Get the profile name from project type and db type.
shared/scripts/run_sql.py:85
↓ 1 callers
Function
get_snowflake_connection
Get Snowflake connection from profiles.yml.
shared/scripts/run_sql.py:150
↓ 1 callers
Method
get_solution_seed_configs
Parse solution_seeds into SolutionSeedConfig objects.
ade_bench/handlers/trial_handler.py:124
↓ 1 callers
Function
get_task_config
Get the task configuration from task.yaml.
scripts_python/create_sandbox.py:56
↓ 1 callers
Method
get_unified_diff
Get unified diff for a specific file.
ade_bench/handlers/file_diff_handler.py:122
↓ 1 callers
Function
import_cprofile_data
(cursor: sqlite3.Cursor, run_id: int, prof_path: Path)
scripts_python/profiling_statistics.py:185
↓ 1 callers
Function
import_cprofile_data
(cursor: sqlite3.Cursor, run_id: int, prof_path: Path)
scripts_python/generate_db_from_profiling_summaries.py:179
↓ 1 callers
Function
import_metadata
(summaries_dir: str = "profiling/summaries", base_dir: str = "profiling")
scripts_python/generate_db_from_profiling_summaries.py:257
↓ 1 callers
Function
import_single_folder
(conn: sqlite3.Connection, folder: Path)
scripts_python/profiling_statistics.py:263
↓ 1 callers
Function
initialize_dynamic_logging
Initialize the Rich logging system with the list of tasks.
ade_bench/utils/logger.py:255
↓ 1 callers
Method
initialize_tasks
Initialize the Rich table with task rows.
ade_bench/utils/logger.py:25
↓ 1 callers
Function
insert_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 callers
Function
is_docker_available
Check if python can connect to an actively running docker daemon.
ade_bench/cli/ab/check.py:174
↓ 1 callers
Function
is_dotenv_available
Check if .env exists and is loadable.
ade_bench/cli/ab/check.py:110
↓ 1 callers
Method
is_error
Check if this failure mode is an infrastructure/harness error vs a task failure.
ade_bench/harness_models.py:34
↓ 1 callers
Function
is_error_result
Check if a result is an error (infrastructure/harness issue) vs a task failure.
ade_bench/utils/results_writer.py:10
↓ 1 callers
Function
is_podman_available
Check if python can connect to an actively running podman daemon.
ade_bench/cli/ab/check.py:190
↓ 1 callers
Function
load_diff_json
Load the JSON diff data from the corresponding file_diffs.json file.
scripts_python/render_file_diffs.py:19
↓ 1 callers
Function
load_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 callers
Method
load_parquet_to_snowflake
Load a Parquet file into Snowflake using internal stage and COPY INTO.
scripts_python/migrate_duckdb_to_snowflake.py:138
↓ 1 callers
Function
load_profiles
Load profiles.yml file.
shared/scripts/run_sql.py:74
↓ 1 callers
Function
load_yaml_file
Load and parse a YAML file.
scripts_python/extract_task_details.py:25
↓ 1 callers
Function
log
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 callers
Function
main
()
shared/scripts/merge_yaml.py:20
↓ 1 callers
Function
main
Main entry point.
shared/scripts/run_sql.py:254
↓ 1 callers
Function
main
()
shared/scripts/dump_tables.py:164
↓ 1 callers
Function
main
()
shared/scripts/detect_failing_equality_tests.py:60
↓ 1 callers
Function
main
()
shared/scripts/compare_tables.py:731
↓ 1 callers
Function
main
()
tasks/airbnb011/setup/deterministic_mutate.py:73
↓ 1 callers
Function
main
()
tasks/airbnb010/setup/deterministic_mutate.py:39
↓ 1 callers
Function
main
Generate HTML dashboard for the latest experiment and open it.
scripts_python/view_results.py:15
↓ 1 callers
Function
main
()
scripts_python/create_sandbox.py:311
↓ 1 callers
Function
main
Main function to generate HTML results.
scripts_python/generate_results_html.py:594
↓ 1 callers
Function
main
()
scripts_python/run_harness.py:11
↓ 1 callers
Function
main
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 callers
Function
main
Main entry point.
scripts_python/analyze.py:305
↓ 1 callers
Function
main
(include: List[str] = None, exclude: List[str] = None, use_database_export: bool = False)
scripts_python/migrate_duckdb_to_snowflake.py:364
↓ 1 callers
Function
main
Interactive wizard for creating new ADE-Bench tasks.
ade_bench/cli/wizard.py:12
↓ 1 callers
Function
make_missing_relation_result
Create a comparison result when one relation is missing.
shared/scripts/compare_tables.py:419
↓ 1 callers
Function
merge_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 callers
Method
merge_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
← previous
next →
201–300 of 626, ranked by callers