MCPcopy Create free account

hub / github.com/SakanaAI/AI-Scientist-v2 / functions

Functions307 in github.com/SakanaAI/AI-Scientist-v2

↓ 1 callersMethoduse_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
Methodabsorb_plot_exec_result
Absorb the result of executing the plotting code from this node.
ai_scientist/treesearch/journal.py:178
Methodappend
(self, node: Node)
ai_scientist/treesearch/journal.py:341
Methodas_openai_tool_dict
(self)
ai_scientist/treesearch/backend/utils.py:116
Functionasync_wrapper
(*args, **kwargs)
ai_scientist/utils/token_tracker.py:145
Methodbuggy_nodes
Return a list of nodes that are considered buggy by the agent.
ai_scientist/treesearch/journal.py:385
Methodchild
(childNode)
ai_scientist/treesearch/utils/viz_templates/template.js:129
Functioncleanup
()
ai_scientist/treesearch/perform_experiments_bfts_with_agentmanager.py:73
Methodcolor
()
ai_scientist/treesearch/utils/viz_templates/template.js:163
Methodconstructor
(x, y, id, isRoot = false)
ai_scientist/treesearch/utils/viz_templates/template.js:36
Methodconstructor
(parent, child, isLeft, isRight)
ai_scientist/treesearch/utils/viz_templates/template.js:139
Functioncreate_client
Create client for vision-language model.
ai_scientist/vlm.py:195
Methoddebug_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
Methoddraft_nodes
Return a list of nodes representing intial coding drafts
ai_scientist/treesearch/journal.py:380
Functiondump_json
(obj: dataclasses_json.DataClassJsonMixin, path: Path)
ai_scientist/treesearch/utils/serialize.py:34
Methodencode_image_to_base64
(image_path)
ai_scientist/treesearch/parallel_agent.py:902
Functionexec_callback
(*args, **kwargs)
ai_scientist/treesearch/perform_experiments_bfts_with_agentmanager.py:95
Functionextract_jsons
Extract all JSON objects from the text. Caveat: This function cannot handle nested JSON objects.
ai_scientist/treesearch/utils/response.py:21
Methodflush
(self)
ai_scientist/treesearch/interpreter.py:77
Functiongenerate
(cfg, jou: Journal, out_path: Path)
ai_scientist/treesearch/utils/tree_export.py:376
Functiongenerate
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
Methodgenerate_nb_trace
Generate a trace of the interactive session in IPython format.
ai_scientist/treesearch/journal.py:345
Methodgenerate_summary_old
(self, include_code: bool = False)
ai_scientist/treesearch/journal.py:550
Functionget_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
Methodget_dataset_value
Get the metric value for a specific dataset
ai_scientist/treesearch/utils/metric.py:97
Methodget_mean_value
Get the mean value across all datasets (or single value if not multi-dataset)
ai_scientist/treesearch/utils/metric.py:103
Methodget_metric_history
Return a list of all metric values in the journal.
ai_scientist/treesearch/journal.py:416
Methodgood_nodes
Return a list of nodes that are not considered buggy by the agent.
ai_scientist/treesearch/journal.py:390
Methodis_leaf
Check if the node is a leaf node in the solution tree.
ai_scientist/treesearch/journal.py:192
Methodis_worst
True if the metric value is the worst possible value.
ai_scientist/treesearch/utils/metric.py:83
Functionjournal2report
Generate a report from a journal, the report will be in markdown format.
ai_scientist/treesearch/journal2report.py:6
Functionload_json
(path: Path, cls: Type[G])
ai_scientist/treesearch/utils/serialize.py:55
Functionmake_llm_call
(client, model, temperature, system_message, prompt)
ai_scientist/vlm.py:53
Functionon_backoff
(details: Dict)
ai_scientist/tools/semantic_scholar.py:12
Methodopenai_tool_choice_dict
(self)
ai_scientist/treesearch/backend/utils.py:127
Functionparse_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
Functionprepare_vlm_prompt
(msg, image_paths, max_images)
ai_scientist/vlm.py:121
Functionprint_cfg
(cfg: Config)
ai_scientist/treesearch/utils/config.py:179
Functionprint_time
()
launch_scientist_bfts.py:31
Functionprocess_stage
(idx, stage_tuple)
ai_scientist/treesearch/log_summarization.py:302
Functionquery
( system_message: str | None, user_message: str | None, func_spec: FunctionSpec | None = None,
ai_scientist/treesearch/backend/backend_openai.py:31
Functionquery
( system_message: str | None, user_message: str | None, func_spec: FunctionSpec | None = None,
ai_scientist/treesearch/backend/backend_anthropic.py:21
Functionredirect_stdout_stderr_to_file
(log_file_path)
launch_scientist_bfts.py:168
Methodrender
(p5)
ai_scientist/treesearch/utils/viz_templates/template.js:167
Methodreset
Reset all token counts and interactions.
ai_scientist/utils/token_tracker.py:99
Methodsave_experiment_notes
Save experimental notes and summaries to files
ai_scientist/treesearch/journal.py:565
FunctionsetBackgroundColor
(color)
ai_scientist/treesearch/utils/viz_templates/template.js:684
Methodstage_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
MethodstartAnimation
()
ai_scientist/treesearch/utils/viz_templates/template.js:68
Functionsync_wrapper
(*args, **kwargs)
ai_scientist/utils/token_tracker.py:185
Methodterm_out
Get the terminal output of the code execution (after truncating it).
ai_scientist/treesearch/journal.py:187
← previousnext →201–300 of 307, ranked by callers