MCPcopy Create free account

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

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

↓ 1 callersFunctionmigrate_field
Apply standard migration logic for a field. Priority order (highest to lowest): 1. Task-specific migration (experiment + task) 2. Ex
scripts_python/analyze.py:73
↓ 1 callersFunctionmigrate_row
Apply migrations to normalize a row's structure based on experiment and task config.
scripts_python/analyze.py:121
↓ 1 callersMethodmodel_name_from_agent_name
(model_name, agent_name)
ade_bench/agents/agent_name.py:16
↓ 1 callersFunctionparse_args
Parse command line arguments.
shared/scripts/run_sql.py:47
↓ 1 callersFunctionparse_args
Parse command line arguments.
scripts_python/analyze.py:273
↓ 1 callersFunctionparse_diff_log
Parse the file diff log content into structured data.
scripts_python/render_file_diffs.py:54
↓ 1 callersFunctionparse_folder_timestamp
(folder_name: str)
scripts_python/profiling_statistics.py:143
↓ 1 callersFunctionparse_folder_timestamp
(folder_name: str)
scripts_python/generate_db_from_profiling_summaries.py:137
↓ 1 callersMethodparse_log_file
Parse the agent log file and extract structured information. Args: log_path: Path to the log file to parse Retu
ade_bench/agents/log_formatter.py:18
↓ 1 callersMethodparse_log_file
Parse the Claude Code agent log file and extract structured information.
ade_bench/agents/installed_agents/claude_code/log_formatter.py:140
↓ 1 callersFunctionprint_ade_summary
()
ade_bench/cli/ab/check.py:42
↓ 1 callersFunctionprint_failure
()
ade_bench/cli/ab/check.py:28
↓ 1 callersFunctionprint_success
()
ade_bench/cli/ab/check.py:35
↓ 1 callersFunctionread_sql
Read SQL from file or stdin.
shared/scripts/run_sql.py:189
↓ 1 callersFunctionrender_diff_html
Render a comparison result as a self-contained HTML page.
shared/scripts/compare_tables.py:474
↓ 1 callersFunctionresolve_relations
Resolve failing test names to (actual, expected) relation pairs.
shared/scripts/detect_failing_equality_tests.py:25
↓ 1 callersFunctionrun_profiling_and_import
( n_runs: int = 200, max_retries: int = 10, profiling_script: str = "./scripts_bash/start_profilin
scripts_python/profiling_statistics.py:619
↓ 1 callersFunctionrun_profiling_session
(script_path: str, extra_args: list)
scripts_python/profiling_statistics.py:605
↓ 1 callersMethodsave_diff_log
Save the diff log to file.
ade_bench/handlers/file_diff_handler.py:307
↓ 1 callersMethodscan
Scan task.yaml files and return matching tasks. Args: task_ids: Only include tasks whose task_id is in this list. db_
ade_bench/utils/task_scanner.py:66
↓ 1 callersFunctionsend_terminal_notification
Send a terminal notification when a run completes. Uses OSC 9 (iTerm2/Windows Terminal) and OSC 777 (Gnome Terminal/VTE) for desktop notifica
ade_bench/utils/notify.py:7
↓ 1 callersMethodset_variant_config
Set the variant configuration for the current task.
ade_bench/agents/sage_agent.py:68
↓ 1 callersFunctionsetup_base_files
Setup base files - copy setup files and run scripts.
ade_bench/setup/base_setup.py:10
↓ 1 callersFunctionsetup_dbt_project
Setup dbt project by copying project files. Returns: Tuple of (success, error_message). error_message is empty string on success.
ade_bench/setup/dbt_setup.py:59
↓ 1 callersFunctionsetup_duckdb
Setup DuckDB by copying the database file. Returns: Tuple of (success, error_message). error_message is empty string on success.
ade_bench/setup/duckdb_setup.py:9
↓ 1 callersFunctionsetup_logger
(name: str)
ade_bench/utils/logger.py:141
↓ 1 callersFunctionsetup_migration
Setup migration by copying migration files.
ade_bench/setup/migration_setup.py:9
↓ 1 callersFunctionsetup_snowflake
Setup Snowflake by cloning database and creating user/role. Args: terminal: Terminal instance session: Tmux session task_
ade_bench/setup/snowflake_setup.py:173
↓ 1 callersMethodsetup_task
Setup a task for the given variant.
ade_bench/setup/setup_orchestrator.py:37
↓ 1 callersMethodshould_exclude_file
Check if file should be excluded from processing.
scripts_python/duckdb_utils.py:102
↓ 1 callersMethodshould_include_file
Check if file should be included in processing.
scripts_python/duckdb_utils.py:118
↓ 1 callersFunctionspin_up_terminal
( client_container_name: str, client_image_name: str, docker_compose_path: Path, docker_name_p
ade_bench/terminal/terminal.py:103
↓ 1 callersMethodstop_services_only
Stop docker compose services but keep containers alive for debugging.
ade_bench/terminal/terminal.py:83
↓ 1 callersMethodstrip_ansi_codes
Remove ANSI color codes from text.
ade_bench/agents/installed_agents/claude_code/log_formatter.py:27
↓ 1 callersMethodto_env_dict
(self, include_os_env: bool = False)
ade_bench/utils/env_model.py:7
↓ 1 callersFunctionupdate_dbt_config
Update dbt_project.yml and profiles.yml files based on variant configuration.
scripts_python/create_sandbox.py:231
↓ 1 callersMethodupdate_task_from_dict
Update a specific task's row from log data dictionary.
ade_bench/utils/logger.py:62
↓ 1 callersFunctionwipe_sandbox_directory
Wipe the contents of the sandbox directory.
scripts_python/create_sandbox.py:32
↓ 1 callersFunctionwrite_results_tsv
Write benchmark results to a TSV file. Args: results: BenchmarkResults object containing all trial results output_path: Path
ade_bench/utils/results_writer.py:114
Method__init__
(self, experiment_dir: Path)
scripts_python/generate_results_html.py:23
Method__init__
Initialize the DuckDB extractor. Args: source_dir: Path to directory containing DuckDB files. If None, uses default.
scripts_python/duckdb_utils.py:19
Method__init__
(self)
scripts_python/migrate_duckdb_to_snowflake.py:53
Method__init__
Runs the Terminal-Bench harness. Args: dataset_path: The path to the dataset. output_path: The path to the o
ade_bench/harness.py:48
Method__init__
Initialize the scanner. Args: tasks_dir: Path to the directory containing task subdirectories.
ade_bench/utils/task_scanner.py:53
Method__init__
Initialize the dataset. Args: dataset_path: Path to the dataset directorygs task_ids: Optional list of specific task
ade_bench/utils/dataset.py:10
Method__init__
(self)
ade_bench/utils/logger.py:16
Method__init__
( self, shared_db_dir: Optional[Path] = None, logger: Optional[logging.Logger] = None )
ade_bench/database/pool_manager.py:48
Method__init__
(self, task_name: str = "unknown")
ade_bench/parsers/base_parser.py:26
Method__init__
Initialize DbtParser with explicit parser type. Args: parser_type: Either "dbt" or "dbt-fusion" to specify which format
ade_bench/parsers/dbt_parser.py:34
Method__init__
( self, client_container_name: str, client_image_name: str, docker_compose_pat
ade_bench/terminal/terminal.py:11
Method__init__
( self, client_container_name: str, client_image_name: str, docker_compose_pat
ade_bench/terminal/docker_compose_manager.py:42
Method__init__
( self, session_name: str, container: Container, commands_path: Path | None =
ade_bench/terminal/tmux_session.py:23
Method__init__
( self, logger=None, terminal=None, session=None, file_diff_handler=No
ade_bench/setup/setup_orchestrator.py:20
Method__init__
(self, **kwargs)
ade_bench/llms/base_llm.py:17
Method__init__
(self)
ade_bench/handlers/file_diff_handler.py:19
Method__init__
( self, timestamp: datetime, directory: str, files: Dict[str, str], co
ade_bench/handlers/file_diff_handler.py:47
Method__init__
(self, before: FileSnapshot, after: FileSnapshot)
ade_bench/handlers/file_diff_handler.py:99
Method__init__
( self, output_dir: Path, enabled: bool = True, exclude_paths: List[str] = Non
ade_bench/handlers/file_diff_handler.py:192
Method__init__
( self, trial_name: str, input_path: Path, output_path: Path | None = None,
ade_bench/handlers/trial_handler.py:174
Method__init__
Initialize the AsciinemaHandler. Args: markers: List of tuples containing (timestamp, label) for each marker
ade_bench/handlers/asciinema_handler.py:11
Method__init__
(self, config_path: Path)
ade_bench/plugins/loader.py:12
Method__init__
( self, dataset_path: Path, task_ids: list[str] | None = None, **kwargs, )
ade_bench/agents/sage_agent.py:29
Method__init__
(cls, name, bases, dct)
ade_bench/agents/base_agent.py:13
Method__init__
(self, **kwargs)
ade_bench/agents/none_agent.py:13
Method__init__
(self, agent_name: AgentName)
ade_bench/agents/agent_factory.py:43
Method__init__
( self, model_name: str | None = None, allowed_tools: list[str] | None = None,
ade_bench/agents/installed_agents/abstract_installed_agent.py:31
Method__init__
(self, additional_args: str = "", model_name: str | None = None, **kwargs)
ade_bench/agents/installed_agents/macro/macro_agent.py:20
Method__init__
(self, **kwargs)
ade_bench/agents/installed_agents/claude_code/claude_code_agent.py:19
Method__init__
(self, **kwargs)
ade_bench/agents/installed_agents/gemini_cli/gemini_cli_agent.py:20
Method__init__
(self, **kwargs)
ade_bench/agents/installed_agents/openai_codex/openai_codex_agent.py:24
Method__iter__
Iterate over the tasks in the dataset.
ade_bench/utils/dataset.py:152
Method__len__
Get the number of tasks in the dataset.
ade_bench/utils/dataset.py:156
Method__str__
Human-readable representation for logging.
ade_bench/utils/timeout_manager.py:39
Method_cProfile_output_path
(self)
ade_bench/harness.py:162
Function_custom_rich_format_error
Custom error formatter that adds a blank line before the error output.
ade_bench/cli/ab/main.py:30
Method_default_docker_compose_path
(self)
ade_bench/handlers/trial_handler.py:288
Method_default_run_tests_path
(self)
ade_bench/handlers/trial_handler.py:292
Method_defaults_path
(self)
ade_bench/handlers/trial_handler.py:284
Method_env
(self)
tests/agents/installed_agents/test_abstract_installed_agent.py:15
Method_env
Environment variables to use when running the agent (e.g. ANTHROPIC_API_KEY).
ade_bench/agents/installed_agents/abstract_installed_agent.py:46
Method_env
(self)
ade_bench/agents/installed_agents/macro/macro_agent.py:26
Method_env
(self)
ade_bench/agents/installed_agents/claude_code/claude_code_agent.py:25
Method_env
(self)
ade_bench/agents/installed_agents/gemini_cli/gemini_cli_agent.py:25
Method_env
(self)
ade_bench/agents/installed_agents/openai_codex/openai_codex_agent.py:29
Method_execute_single_trial
Execute a single task and return its results. Args: task_path: Path to the task directory task_key: Optional key for
ade_bench/harness.py:1369
Method_format_time
Format time in seconds to a readable string.
scripts_python/generate_results_html.py:89
Method_get_network_name
(self, container_name: str)
ade_bench/agents/base_agent.py:59
Method_install_agent_script
(self)
tests/agents/installed_agents/test_abstract_installed_agent.py:19
Method_install_agent_script
Script to install the agent in the container.
ade_bench/agents/installed_agents/abstract_installed_agent.py:54
Method_install_agent_script
Use a local or default Macro install script depending on environment.
ade_bench/agents/installed_agents/macro/macro_agent.py:32
Method_install_agent_script
(self)
ade_bench/agents/installed_agents/claude_code/claude_code_agent.py:31
Method_install_agent_script
(self)
ade_bench/agents/installed_agents/gemini_cli/gemini_cli_agent.py:31
Method_install_agent_script
(self)
ade_bench/agents/installed_agents/openai_codex/openai_codex_agent.py:35
Method_log_output_path
(self)
ade_bench/harness.py:153
Method_normalize_solution_seeds
Normalize mixed str/dict solution_seeds from YAML into dicts.
ade_bench/utils/task_scanner.py:35
Method_panes_path
(self)
ade_bench/handlers/trial_handler.py:374
Method_parse_agent_output
Parse Macro agent output using the Macro parser.
ade_bench/agents/installed_agents/macro/macro_agent.py:61
Method_parse_agent_output
Parse Claude agent output to extract metrics.
ade_bench/agents/installed_agents/claude_code/claude_code_agent.py:55
Method_parse_agent_output
Parse Gemini agent output to extract metrics.
ade_bench/agents/installed_agents/gemini_cli/gemini_cli_agent.py:52
Method_parse_agent_output
Parse Codex agent output to extract metrics.
ade_bench/agents/installed_agents/openai_codex/openai_codex_agent.py:72
← previousnext →301–400 of 626, ranked by callers