Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SamuelSchmidgall/AgentLaboratory
/ functions
Functions
190 in github.com/SamuelSchmidgall/AgentLaboratory
⨍
Functions
190
◇
Types & classes
23
↳
Endpoints
7
↓ 25 callers
Function
extract_prompt
(text, word)
utils.py:235
↓ 16 callers
Function
query_model
(model_str, prompt, system_prompt, openai_api_key=None, gemini_api_key=None, anthropic_api_key=None, tries=5,
inference.py:35
↓ 16 callers
Method
set_agent_attr
Set attribute for all agents @param attr: (str) agent attribute @param obj: (object) object attribute @return: None
ai_lab_repo.py:115
↓ 10 callers
Method
inference
(self, research_topic, phase, step, feedback="", temp=None)
agents.py:247
↓ 9 callers
Method
reset_agents
Reset all agent states @return: None
ai_lab_repo.py:128
↓ 8 callers
Method
human_in_loop
Get human feedback for phase output @param phase: (str) current phase @param phase_prod: (str) current phase result @
ai_lab_repo.py:547
↓ 5 callers
Method
__init__
(self, model="gpt-4o-mini", notes=None, max_steps=100, openai_api_key=None)
agents.py:205
↓ 5 callers
Function
execute_code
(code_str, timeout=600, MAX_LEN=1000)
tools.py:306
↓ 5 callers
Function
get_score
(outlined_plan, latex, reward_model_llm, reviewer_type=None, attempts=3, openai_api_key=None)
agents.py:36
↓ 5 callers
Method
reset
(self)
agents.py:279
↓ 5 callers
Function
save_to_file
Utility function to save data as plain text.
utils.py:186
↓ 4 callers
Method
system_prompt
Produce a system prompt for the mle-solver to solve ml problems @param commands: (bool) whether to use command prompt @return
mlesolver.py:422
↓ 3 callers
Method
_normalize
(self, arr)
tools.py:81
↓ 3 callers
Method
find_papers_by_str
(self, query, N=20)
tools.py:229
↓ 3 callers
Function
get_score
(outlined_plan, code, code_return, REWARD_MODEL_LLM, attempts=3, openai_api_key=None)
mlesolver.py:141
↓ 3 callers
Function
remove_figures
Remove a directory if it exists.
utils.py:168
↓ 3 callers
Method
retrieve_full_paper_text
(self, query, MAX_LEN=50000)
tools.py:262
↓ 3 callers
Function
update_papers_from_uploads
()
app.py:25
↓ 2 callers
Method
__init__
(self)
papersolver.py:48
↓ 2 callers
Method
add_review
(self, review, arx_eng, agentrxiv=False, GLOBAL_AGENTRXIV=None)
agents.py:714
↓ 2 callers
Method
clean_text
(text)
papersolver.py:326
↓ 2 callers
Method
clean_text
(text)
mlesolver.py:245
↓ 2 callers
Function
code_repair
(code, error, ctype, REPAIR_LLM, openai_api_key=None)
mlesolver.py:166
↓ 2 callers
Function
compile_latex
(latex_code, output_path, compile=True, timeout=30)
utils.py:127
↓ 2 callers
Method
format_review
(self)
agents.py:733
↓ 2 callers
Method
generate_code_lines
Generate well-formatted code lines with line numbers @param code: (list) list of code line strings @return: (str) code lines
mlesolver.py:453
↓ 2 callers
Function
last_boxed_only_string
(string)
utils.py:312
↓ 2 callers
Method
parse_command
(self, *args)
papersolver.py:77
↓ 2 callers
Method
parse_command
(self, *args)
mlesolver.py:126
↓ 2 callers
Method
perform_research
Loop through all research phases @return: None
ai_lab_repo.py:139
↓ 2 callers
Method
process_command
Take command from language model and execute if valid @param model_resp: (str) language model output @return: (tuple) tuple c
papersolver.py:399
↓ 2 callers
Method
process_command
Take command from language model and execute if valid @param model_resp: (str) language model output @return: (tuple) tuple c
mlesolver.py:329
↓ 2 callers
Function
remove_boxed
(s)
utils.py:298
↓ 2 callers
Function
remove_directory
Remove a directory if it exists.
utils.py:174
↓ 2 callers
Method
retrieve_full_text
(self, arxiv_id)
ai_lab_repo.py:597
↓ 2 callers
Method
role_description
(self)
agents.py:289
↓ 2 callers
Function
run_app
(port=5000)
app.py:159
↓ 2 callers
Method
save_state
Save state for phase @param phase: (str) phase string @return: None
ai_lab_repo.py:106
↓ 2 callers
Method
set_model
(self, model)
ai_lab_repo.py:102
↓ 2 callers
Function
strip_string
(string)
utils.py:414
↓ 2 callers
Method
system_prompt
Produce a system prompt for the paper-solver @param commands: (bool) whether to use command prompt @return: (str) system prom
papersolver.py:481
↓ 1 callers
Method
__init__
(self)
mlesolver.py:87
↓ 1 callers
Method
_common_code_errors
Some general tips to avoid common code errors, also TF has many errors so we avoid this and ask to use pytorch @return: (str) common
mlesolver.py:533
↓ 1 callers
Method
_process_query
Process query string to fit within MAX_QUERY_LENGTH while preserving as much information as possible
tools.py:205
↓ 1 callers
Function
clean_answer
(s)
utils.py:293
↓ 1 callers
Method
clean_text
Fix minor corrections :return: (str) corrected text
agents.py:234
↓ 1 callers
Method
command_descriptions
(self, phase)
agents.py:292
↓ 1 callers
Method
command_descriptions
Provide command descriptions @return: (str) command descriptions
papersolver.py:552
↓ 1 callers
Method
command_descriptions
Provide command descriptions @return: (str) command descriptions
mlesolver.py:545
↓ 1 callers
Method
context
(self, phase)
agents.py:283
↓ 1 callers
Function
curr_cost_est
()
inference.py:12
↓ 1 callers
Method
data_preparation
Perform data preparation phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:343
↓ 1 callers
Method
docstring
(self)
papersolver.py:54
↓ 1 callers
Method
docstring
(self)
mlesolver.py:91
↓ 1 callers
Method
execute_command
(self, *args)
papersolver.py:63
↓ 1 callers
Method
execute_command
(self, *args)
mlesolver.py:99
↓ 1 callers
Function
extract_json_between_markers
(llm_output)
agents.py:7
↓ 1 callers
Function
fix_a_slash_b
(string)
utils.py:374
↓ 1 callers
Function
fix_fracs
(string)
utils.py:342
↓ 1 callers
Function
fix_sqrt
(string)
utils.py:399
↓ 1 callers
Method
gen_initial_code
(self)
mlesolver.py:250
↓ 1 callers
Method
gen_initial_report
(self)
papersolver.py:330
↓ 1 callers
Method
generate_dataset_descr_prompt
Generate description prompt for kaggle dataset @param data_loader: (DataLoader) data loader @return: (str) description prompt
mlesolver.py:504
↓ 1 callers
Method
generate_paper_lines
Generate well-formatted code lines with line numbers @param code: (list) list of code line strings @return: (str) code lines
papersolver.py:470
↓ 1 callers
Method
generate_readme
(self)
agents.py:304
↓ 1 callers
Method
history_str
Well-formatted history string @return: (str) history string
mlesolver.py:407
↓ 1 callers
Method
initial_solve
Initialize the solver and get an initial set of papers and a return @return: None
papersolver.py:306
↓ 1 callers
Method
initial_solve
Initialize the solver and get an initial set of code and a return @return: None
mlesolver.py:225
↓ 1 callers
Method
initialize_server
(self)
ai_lab_repo.py:584
↓ 1 callers
Function
is_equiv
(str1, str2, verbose=False)
utils.py:276
↓ 1 callers
Method
literature_review
Perform literature review phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:465
↓ 1 callers
Method
matches_command
(self, cmd_str)
papersolver.py:72
↓ 1 callers
Method
matches_command
(self, cmd_str)
mlesolver.py:122
↓ 1 callers
Method
num_papers
()
ai_lab_repo.py:594
↓ 1 callers
Function
parse_arguments
()
ai_lab_repo.py:653
↓ 1 callers
Function
parse_yaml
(yaml_file_loc)
ai_lab_repo.py:666
↓ 1 callers
Method
phase_prompt
(self, phase)
agents.py:286
↓ 1 callers
Method
phase_prompt
Describe system role and general tips for mle-solver @return: (str) system role
papersolver.py:568
↓ 1 callers
Method
phase_prompt
Describe system role and general tips for mle-solver @return: (str) system role
mlesolver.py:512
↓ 1 callers
Method
plan_formulation
Perform plan formulation phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:414
↓ 1 callers
Method
read_pdf_pypdf2
(pdf_path)
ai_lab_repo.py:604
↓ 1 callers
Method
reflect_code
Provide a reflection on produced behavior for next execution @return: (str) language model-produced reflection
mlesolver.py:319
↓ 1 callers
Method
reflection
Reflect on your future plans and next steps to improve the code @param reflect_prompt: (str) reflection prompt @param code_st
mlesolver.py:494
↓ 1 callers
Function
remove_right_units
(string)
utils.py:389
↓ 1 callers
Method
report_refinement
Perform report refinement phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:207
↓ 1 callers
Method
report_writing
Perform report writing phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:240
↓ 1 callers
Method
results_interpretation
Perform results interpretation phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:274
↓ 1 callers
Method
results_str
Provide results as list of results in human-readable format. :param results: (list(dict)) list of results from search :return
tools.py:157
↓ 1 callers
Method
retrieve_ds
Retrieves the top N datasets matching the query, weighted by likes and downloads. :param query: The search query string. :par
tools.py:88
↓ 1 callers
Method
role_description
(self)
agents.py:361
↓ 1 callers
Method
role_description
(self)
agents.py:711
↓ 1 callers
Method
role_description
Provide role description @return: (str) role description
papersolver.py:560
↓ 1 callers
Method
role_description
Provide role description @return: (str) role description
mlesolver.py:525
↓ 1 callers
Method
run_server
(self, port)
ai_lab_repo.py:649
↓ 1 callers
Method
running_experiments
Perform running experiments phase @return: (bool) whether to repeat the phase
ai_lab_repo.py:310
↓ 1 callers
Method
search_agentrxiv
(self, search_query, num_papers)
ai_lab_repo.py:613
↓ 1 callers
Method
solve
(self)
papersolver.py:269
↓ 1 callers
Method
solve
(self)
mlesolver.py:276
Method
__init__
(self, model="gpt-4o-mini", notes=None, openai_api_key=None)
agents.py:185
Method
__init__
(self, model="gpt4omini", notes=None, max_steps=100, openai_api_key=None)
agents.py:300
next →
1–100 of 190, ranked by callers