MCPcopy Create free account

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

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

↓ 76 callersFunctionopen
Open a specific run in your browser.
ade_bench/cli/ab/runs.py:66
↓ 56 callersFunctionlog_harness_info
( logger: logging.Logger, task: str, stage: str, message: str, timestamp: Optional[datetim
ade_bench/utils/logger.py:239
↓ 36 callersMethodrun
Run the benchmark with all configured plugin sets.
ade_bench/harness.py:1534
↓ 34 callersFunctionwrite_parquet
Helper: create a parquet file from a SQL query.
tests/scripts/test_compare_tables.py:18
↓ 25 callersMethodcopy_to_container
( self, paths: list[Path] | Path, container_dir: str | None = None, container_
ade_bench/terminal/terminal.py:88
↓ 19 callersFunctionesc
(s)
shared/scripts/compare_tables.py:478
↓ 17 callersFunctionload_module
()
tests/scripts/test_compare_tables.py:8
↓ 15 callersMethodsend_keys
Execute a command in the tmux session. Args: keys (str): The keys to send to the tmux session. block (bool):
ade_bench/terminal/tmux_session.py:225
↓ 14 callersMethodload
Load the plugin sets configuration from YAML.
ade_bench/plugins/loader.py:16
↓ 13 callersFunctionclean_text_for_spreadsheet
Clean text for spreadsheet compatibility.
scripts_python/extract_task_details.py:14
↓ 13 callersFunctionlist
List recent benchmark runs.
ade_bench/cli/ab/runs.py:12
↓ 12 callersMethodget_setting
(key, default=None)
ade_bench/config.py:14
↓ 9 callersMethod_configure_mcp_servers
Configure MCP servers after agent installation.
ade_bench/agents/installed_agents/abstract_installed_agent.py:97
↓ 8 callersFunction_make_session
Create a mock TmuxSession with configurable exec_run results.
tests/agents/installed_agents/test_abstract_installed_agent.py:26
↓ 8 callersMethodresolve_plugin_sets
Resolve which plugin sets to use for a run. Args: plugin_set_names: Explicit plugin set names from --plugin-set, or None for defa
ade_bench/plugins/loader.py:27
↓ 8 callersFunctionwrite_parquet
(tmpdir: str, name: str, sql: str)
tests/scripts/test_compare_tables_html.py:19
↓ 7 callersMethodcapture_snapshot
Capture a snapshot of files in the container directory using tar archive.
ade_bench/handlers/file_diff_handler.py:223
↓ 7 callersFunctioncopy_item
Generic function to copy files or directories.
scripts_python/create_sandbox.py:74
↓ 7 callersFunctionload_module
()
tests/scripts/test_detect_failing_equality_tests.py:5
↓ 6 callersMethod_should_exclude_path
Check if a file path should be excluded from diffing.
ade_bench/handlers/file_diff_handler.py:216
↓ 6 callersFunctionadd_failure
(message)
ade_bench/cli/ab/check.py:20
↓ 6 callersMethodinstall
Install skills from the plugin set into the container. Args: plugin_set: The plugin set configuration terminal: The D
ade_bench/plugins/skills_handler.py:13
↓ 6 callersMethodis_compatible_with_agent
Check if this plugin set is compatible with the given agent.
ade_bench/harness_models.py:270
↓ 6 callersMethodkill_session
Kill the tmux session and all processes running in it.
ade_bench/terminal/tmux_session.py:302
↓ 5 callersFunction_clean_text
Remove line breaks and normalize whitespace for display.
ade_bench/cli/ab/tasks.py:16
↓ 5 callersFunctionload_module
()
tests/scripts/test_compare_tables_html.py:9
↓ 5 callersFunctionload_module
()
tests/scripts/test_dump_tables.py:8
↓ 5 callersMethodparse
(self, content: str)
ade_bench/parsers/dbt_parser.py:205
↓ 4 callersFunction_copy_config_file
Helper to copy a configuration file to the container.
ade_bench/setup/agent_setup.py:11
↓ 4 callersMethod_load_task
Load a single task, applying all filters and key selection logic.
ade_bench/utils/dataset.py:121
↓ 4 callersMethod_run_docker_compose_command
Run a docker-compose command with the appropriate environment variables.
ade_bench/terminal/docker_compose_manager.py:74
↓ 4 callersFunction_variant_field
Extract a comma-separated unique set of a variant field.
ade_bench/cli/ab/tasks.py:33
↓ 4 callersFunctionadd_success
(message)
ade_bench/cli/ab/check.py:24
↓ 4 callersFunctioncheck_provider_key
(keyname)
ade_bench/cli/ab/check.py:99
↓ 4 callersFunctioncreate_histogram
(data: np.ndarray, title: str, xlabel: str, output_path: Path)
scripts_python/generate_function_charts_from_db.py:14
↓ 4 callersFunctioncreate_histogram
(data: np.ndarray, title: str, xlabel: str, output_path: Path)
scripts_python/profiling_statistics.py:317
↓ 4 callersMethodgenerate_all
Generate all HTML pages for the experiment.
scripts_python/generate_results_html.py:28
↓ 4 callersMethodget_by_name
Get a plugin set by name.
ade_bench/harness_models.py:291
↓ 4 callersMethodget_timeouts_for_task
Resolve all timeouts for a task, applying overrides as needed. Task-specific overrides take precedence over global config.
ade_bench/utils/timeout_manager.py:68
↓ 4 callersMethodsend_command
(self, command: TerminalCommand)
ade_bench/terminal/tmux_session.py:272
↓ 3 callersFunction_execute_queries
Execute multiple SQL queries separated by semicolons.
ade_bench/setup/snowflake_setup.py:19
↓ 3 callersMethod_get_repo_root
Get the git repository root directory. This handles both regular repos and worktrees by finding the actual repository root, not the w
ade_bench/handlers/trial_handler.py:240
↓ 3 callersMethod_run_agent_commands
(self, task_prompt: str)
ade_bench/agents/installed_agents/openai_codex/openai_codex_agent.py:38
↓ 3 callersMethodcapture_pane
(self, capture_entire: bool = False)
ade_bench/terminal/tmux_session.py:285
↓ 3 callersFunctioncheck_file_exists
(folder_path: Path, filename: str)
scripts_python/profiling_statistics.py:150
↓ 3 callersFunctioncheck_file_exists
(folder_path: Path, filename: str)
scripts_python/generate_db_from_profiling_summaries.py:144
↓ 3 callersFunctioncreate_scatterplot
( x_data: np.ndarray, y_data: np.ndarray, title: str, xlabel: str, ylabel: str, output_path: Path )
scripts_python/generate_function_charts_from_db.py:54
↓ 3 callersFunctioncreate_scatterplot
( x_data: np.ndarray, y_data: np.ndarray, title: str, xlabel: str, ylabel: str, output_path: Path )
scripts_python/profiling_statistics.py:357
↓ 3 callersMethodfind_database_file
Find a database file by name and type.
ade_bench/database/pool_manager.py:110
↓ 3 callersFunctionformat_log_line
(log_data: dict)
ade_bench/utils/logger.py:176
↓ 3 callersMethodfrom_dict
Create diff from dictionary.
ade_bench/handlers/file_diff_handler.py:178
↓ 3 callersFunctiongenerate_task_snowflake_credentials
Generate Snowflake credentials for a specific task (the user created during setup).
ade_bench/setup/setup_utils.py:11
↓ 3 callersMethodget_agent
(self, **kwargs)
ade_bench/agents/agent_factory.py:29
↓ 3 callersMethodget_by_names
Get multiple plugin sets by name. Raises if any not found.
ade_bench/harness_models.py:298
↓ 3 callersFunctionget_or_create_function
(cursor: sqlite3.Cursor, func_tuple: tuple)
scripts_python/profiling_statistics.py:154
↓ 3 callersFunctionget_or_create_function
(cursor: sqlite3.Cursor, func_tuple: tuple)
scripts_python/generate_db_from_profiling_summaries.py:148
↓ 3 callersMethodhandle_phase_diffing
Handle file diffing operations for a specific phase.
ade_bench/handlers/file_diff_handler.py:417
↓ 3 callersMethodinstall_all
Return True if all skills should be installed from this origin.
ade_bench/harness_models.py:253
↓ 3 callersMethodstart
(self)
ade_bench/terminal/tmux_session.py:108
↓ 3 callersMethodstop
(self)
ade_bench/terminal/terminal.py:75
↓ 3 callersFunctionsummarize_results
Generate a JSON summary of benchmark results.
scripts_python/summarize_results.py:10
↓ 3 callersMethodto_dict
Convert diff to dictionary for serialization.
ade_bench/handlers/file_diff_handler.py:167
↓ 2 callersFunction_build_tsv_row
Build a single TSV export row from a TaskInfo and prompt.
ade_bench/cli/ab/tasks.py:39
↓ 2 callersMethod_create_env_setup_file
(self)
ade_bench/agents/installed_agents/abstract_installed_agent.py:70
↓ 2 callersMethod_create_table
Create a new table with current task data.
ade_bench/utils/logger.py:89
↓ 2 callersFunction_fetch_rows
Fetch rows from a table as a list of dicts.
shared/scripts/compare_tables.py:409
↓ 2 callersFunction_get_most_recent_run_with_results
Get the most recent run directory that has results.
ade_bench/cli/ab/save.py:15
↓ 2 callersFunction_get_snowflake_connection
Get Snowflake connection using environment variables for setup operations.
ade_bench/setup/snowflake_setup.py:31
↓ 2 callersMethod_is_enter_key
(self, key: str)
ade_bench/terminal/tmux_session.py:135
↓ 2 callersMethod_is_executing_command
(self, key: str)
ade_bench/terminal/tmux_session.py:142
↓ 2 callersFunction_render_row_table
Render a list of row dicts as an HTML table.
shared/scripts/compare_tables.py:656
↓ 2 callersMethod_save_catalog
Save the database catalog to disk.
ade_bench/database/pool_manager.py:73
↓ 2 callersFunction_save_run
Save a run to the saved directory. Returns True if successful.
ade_bench/cli/ab/save.py:34
↓ 2 callersMethod_tmux_send_keys
(self, keys: list[str])
ade_bench/terminal/tmux_session.py:83
↓ 2 callersFunction_truncate_middle
Truncate text in the middle with ellipsis to fit available space.
ade_bench/cli/ab/tasks.py:25
↓ 2 callersFunction_wrap_html
Wrap content in a full HTML page with VS Code dark theme styling.
shared/scripts/compare_tables.py:676
↓ 2 callersMethod_write_detail_page
Write a detail page using the template.
scripts_python/generate_results_html.py:538
↓ 2 callersMethod_write_marker
Write a single marker event to the output file.
ade_bench/handlers/asciinema_handler.py:87
↓ 2 callersMethodcopy_to_container
( self, paths: list[Path] | Path, container_dir: str | None = None, container_
ade_bench/terminal/tmux_session.py:289
↓ 2 callersMethodcreate_diff
Create a diff between two snapshots.
ade_bench/handlers/file_diff_handler.py:284
↓ 2 callersMethodcreate_session
Create a new tmux session with the given name.
ade_bench/terminal/terminal.py:46
↓ 2 callersFunctionflip_hosts
Flip IS_SUPERHOST for given host IDs. UPDATED_AT stays at 1980-01-01.
tasks/airbnb011/setup/deterministic_mutate.py:25
↓ 2 callersFunctionflip_hosts
Flip IS_SUPERHOST for given host IDs and set UPDATED_AT.
tasks/airbnb010/setup/deterministic_mutate.py:22
↓ 2 callersMethodformat_tool_result
Format tool result output, limiting length.
ade_bench/agents/installed_agents/claude_code/log_formatter.py:107
↓ 2 callersFunctionformat_trial_result
Format a single trial result with calculated statistics. This is a shared utility used by both the TSV writer and summarize_results. Ar
ade_bench/utils/results_writer.py:59
↓ 2 callersFunctiongenerate_html
Generate HTML content from parsed diff phases.
scripts_python/render_file_diffs.py:156
↓ 2 callersMethodget_content_by_hash
Get file content by hash.
ade_bench/handlers/file_diff_handler.py:28
↓ 2 callersMethodget_defaults
Get all plugin sets marked as default.
ade_bench/harness_models.py:287
↓ 2 callersFunctionget_failure_type
Determine the failure type based on the result. Returns: - Empty string if test passed - "eval_error" if dbt compile passed
ade_bench/utils/results_writer.py:15
↓ 2 callersFunctionget_file_content
Get file content from optimized JSON data structure.
scripts_python/render_file_diffs.py:33
↓ 2 callersMethodget_file_content
Get file content by path.
ade_bench/handlers/file_diff_handler.py:64
↓ 2 callersFunctionget_latest_experiment_with_results
Find the most recent experiment directory that has results.json.
scripts_python/utils.py:9
↓ 2 callersMethodget_migration_path
Get the path to a specific migration directory.
ade_bench/handlers/trial_handler.py:506
↓ 2 callersMethodget_parser
(parser_name: ParserName | str, **kwargs)
ade_bench/parsers/parser_factory.py:25
↓ 2 callersMethodget_snapshot_count
Get the number of snapshots captured.
ade_bench/handlers/file_diff_handler.py:374
↓ 2 callersMethodget_snowflake_connection
Get Snowflake connection using environment variables.
scripts_python/migrate_duckdb_to_snowflake.py:89
↓ 2 callersFunctionrender_diff_log_html
Render a file diff log as HTML. Args: diff_log_path: Path to the file_diff_log.txt file task_id: Optional task ID to include in t
scripts_python/render_file_diffs.py:512
↓ 2 callersFunctionsetup_agent_config
Setup agent-specific configuration files and resources.
ade_bench/setup/agent_setup.py:29
↓ 2 callersMethodstart
(self)
ade_bench/terminal/terminal.py:72
↓ 2 callersMethodstart
(self)
ade_bench/terminal/docker_compose_manager.py:125
next →1–100 of 626, ranked by callers