MCPcopy Create free account

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

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

↓ 2 callersMethodstop
Stop and remove the docker compose services.
ade_bench/terminal/docker_compose_manager.py:135
↓ 2 callersFunctionupdate_file_in_container
Read file from container, apply update function, write back to container.
ade_bench/setup/setup_utils.py:31
↓ 1 callersMethod__init__
(self, **kwargs)
ade_bench/agents/base_agent.py:56
↓ 1 callersMethod_calculate_duration
Calculate total experiment duration.
scripts_python/generate_results_html.py:110
↓ 1 callersMethod_calculate_pass_at_k
(self, k: int, task_counts: dict[str, list[int]])
ade_bench/harness_models.py:124
↓ 1 callersFunction_clone_database
Clone a Snowflake database. Returns: Tuple of (success, error_message). error_message is None on success.
ade_bench/setup/snowflake_setup.py:59
↓ 1 callersFunction_compare_tables_inner
Inner comparison logic operating on an open DuckDB connection.
shared/scripts/compare_tables.py:37
↓ 1 callersMethod_compute_diff
Compute the differences between the two snapshots.
ade_bench/handlers/file_diff_handler.py:107
↓ 1 callersFunction_copy_to_clipboard
Copy task data as TSV to the system clipboard.
ade_bench/cli/ab/tasks.py:142
↓ 1 callersMethod_copy_tsv_files
Copy TSV files as plaintext.
scripts_python/generate_results_html.py:220
↓ 1 callersMethod_create_agent_for_task
Create a fresh agent for a specific task. Args: task_id: The ID of the task to create an agent for. Returns:
ade_bench/harness.py:169
↓ 1 callersMethod_create_status_message
Create the final status message based on parsed results. The PASS/FAIL logic here mirrors harness._is_resolved() exactly: -
ade_bench/parsers/dbt_parser.py:45
↓ 1 callersMethod_create_tar_archive
(paths: list[Path], container_filename: str | None)
ade_bench/terminal/docker_compose_manager.py:164
↓ 1 callersFunction_create_user_and_role
Create user and role for the database. Returns: Tuple of (success, error_message). error_message is None on success.
ade_bench/setup/snowflake_setup.py:97
↓ 1 callersFunction_detect_systematic_columns
Pre-scan: try excluding each column from EXCEPT ALL to find systematic diffs. If excluding a single column recovers >threshold of rows as exact m
shared/scripts/compare_tables.py:325
↓ 1 callersFunction_detect_systematic_diffs
Detect columns that differ in >= threshold fraction of paired rows. Returns a dict of column_name -> {diff_count, total_paired, sample_values}.
shared/scripts/compare_tables.py:375
↓ 1 callersMethod_disable_console_handlers
Disable console handlers to prevent double printing.
ade_bench/utils/logger.py:46
↓ 1 callersMethod_ends_with_newline
(self, key: str)
ade_bench/terminal/tmux_session.py:138
↓ 1 callersMethod_execute_queries
Execute multiple SQL queries separated by semicolons.
scripts_python/migrate_duckdb_to_snowflake.py:61
↓ 1 callersMethod_execute_tasks
Execute all tasks in parallel and collect their results.
ade_bench/harness.py:1441
↓ 1 callersMethod_execute_trials
Execute trials for the current skill set. Returns: BenchmarkResults: The results of the trials.
ade_bench/harness.py:1559
↓ 1 callersMethod_extract_comparison_artifacts
Extract comparison artifacts from the container after test execution.
ade_bench/harness.py:1084
↓ 1 callersMethod_extract_csv_files
Extract CSV files and schema file from the database after agent execution.
ade_bench/harness.py:1025
↓ 1 callersMethod_extract_duckdb_csv
Extract CSV files from DuckDB database.
ade_bench/harness.py:1130
↓ 1 callersMethod_extract_table_name
Extract table name from Parquet filename.
scripts_python/migrate_duckdb_to_snowflake.py:211
↓ 1 callersMethod_extract_table_schema
Extract schema information from a database table.
ade_bench/harness.py:1001
↓ 1 callersMethod_format_duration_seconds
Format duration in seconds to H:MM:SS format.
scripts_python/generate_results_html.py:100
↓ 1 callersFunction_fuzzy_match_rows
Fuzzy-match unmatched rows by similarity score.
shared/scripts/compare_tables.py:230
↓ 1 callersMethod_generate_data_comparisons_page
Generate data comparisons detail page from diff.html artifacts.
scripts_python/generate_results_html.py:457
↓ 1 callersMethod_generate_diffs_page
Generate diffs detail page.
scripts_python/generate_results_html.py:432
↓ 1 callersMethod_generate_panes_page
Generate panes detail page. Displays terminal output in chronological order: 1. pre-agent.txt (setup before agent runs) 2. HT
scripts_python/generate_results_html.py:337
↓ 1 callersMethod_generate_results_page
Generate results.json detail page.
scripts_python/generate_results_html.py:319
↓ 1 callersMethod_generate_solution_tests
Generate solution tests for tables specified in solution_seeds. Args: trial_handler: The trial handler containing task configurat
ade_bench/harness.py:454
↓ 1 callersMethod_generate_summary_page
Generate the main summary page.
scripts_python/generate_results_html.py:126
↓ 1 callersMethod_generate_task_detail_pages
Generate detail pages for a specific task.
scripts_python/generate_results_html.py:285
↓ 1 callersMethod_get_base_task_id
Extract base task ID from variant task ID. e.g., 'foo.hard.1-of-1' -> 'foo' 'foo.base.2-of-3' -> 'foo' 'foo' -> '
scripts_python/generate_results_html.py:246
↓ 1 callersMethod_get_dbt_dynamic_env
Get dynamic environment variables for dbt MCP server.
ade_bench/agents/installed_agents/abstract_installed_agent.py:73
↓ 1 callersMethod_get_expected_test_count
Get the expected test count from the '[ade-bench] expected_test_count=N' line.
ade_bench/parsers/dbt_parser.py:200
↓ 1 callersMethod_get_git_commit_hash
Get the current git commit hash.
ade_bench/harness.py:1247
↓ 1 callersMethod_get_task_success_counts
(self)
ade_bench/harness_models.py:110
↓ 1 callersMethod_get_trial_name
( self, task_path: Path, task_key: str | None, attempt: int, )
ade_bench/harness.py:1431
↓ 1 callersMethod_get_user
Get the user who ran the experiment. First tries to get the git user.name, falls back to system username.
ade_bench/harness.py:1258
↓ 1 callersMethod_has_test_results
Check if the content contains actual test results (not just compilation errors).
ade_bench/parsers/dbt_parser.py:190
↓ 1 callersMethod_init_dataset
(self)
ade_bench/harness.py:197
↓ 1 callersMethod_init_file_logger
Initialize or reinitialize file logging for the current run path.
ade_bench/harness.py:232
↓ 1 callersMethod_init_logger
Initialize console logging. File logging is initialized per plugin set.
ade_bench/harness.py:223
↓ 1 callersMethod_init_plugin_sets
Load and resolve plugin sets from configuration.
ade_bench/harness.py:205
↓ 1 callersMethod_init_solution_dict
(self)
ade_bench/agents/sage_agent.py:46
↓ 1 callersMethod_install_skill_origin
Install skills from a single skill origin. Args: skill_origin: The skill origin configuration terminal: The Docker co
ade_bench/plugins/skills_handler.py:27
↓ 1 callersMethod_is_resolved
(self, parser_result: ParserResult | None)
ade_bench/harness.py:244
↓ 1 callersMethod_load_all_ready_tasks
Load all tasks from directory, filtering to only status=ready.
ade_bench/utils/dataset.py:111
↓ 1 callersMethod_load_catalog
Load the database catalog from disk.
ade_bench/database/pool_manager.py:65
↓ 1 callersMethod_load_experiment_data
Load experiment data from results.json and run_metadata.json.
scripts_python/generate_results_html.py:48
↓ 1 callersMethod_load_experiment_set
Load tasks from an experiment set file.
ade_bench/utils/dataset.py:80
↓ 1 callersMethod_load_specific_tasks
Load specific tasks from the task_ids.
ade_bench/utils/dataset.py:44
↓ 1 callersMethod_load_task_yaml_contents
Load task.yaml contents for each task and return as a dictionary.
scripts_python/generate_results_html.py:266
↓ 1 callersMethod_load_wildcard_tasks
Load all tasks that start with the given prefix.
ade_bench/utils/dataset.py:70
↓ 1 callersMethod_parse_agent_output
Parse the agent output to extract metrics. Override in subclasses if needed.
ade_bench/agents/installed_agents/abstract_installed_agent.py:288
↓ 1 callersMethod_parse_json_response
Parse the JSON response data to extract metrics.
ade_bench/parsers/claude_parser.py:83
↓ 1 callersMethod_parse_json_response
Parse the JSON response data to extract metrics.
ade_bench/parsers/macro_parser.py:72
↓ 1 callersMethod_parse_results
( self, trial_handler: TrialHandler, post_agent_pane: str, )
ade_bench/harness.py:481
↓ 1 callersMethod_pass_at_k_estimator
Calculates 1 - comb(n - c, k) / comb(n, k).
ade_bench/harness_models.py:118
↓ 1 callersMethod_prepare_keys
Prepare keys for sending to the terminal. Args: keys (str | list[str]): The keys to send to the terminal. bl
ade_bench/terminal/tmux_session.py:160
↓ 1 callersMethod_prevent_execution
(self, keys: list[str])
ade_bench/terminal/tmux_session.py:145
↓ 1 callersMethod_process_recording_line
Process a single line from the recording, inserting markers as needed.
ade_bench/handlers/asciinema_handler.py:60
↓ 1 callersMethod_rebuild_table
Rebuild the table with current task data.
ade_bench/utils/logger.py:115
↓ 1 callersMethod_remove_excluded_tasks
Remove tasks that are in the excluded list.
ade_bench/utils/dataset.py:146
↓ 1 callersMethod_run_agent
( self, session: TmuxSession, trial_handler: TrialHandler, agent: BaseAgent,
ade_bench/harness.py:539
↓ 1 callersMethod_run_agent_commands
Commands to run the agent with the given task prompt.
ade_bench/agents/installed_agents/abstract_installed_agent.py:64
↓ 1 callersMethod_run_agent_commands
Return the command(s) that will run the Macro agent inside the container.
ade_bench/agents/installed_agents/macro/macro_agent.py:38
↓ 1 callersMethod_run_agent_with_timeout
( self, trial_handler: TrialHandler, session: TmuxSession, logging_dir: Path,
ade_bench/harness.py:508
↓ 1 callersMethod_run_setup
Run comprehensive setup including files, migration, and scripts.
ade_bench/harness.py:631
↓ 1 callersMethod_run_tests
Run tests with proper timeout management.
ade_bench/harness.py:391
↓ 1 callersMethod_run_trial
( self, trial_handler: TrialHandler, config: dict, )
ade_bench/harness.py:691
↓ 1 callersMethod_save_diff_summary
Save a human-readable summary of file changes.
ade_bench/handlers/file_diff_handler.py:328
↓ 1 callersMethod_save_diff_to_trial_log
Save detailed diff information to the trial log file.
ade_bench/handlers/file_diff_handler.py:448
↓ 1 callersFunction_save_most_recent
Save the most recent run with results.
ade_bench/cli/ab/save.py:72
↓ 1 callersMethod_send_blocking_keys
( self, keys: list[str], max_timeout_sec: float, )
ade_bench/terminal/tmux_session.py:187
↓ 1 callersMethod_send_non_blocking_keys
( self, keys: list[str], min_timeout_sec: float, )
ade_bench/terminal/tmux_session.py:202
↓ 1 callersMethod_setup_test_env
(self, terminal: Terminal, trial_handler: TrialHandler)
ade_bench/harness.py:282
↓ 1 callersMethod_tmux_capture_pane
(self, capture_entire: bool = False)
ade_bench/terminal/tmux_session.py:92
↓ 1 callersMethod_update_metadata_on_end
(self, results: BenchmarkResults)
ade_bench/harness.py:1362
↓ 1 callersFunction_update_snowflake_files
(session, project_name: str, task_id: str, project_dir: Path)
ade_bench/setup/dbt_setup.py:47
↓ 1 callersFunction_values_match
Check if two values match, with numeric tolerance for floats.
shared/scripts/compare_tables.py:212
↓ 1 callersMethod_write_merged_recording
Write a new recording file with markers merged in at the correct timestamps.
ade_bench/handlers/asciinema_handler.py:41
↓ 1 callersMethod_write_panes_page
Write the panes page with pre-built HTML content. Unlike _write_detail_page which escapes content, this method accepts raw HTML that
scripts_python/generate_results_html.py:564
↓ 1 callersMethod_write_remaining_markers
Write any remaining markers that come after all recorded events.
ade_bench/handlers/asciinema_handler.py:93
↓ 1 callersMethod_write_results
(self, results: BenchmarkResults)
ade_bench/harness.py:1240
↓ 1 callersMethod_write_run_metadata
(self)
ade_bench/harness.py:1337
↓ 1 callersMethodadd_exclusions
Add files to exclusion list.
scripts_python/migrate_duckdb_to_snowflake.py:69
↓ 1 callersMethodadd_file_content
Add file content and return its hash.
ade_bench/handlers/file_diff_handler.py:22
↓ 1 callersMethodanalyze_duckdb_schema
Analyze schema and tables in a DuckDB file.
scripts_python/duckdb_utils.py:55
↓ 1 callersMethodanalyze_duckdb_schema
Analyze schema and tables in a DuckDB file.
scripts_python/migrate_duckdb_to_snowflake.py:77
↓ 1 callersFunctioncheck_anthropic_key
Check ANTHROPIC_API_KEY.
ade_bench/cli/ab/check.py:138
↓ 1 callersFunctioncheck_gemini_key
Check GEMINI_API_KEY.
ade_bench/cli/ab/check.py:158
↓ 1 callersFunctioncheck_macro_key
Check MACRO_API_KEY.
ade_bench/cli/ab/check.py:128
↓ 1 callersFunctioncheck_openai_key
Check OPENAI_API_KEY.
ade_bench/cli/ab/check.py:148
↓ 1 callersFunctioncheck_task_exists
Check if task exists in tasks directory.
scripts_python/create_sandbox.py:16
↓ 1 callersFunctioncompare_tables
Compare two parquet files and return a structured diff.
shared/scripts/compare_tables.py:9
↓ 1 callersMethodconvert_all
Convert all discovered DuckDB files to Snowflake databases.
scripts_python/migrate_duckdb_to_snowflake.py:306
← previousnext →101–200 of 626, ranked by callers