Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SakanaAI/AI-Scientist-v2
/ functions
Functions
307 in github.com/SakanaAI/AI-Scientist-v2
⨍
Functions
307
◇
Types & classes
36
↓ 1 callers
Method
use_tool
Abstract method that should be implemented by subclasses to define the functionality of the tool.
ai_scientist/tools/base_tool.py:26
Method
__deepcopy__
(self, memo)
ai_scientist/treesearch/journal.py:128
Method
__enter__
(self)
ai_scientist/treesearch/parallel_agent.py:2333
Method
__eq__
(self, other)
ai_scientist/treesearch/journal.py:196
Method
__eq__
(self, other: Any)
ai_scientist/treesearch/utils/metric.py:61
Method
__eq__
Compare equality of metric values
ai_scientist/treesearch/utils/metric.py:237
Method
__exit__
(self, exc_type, exc_val, exc_tb)
ai_scientist/treesearch/parallel_agent.py:2367
Method
__getitem__
(self, idx: int)
ai_scientist/treesearch/journal.py:367
Method
__getitem__
(self, idx)
ai_scientist/ideas/i_cant_believe_its_not_betterrealworld.py:151
Method
__getitem__
(self, idx)
ai_scientist/ideas/i_cant_believe_its_not_better.py:183
Method
__getstate__
Return state for pickling
ai_scientist/treesearch/journal.py:145
Method
__gt__
True if self is a _better_ (not necessarily larger) metric value than other
ai_scientist/treesearch/utils/metric.py:34
Method
__gt__
(self, other)
ai_scientist/treesearch/utils/metric.py:171
Method
__hash__
(self)
ai_scientist/treesearch/journal.py:199
Method
__init__
( self, name: str = "SearchSemanticScholar", description: str = ( "Search
ai_scientist/tools/semantic_scholar.py:20
Method
__init__
(self, name: str, description: str, parameters: List[Dict[str, Any]])
ai_scientist/tools/base_tool.py:20
Method
__init__
Token counts for prompt, completion, reasoning, and cached. Reasoning tokens are included in completion tokens. Cached tokens
ai_scientist/utils/token_tracker.py:11
Method
__init__
(self, queue)
ai_scientist/treesearch/interpreter.py:71
Method
__init__
Simulates a standalone Python REPL with an execution time limit. Args: working_dir (Path | str): working directory of th
ai_scientist/treesearch/interpreter.py:82
Method
__init__
(self, name: str, description: str, code: str, base_node: Node)
ai_scientist/treesearch/parallel_agent.py:226
Method
__init__
(self, name: str, description: str)
ai_scientist/treesearch/parallel_agent.py:241
Method
__init__
(self, name: str, description: str)
ai_scientist/treesearch/parallel_agent.py:249
Method
__init__
( self, task_desc, cfg, memory_summary=None, evaluation_metrics=None,
ai_scientist/treesearch/parallel_agent.py:257
Method
__init__
( self, task_desc: str, cfg: Config, journal: Journal, stage_name=None
ai_scientist/treesearch/parallel_agent.py:1143
Method
__init__
(self, task_desc: str, cfg: Any, workspace_dir: Path)
ai_scientist/treesearch/agent_manager.py:124
Method
__init__
(self, hf_dataset, transform=None)
ai_scientist/ideas/i_cant_believe_its_not_betterrealworld.py:144
Method
__init__
(self, hf_dataset, transform=None)
ai_scientist/ideas/i_cant_believe_its_not_better.py:176
Method
__len__
Return the number of nodes in the journal.
ai_scientist/treesearch/journal.py:370
Method
__len__
(self)
ai_scientist/ideas/i_cant_believe_its_not_betterrealworld.py:148
Method
__len__
(self)
ai_scientist/ideas/i_cant_believe_its_not_better.py:180
Method
__post_init__
(self)
ai_scientist/treesearch/journal.py:120
Method
__post_init__
(self)
ai_scientist/treesearch/utils/metric.py:26
Method
__post_init__
(self)
ai_scientist/treesearch/backend/utils.py:111
Method
__repr__
(self)
ai_scientist/treesearch/utils/metric.py:64
Method
__repr__
(self)
ai_scientist/treesearch/utils/metric.py:336
Method
__setstate__
Set state during unpickling
ai_scientist/treesearch/journal.py:153
Method
__str__
(self)
ai_scientist/treesearch/utils/metric.py:67
Method
__str__
(self)
ai_scientist/treesearch/utils/metric.py:339
Method
_create_stage_analysis_prompt
Create detailed prompt to determine next stage configuration
ai_scientist/treesearch/agent_manager.py:831
Method
_evaluate_stage_progression
Evaluate whether experiment is ready for next stage
ai_scientist/treesearch/agent_manager.py:1151
Method
_get_response
Get structured response from LLM for stage configuration. Args: prompt: The analysis prompt to send to the LLM Returns:
ai_scientist/treesearch/agent_manager.py:978
Method
_process_node_wrapper
Wrapper function that creates a fresh environment for each process
ai_scientist/treesearch/parallel_agent.py:1410
Method
_prompt_ablation_resp_fmt
(self)
ai_scientist/treesearch/parallel_agent.py:442
Method
_prompt_debug_resp_fmt
(self)
ai_scientist/treesearch/parallel_agent.py:418
Method
_prompt_environment
(self)
ai_scientist/treesearch/parallel_agent.py:274
Method
_prompt_hyperparam_tuning_resp_fmt
(self)
ai_scientist/treesearch/parallel_agent.py:430
Method
_prompt_impl_guideline
(self)
ai_scientist/treesearch/parallel_agent.py:298
Method
_prompt_resp_fmt
(self)
ai_scientist/treesearch/parallel_agent.py:397
Method
_run_session
( self, code_inq: Queue, result_outq: Queue, event_outq: Queue )
ai_scientist/treesearch/interpreter.py:130
Method
_save_stage_summary
Save comprehensive stage completion summary
ai_scientist/treesearch/agent_manager.py:943
Method
absorb_plot_exec_result
Absorb the result of executing the plotting code from this node.
ai_scientist/treesearch/journal.py:178
Method
append
(self, node: Node)
ai_scientist/treesearch/journal.py:341
Method
as_openai_tool_dict
(self)
ai_scientist/treesearch/backend/utils.py:116
Function
async_wrapper
(*args, **kwargs)
ai_scientist/utils/token_tracker.py:145
Method
buggy_nodes
Return a list of nodes that are considered buggy by the agent.
ai_scientist/treesearch/journal.py:385
Method
child
(childNode)
ai_scientist/treesearch/utils/viz_templates/template.js:129
Function
cleanup
()
ai_scientist/treesearch/perform_experiments_bfts_with_agentmanager.py:73
Method
color
()
ai_scientist/treesearch/utils/viz_templates/template.js:163
Method
constructor
(x, y, id, isRoot = false)
ai_scientist/treesearch/utils/viz_templates/template.js:36
Method
constructor
(parent, child, isLeft, isRight)
ai_scientist/treesearch/utils/viz_templates/template.js:139
Function
create_client
Create client for vision-language model.
ai_scientist/vlm.py:195
Method
debug_depth
Length of the current debug path - 0 if the node is not a debug node (parent is not buggy) - 1 if the parent is buggy but the
ai_scientist/treesearch/journal.py:203
Method
draft_nodes
Return a list of nodes representing intial coding drafts
ai_scientist/treesearch/journal.py:380
Function
dump_json
(obj: dataclasses_json.DataClassJsonMixin, path: Path)
ai_scientist/treesearch/utils/serialize.py:34
Method
encode_image_to_base64
(image_path)
ai_scientist/treesearch/parallel_agent.py:902
Function
exec_callback
(*args, **kwargs)
ai_scientist/treesearch/perform_experiments_bfts_with_agentmanager.py:95
Function
extract_jsons
Extract all JSON objects from the text. Caveat: This function cannot handle nested JSON objects.
ai_scientist/treesearch/utils/response.py:21
Method
flush
(self)
ai_scientist/treesearch/interpreter.py:77
Function
generate
(cfg, jou: Journal, out_path: Path)
ai_scientist/treesearch/utils/tree_export.py:376
Function
generate
Generate a textual preview of a directory, including an overview of the directory structure and previews of individual files
ai_scientist/treesearch/utils/data_preview.py:121
Method
generate_nb_trace
Generate a trace of the interactive session in IPython format.
ai_scientist/treesearch/journal.py:345
Method
generate_summary_old
(self, include_code: bool = False)
ai_scientist/treesearch/journal.py:550
Function
get_batch_responses_from_vlm
Get multiple responses from vision-language model for the same input. Args: msg: Text message to send image_paths: Path(s) to ima
ai_scientist/vlm.py:251
Method
get_dataset_value
Get the metric value for a specific dataset
ai_scientist/treesearch/utils/metric.py:97
Method
get_mean_value
Get the mean value across all datasets (or single value if not multi-dataset)
ai_scientist/treesearch/utils/metric.py:103
Method
get_metric_history
Return a list of all metric values in the journal.
ai_scientist/treesearch/journal.py:416
Method
good_nodes
Return a list of nodes that are not considered buggy by the agent.
ai_scientist/treesearch/journal.py:390
Method
is_leaf
Check if the node is a leaf node in the solution tree.
ai_scientist/treesearch/journal.py:192
Method
is_worst
True if the metric value is the worst possible value.
ai_scientist/treesearch/utils/metric.py:83
Function
journal2report
Generate a report from a journal, the report will be in markdown format.
ai_scientist/treesearch/journal2report.py:6
Function
load_json
(path: Path, cls: Type[G])
ai_scientist/treesearch/utils/serialize.py:55
Function
make_llm_call
(client, model, temperature, system_message, prompt)
ai_scientist/vlm.py:53
Function
on_backoff
(details: Dict)
ai_scientist/tools/semantic_scholar.py:12
Method
openai_tool_choice_dict
(self)
ai_scientist/treesearch/backend/utils.py:127
Function
parse_markdown_to_dict
Reads a file that contains lines of the form: "Key": "Value", "Another Key": "Another Value", ... including possibl
ai_scientist/treesearch/utils/serialize.py:60
Function
prepare_vlm_prompt
(msg, image_paths, max_images)
ai_scientist/vlm.py:121
Function
print_cfg
(cfg: Config)
ai_scientist/treesearch/utils/config.py:179
Function
print_time
()
launch_scientist_bfts.py:31
Function
process_stage
(idx, stage_tuple)
ai_scientist/treesearch/log_summarization.py:302
Function
query
( system_message: str | None, user_message: str | None, func_spec: FunctionSpec | None = None,
ai_scientist/treesearch/backend/backend_openai.py:31
Function
query
( system_message: str | None, user_message: str | None, func_spec: FunctionSpec | None = None,
ai_scientist/treesearch/backend/backend_anthropic.py:21
Function
redirect_stdout_stderr_to_file
(log_file_path)
launch_scientist_bfts.py:168
Method
render
(p5)
ai_scientist/treesearch/utils/viz_templates/template.js:167
Method
reset
Reset all token counts and interactions.
ai_scientist/utils/token_tracker.py:99
Method
save_experiment_notes
Save experimental notes and summaries to files
ai_scientist/treesearch/journal.py:565
Function
setBackgroundColor
(color)
ai_scientist/treesearch/utils/viz_templates/template.js:684
Method
stage_name
Return the stage of the node: - "stage" if the node is an initial solution draft - "debug" if the node is the result of a deb
ai_scientist/treesearch/journal.py:159
Method
startAnimation
()
ai_scientist/treesearch/utils/viz_templates/template.js:68
Function
sync_wrapper
(*args, **kwargs)
ai_scientist/utils/token_tracker.py:185
Method
term_out
Get the terminal output of the code execution (after truncating it).
ai_scientist/treesearch/journal.py:187
← previous
next →
201–300 of 307, ranked by callers