MCPcopy Create free account

hub / github.com/HKUDS/AI-Researcher / functions

Functions951 in github.com/HKUDS/AI-Researcher

↓ 370 callersMethodinfo
(self, *args: str, **kwargs: dict)
research_agent/inno/logger.py:29
↓ 232 callersMethodget
get: get the record by the id. Args: record_id: the id of the record. collection: the name of the collection
research_agent/inno/memory/rag_memory.py:130
↓ 47 callersMethodrun
( self, agent: Agent, messages: List, context_variables: dict = {}, mo
research_agent/inno/core.py:267
↓ 38 callersMethodclose
(self)
research_agent/inno/environment/browser_env.py:569
↓ 28 callersMethodstep
Execute an action in the browser environment and return the observation.
research_agent/inno/environment/browser_env.py:546
↓ 25 callersMethodwrite_temp_log
Write intermediate results to temporary log file
paper_agent/section_composer.py:51
↓ 24 callersMethodchat
(self, prompt: str, system_prompt: str = None, temperature: flo
benchmark_collection/utils/openai_utils.py:101
↓ 22 callersMethodstep
(self, epoch)
examples/gnn_nodeformer/project/experiments/run_enhanced_experiments.py:39
↓ 21 callersMethodsearch
Search Bing and return the results formatted in Markdown. If a Bing API key is available, the API is used to perform the search. If no API key is avai
research_agent/inno/environment/markdown_browser/markdown_search.py:57
↓ 19 callersMethodrun_command
communicate with docker container and execute command, support stream output Args: command: the command to execu
research_agent/inno/environment/docker_env.py:138
↓ 16 callersMethodtrain
Train for specified number of epochs. Args: num_epochs: Number of epochs to train for Returns:
examples/fsq/project/training/trainer.py:73
↓ 14 callersFunctionevaluate
Evaluate model Args: model: Diffusion model data: PyG data object mask: Evaluation mask device: Computing de
examples/gnn_difformer/project/training/train.py:63
↓ 13 callersMethodcount
count: count the number of records in the memery. Args: collection_name: the name of the collection to count.
research_agent/inno/memory/rag_memory.py:157
↓ 13 callersMethodevaluate
(self, user_graph, user_item_graph, item_graph, test_mat, k=20)
examples/hgcl/project/training/trainer.py:40
↓ 13 callersFunctionto_web_obs
(obs)
research_agent/inno/tools/inno_tools/web_tools.py:78
↓ 12 callersMethodregister_page_converter
Register a page text converter.
research_agent/inno/environment/markdown_browser/mdconvert.py:1031
↓ 12 callersFunctionto_web_obs
(obs)
research_agent/inno/tools/web_tools.py:73
↓ 11 callersMethod_set_page_content
Sets the text content of the current page.
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:187
↓ 11 callersFunctiondebug_print
(debug: bool, *args: str, **kwargs: dict)
research_agent/inno/util.py:42
↓ 10 callersMethodadd_node
Add a node, return its node_id if it exists, otherwise create a new node. Parameters: node_name (str): The node name.
research_agent/inno/workflow/flowgraph.py:15
↓ 10 callersMethodevaluate
Evaluate model performance. Args: num_samples: Number of samples for FID calculation Returns:
examples/fsq/project/testing/evaluator.py:97
↓ 10 callersMethodnormalize_title
Normalize title for file naming
paper_agent/section_composer.py:84
↓ 10 callersMethodregister_page_converter
Register a page text converter.
research_agent/inno/environment/mdconvert.py:1037
↓ 9 callersMethod_wrap_title
(self, title: str, color: str = None)
research_agent/inno/logger.py:24
↓ 9 callersMethodload_checkpoint
Load checkpoint data if exists
paper_agent/section_composer.py:73
↓ 9 callersFunctionload_dataset
Load and preprocess graph dataset Args: name (str): Dataset name (Cora, CiteSeer, or PubMed) normalize_features (bool):
examples/gnn_difformer/project/data_processing/dataset.py:6
↓ 9 callersMethodsave_checkpoint
Save checkpoint data
paper_agent/section_composer.py:63
↓ 9 callersFunctionwrap_return_value
(web_obs: WebObservation, action_description: str = "")
research_agent/inno/tools/web_tools.py:96
↓ 9 callersFunctionwrap_return_value
(web_obs: WebObservation, action_description: str = "")
research_agent/inno/tools/inno_tools/web_tools.py:101
↓ 8 callersFunctionaccuracy
Compute accuracy for node classification. Args: y_pred (torch.Tensor): Model predictions y_true (torch.Tensor): Ground truth
examples/gnn_nodeformer/project/testing/metrics.py:5
↓ 8 callersMethodget_checkpoint_path
Get checkpoint directory path for the target paper
paper_agent/section_composer.py:58
↓ 8 callersFunctiontrain_epoch
Train for one epoch Args: model: Diffusion model data: PyG data object optimizer: PyTorch optimizer device:
examples/gnn_difformer/project/training/train.py:23
↓ 7 callersMethod_append_ext
Append a unique non-None, non-empty extension to a list of extensions.
research_agent/inno/environment/mdconvert.py:1009
↓ 7 callersMethod_append_ext
Append a unique non-None, non-empty extension to a list of extensions.
research_agent/inno/environment/markdown_browser/mdconvert.py:1003
↓ 7 callersMethod_write_log
(self, message: str)
research_agent/inno/logger.py:15
↓ 7 callersMethodevaluate
(self)
examples/con_flowmatching/project/training/trainer.py:77
↓ 7 callersFunctionget_data_loaders
Create data loaders for training and testing. Args: data_dir (str): Directory containing dataset batch_size (int): Batch size
examples/fsq/project/data_processing/dataset.py:45
↓ 7 callersFunctionget_train_val_test_split
Get training, validation and test masks Args: data: PyG data object Returns: train_mask: Training mask
examples/gnn_difformer/project/data_processing/dataset.py:30
↓ 7 callersFunctionto_string
(idx)
research_agent/inno/tools/inno_tools/web_tools.py:373
↓ 7 callersMethodtrain_epoch
(self, epoch)
examples/con_flowmatching/project/training/trainer.py:36
↓ 6 callersMethod_wrap_timestamp
(self, timestamp: str, color: bool = True)
research_agent/inno/logger.py:53
↓ 6 callersFunctionfunction_to_json
Converts a Python function into a JSON-serializable dictionary that describes the function's signature, including its name, description,
research_agent/inno/util.py:211
↓ 6 callersFunctionget_data_loaders
Create data loaders for CIFAR-10 dataset.
examples/con_flowmatching/project/data_processing/cifar10.py:7
↓ 6 callersMethodget_random_template
Randomly select a writing template from the templates directory
paper_agent/section_composer.py:98
↓ 6 callersFunctionmetrics
(uids, predictions, topk, test_labels)
examples/dccf/project/model/utils.py:5
↓ 6 callersFunctionscipy_sparse_mat_to_torch_sparse_tensor
(sparse_mx)
examples/dccf/project/model/utils.py:44
↓ 6 callersFunctionsetup_logging
(research_field)
paper_agent/section_composer.py:13
↓ 6 callersFunctionstr_to_bool
convert string to bool
research_agent/constant.py:7
↓ 5 callersMethod_get
(self, metadata: Dict[str, str], keys: List[str], default: Union[str, None] = None)
research_agent/inno/environment/mdconvert.py:354
↓ 5 callersMethod_get
(self, metadata: Dict[str, str], keys: List[str], default: Union[str, None] = None)
research_agent/inno/environment/markdown_browser/mdconvert.py:344
↓ 5 callersFunction_get_browser_state
Get the current state of the browser, including the header and content.
research_agent/inno/tools/file_surfer_tool.py:31
↓ 5 callersMethodconvert_soup
(self, soup: Any)
research_agent/inno/environment/mdconvert.py:129
↓ 5 callersMethodconvert_soup
(self, soup: Any)
research_agent/inno/environment/markdown_browser/mdconvert.py:119
↓ 5 callersFunctionevaluate_model
Evaluate model by generating samples and calculating FID score
examples/con_flowmatching/project/testing/evaluator.py:60
↓ 5 callersFunctionget_data_loaders
(data_dir, batch_size=128, num_workers=4)
examples/rotation_vq/project/data_processing/cifar10.py:44
↓ 5 callersFunctioninit_env_file
初始化.env文件如果不存在
web_ai_researcher.py:585
↓ 5 callersMethodtrain_epoch
Train for one epoch. Args: epoch: Current epoch number Returns: dict: Dictionary contain
examples/fsq/project/training/trainer.py:23
↓ 4 callersMethod_convert
Helper function that converts and HTML string.
research_agent/inno/environment/mdconvert.py:185
↓ 4 callersMethod_convert
Helper function that converts and HTML string.
research_agent/inno/environment/markdown_browser/mdconvert.py:175
↓ 4 callersFunction_get_browser_state
Get the current state of the browser, including the header and content.
research_agent/inno/tools/terminal_tools.py:20
↓ 4 callersMethod_markdown_link
Create a Markdown hyperlink, escaping the URLs as appropriate. Args: anchor: The anchor text of the hyperlink href: T
research_agent/inno/environment/markdown_browser/markdown_search.py:264
↓ 4 callersMethodadd_edge
Add an edge from node u to node v, and add or update node attributes at the same time. Parameters: u (str): The start no
research_agent/inno/workflow/flowgraph.py:44
↓ 4 callersFunctioncombined_loss
Compute combined loss with classification and edge regularization. Args: pred (torch.Tensor): Model predictions labels (torch
examples/gnn_nodeformer/project/training/loss.py:46
↓ 4 callersMethodconvert
(self, local_path, **kwargs)
research_agent/inno/environment/mdconvert.py:448
↓ 4 callersMethodconvert_response
( self, response: requests.Response, **kwargs: Any )
research_agent/inno/environment/markdown_browser/mdconvert.py:911
↓ 4 callersFunctioncreate_adj_matrices
Create adjacency matrices for heterogeneous graphs
examples/hgcl/project/data_processing/dataset.py:16
↓ 4 callersFunctioncreate_file
Create a file with the given path and content. Use this function when there is a need to create a new file with initial content. Args:
research_agent/inno/tools/terminal_tools.py:177
↓ 4 callersFunctionedge_index_to_adj
Convert edge_index to adjacency matrix. Args: edge_index (torch.Tensor): Edge index tensor of shape [2, E] num_nodes (int): N
examples/gnn_nodeformer/project/experiments/edge_index_to_adj.py:5
↓ 4 callersFunctionfind_node_by_class
递归查找具有特定class的节点
research_agent/inno/tools/inno_tools/web_tools.py:425
↓ 4 callersMethodget_logger
(cls)
research_agent/inno/logger.py:127
↓ 4 callersMethodhandle_tool_calls
( self, tool_calls: List[ChatCompletionMessageToolCall], functions: List[AgentFunction
research_agent/inno/core.py:179
↓ 4 callersFunctionload_data
Load Yelp dataset
examples/hgcl/project/data_processing/dataset.py:7
↓ 4 callersFunctionload_dataset
Load a dataset from PyTorch Geometric. Args: dataset_name (str): Name of the dataset ('Cora', 'CiteSeer', or 'PubMed') Retur
examples/gnn_nodeformer/project/data_processing/data_loader.py:9
↓ 4 callersMethodquery
query: query the memery. Args: query_texts: the query texts to search in the memery. collection: the name of
research_agent/inno/memory/rag_memory.py:87
↓ 4 callersMethodset_address
Sets the address of the current page. This will result in the page being fetched via the underlying requests session. Arguments:
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:146
↓ 4 callersMethodset_logger
(cls, new_logger)
research_agent/inno/logger.py:131
↓ 4 callersMethodtest_epoch
(self)
examples/rotation_vq/project/training/trainer.py:73
↓ 4 callersMethodtrain_epoch
(self, epoch)
examples/rotation_vq/project/training/trainer.py:16
↓ 4 callersMethodtrain_epoch
(self, user_graph, user_item_graph, item_graph)
examples/hgcl/project/training/trainer.py:16
↓ 3 callersMethod__init__
(self, lambda_reg=1.0)
examples/gnn_difformer/project/model/diffusion.py:30
↓ 3 callersMethod__init__
(self, hidden_dim, lambda_reg=1.0)
examples/gnn_difformer/project/model/improved_diffusion.py:45
↓ 3 callersMethod_convert
(self, local_path: str, extensions: List[Union[str, None]], **kwargs)
research_agent/inno/environment/mdconvert.py:964
↓ 3 callersMethod_convert
(self, local_path: str, extensions: List[Union[str, None]], **kwargs)
research_agent/inno/environment/markdown_browser/mdconvert.py:958
↓ 3 callersMethod_get_metadata
(self, local_path)
research_agent/inno/environment/mdconvert.py:595
↓ 3 callersMethod_get_metadata
(self, local_path)
research_agent/inno/environment/markdown_browser/mdconvert.py:588
↓ 3 callersMethod_guess_ext_magic
Use puremagic (a Python implementation of libmagic) to guess a file's extension based on the first few bytes.
research_agent/inno/environment/mdconvert.py:1020
↓ 3 callersMethod_guess_ext_magic
Use puremagic (a Python implementation of libmagic) to guess a file's extension based on the first few bytes.
research_agent/inno/environment/markdown_browser/mdconvert.py:1014
↓ 3 callersFunctionadd_env_var
添加或更新单个环境变量 Args: key: 环境变量名 value: 环境变量值 from_frontend: 是否来自前端配置,默认为True
web_ai_researcher.py:672
↓ 3 callersMethodadd_query
add_query: add the queries to the memery. Args: queries: the queries to add to the memery. Should be in the format of
research_agent/inno/memory/rag_memory.py:45
↓ 3 callersMethodcalculate_fid
Calculate FID score between real and generated images
examples/con_flowmatching/project/testing/evaluator.py:42
↓ 3 callersFunctionconvert_tools_to_description
(tools: list[dict])
research_agent/inno/fn_call_converter.py:281
↓ 3 callersFunctioncreate_directory
Create a directory if it does not exist. Use this function when there is a need to create a new directory. Args: path: The path of th
research_agent/inno/tools/terminal_tools.py:227
↓ 3 callersFunctionfilter_empty_conversations
过滤掉完全空的对话记录,并处理部分空内容
web_ai_researcher.py:960
↓ 3 callersMethodfuse_subsections
Fuse all subsections into one complete section
paper_agent/section_composer.py:129
↓ 3 callersFunctionget_file_md5
(file_path)
research_agent/inno/io_utils.py:16
↓ 3 callersFunctionget_latest_logs
(max_lines=500, state=None, queue_source=None, last_index=0)
web_ai_researcher.py:405
↓ 3 callersFunctionget_node_links
获取节点内所有的链接
research_agent/inno/tools/inno_tools/web_tools.py:404
↓ 3 callersFunctionget_node_text
(node_idx)
research_agent/inno/tools/inno_tools/web_tools.py:392
↓ 3 callersMethodinit_container
(self)
research_agent/inno/environment/docker_env.py:41
↓ 3 callersFunctionload_data
(path)
examples/dccf/project/data_processing/dataset.py:28
next →1–100 of 951, ranked by callers