Index

_

__init__() (ml_research_tools.BaseTool method) (ml_research_tools.cache.redis.RedisCache method) (ml_research_tools.cache.RedisCache method) (ml_research_tools.core.base_tool.BaseTool method) (ml_research_tools.core.llm_tools.LLMClient method) (ml_research_tools.core.LLMClient method) (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) (ml_research_tools.doc.ask_document_tool.DocumentChat method) (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool method) (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method)

__post_init__() (ml_research_tools.core.config.LLMPresets method) (ml_research_tools.core.LLMPresets method)

A

add_arguments() (ml_research_tools.BaseTool class method) (ml_research_tools.core.base_tool.BaseTool class method) (ml_research_tools.doc.ask_document_tool.AskDocumentTool class method) (ml_research_tools.doc.AskDocumentTool class method) (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool class method) (ml_research_tools.kube.pod_forward_tool.PodForwardTool class method) (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool class method) add_assistant_message() (ml_research_tools.doc.ask_document_tool.DocumentChat method) add_config_args() (in module ml_research_tools.core.config)

add_subparser() (ml_research_tools.BaseTool class method) (ml_research_tools.core.base_tool.BaseTool class method) add_user_message() (ml_research_tools.doc.ask_document_tool.DocumentChat method) all_states (ml_research_tools.exp.wandb_store.ExperimentStore property) all_tags (ml_research_tools.exp.wandb_store.ExperimentStore property) api_key (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute) AskDocumentTool (class in ml_research_tools.doc) (class in ml_research_tools.doc.ask_document_tool)

B

base_url (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute)

BaseTool (class in ml_research_tools) (class in ml_research_tools.core.base_tool)

C

cached() (in module ml_research_tools.cache) (in module ml_research_tools.cache.redis) calculate_a_paper_sizes() (in module ml_research_tools.tex.presets.sizes) call() (ml_research_tools.core.llm_tools.LLMClient method) (ml_research_tools.core.LLMClient method) can_handle() (ml_research_tools.doc.ask_document_tool.CodeDocumentParser class method) (ml_research_tools.doc.ask_document_tool.DocumentParser class method) (ml_research_tools.doc.ask_document_tool.LatexDocumentParser class method) (ml_research_tools.doc.ask_document_tool.PDFDocumentParser class method) (ml_research_tools.doc.ask_document_tool.TextDocumentParser class method) (ml_research_tools.doc.ask_document_tool.URLParser class method) clear() (ml_research_tools.cache.redis.RedisCache method) (ml_research_tools.cache.RedisCache method) CodeDocumentParser (class in ml_research_tools.doc.ask_document_tool) Config (class in ml_research_tools) (class in ml_research_tools.core) (class in ml_research_tools.core.config)

config (ml_research_tools.core.llm_tools.LLMClient attribute) (ml_research_tools.core.LLMClient attribute) content (ml_research_tools.core.llm_tools.Message attribute) create_default_llm_client() (in module ml_research_tools.core) (in module ml_research_tools.core.service_factories) create_diff_file() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method) create_latexdiff_file() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method) create_llm_client() (in module ml_research_tools.core) (in module ml_research_tools.core.llm_tools) create_progress() (ml_research_tools.BaseTool method) (ml_research_tools.core.base_tool.BaseTool method) create_redis_cache() (in module ml_research_tools.core) (in module ml_research_tools.core.service_factories) create_redis_client() (in module ml_research_tools.cache) (in module ml_research_tools.cache.redis)

D

db (ml_research_tools.core.config.RedisConfig attribute) default (ml_research_tools.core.config.LLMPresets attribute) (ml_research_tools.core.LLMPresets attribute) delete() (ml_research_tools.cache.redis.RedisCache method) (ml_research_tools.cache.RedisCache method) delete_outdated_logs() (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool method) description (ml_research_tools.BaseTool attribute) (ml_research_tools.core.base_tool.BaseTool attribute) (ml_research_tools.doc.ask_document_tool.AskDocumentTool attribute) (ml_research_tools.doc.AskDocumentTool attribute) (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool attribute) (ml_research_tools.kube.pod_forward_tool.PodForwardTool attribute) (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool attribute)

display_pod_table() (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) DocumentChat (class in ml_research_tools.doc.ask_document_tool) DocumentParser (class in ml_research_tools.doc.ask_document_tool) download_wandb_logs() (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool method) draw_text() (ml_research_tools.tex.backend.svg_patch.RendererInkSVG method)

E

enabled (ml_research_tools.cache.redis.RedisCache property) (ml_research_tools.cache.RedisCache property) (ml_research_tools.core.config.RedisConfig attribute) estimate_token_count_with_cache() (in module ml_research_tools.doc.ask_document_tool) execute() (ml_research_tools.BaseTool method) (ml_research_tools.core.base_tool.BaseTool method) (ml_research_tools.doc.ask_document_tool.AskDocumentTool method) (ml_research_tools.doc.AskDocumentTool method) (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool method) (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method)

execute_from_args() (ml_research_tools.BaseTool method) (ml_research_tools.core.base_tool.BaseTool method) exists() (ml_research_tools.cache.redis.RedisCache method) (ml_research_tools.cache.RedisCache method) ExperimentStore (class in ml_research_tools.exp.wandb_store) extract_content() (ml_research_tools.doc.ask_document_tool.CodeDocumentParser class method) (ml_research_tools.doc.ask_document_tool.DocumentParser class method) (ml_research_tools.doc.ask_document_tool.PDFDocumentParser class method) (ml_research_tools.doc.ask_document_tool.TextDocumentParser class method) (ml_research_tools.doc.ask_document_tool.URLParser class method)

F

FigureCanvasInkSVG (class in ml_research_tools.tex.backend.svg_patch) FigureManagerInkSVG (class in ml_research_tools.tex.backend.svg_patch) file (ml_research_tools.core.config.LoggingConfig attribute) find_pod_by_pattern() (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) fixed_dpi (ml_research_tools.tex.backend.svg_patch.FigureCanvasInkSVG attribute)

font_scale (ml_research_tools.tex.backend.svg_patch.SvgInkConfig attribute) forward_port() (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) from_dict() (ml_research_tools.Config class method) (ml_research_tools.core.Config class method) (ml_research_tools.core.config.Config class method)

G

generate_cache_key() (in module ml_research_tools.cache) (in module ml_research_tools.cache.redis) generate_completion_params() (in module ml_research_tools.core) (in module ml_research_tools.core.llm_tools) generate_document_cache_key() (in module ml_research_tools.doc.ask_document_tool) get() (ml_research_tools.cache.redis.RedisCache method) (ml_research_tools.cache.RedisCache method) (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) get_config() (in module ml_research_tools) (in module ml_research_tools.core) (in module ml_research_tools.core.config) (ml_research_tools.core.config.LLMPresets method) (ml_research_tools.core.LLMPresets method) (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) get_console() (in module ml_research_tools.core.logging_tools)

get_experiment() (ml_research_tools.exp.wandb_store.ExperimentStore method) get_from_cache() (in module ml_research_tools.cache) (in module ml_research_tools.cache.redis) get_llm_config() (in module ml_research_tools.core.llm_tools) get_logger() (in module ml_research_tools) (in module ml_research_tools.core) (in module ml_research_tools.core.logging_tools) get_openai_client() (ml_research_tools.core.llm_tools.LLMClient method) (ml_research_tools.core.LLMClient method) get_or_create() (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) get_parser_for_document() (in module ml_research_tools.doc.ask_document_tool) get_running_pods() (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) get_typed() (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) golden_figure() (in module ml_research_tools.tex.presets.nice_figures) golden_subplots() (in module ml_research_tools.tex.presets.nice_figures) groupby() (ml_research_tools.exp.wandb_store.ExperimentStore static method)

H

has() (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method)

height (ml_research_tools.tex.presets.sizes.PaperSize attribute) host (ml_research_tools.core.config.RedisConfig attribute)

I

ignore_alpha (ml_research_tools.tex.backend.svg_patch.SvgInkConfig attribute) ignore_color (ml_research_tools.tex.backend.svg_patch.SvgInkConfig attribute)

ignore_font_size (ml_research_tools.tex.backend.svg_patch.SvgInkConfig attribute) ignore_style (ml_research_tools.tex.backend.svg_patch.SvgInkConfig attribute) ignore_weight (ml_research_tools.tex.backend.svg_patch.SvgInkConfig attribute)

L

LatexDocumentParser (class in ml_research_tools.doc.ask_document_tool) LatexGrammarTool (class in ml_research_tools.tex.latex_grammar_tool) level (ml_research_tools.core.config.LoggingConfig attribute) llm (ml_research_tools.Config property) (ml_research_tools.core.Config property) (ml_research_tools.core.config.Config property) llm_presets (ml_research_tools.Config attribute) (ml_research_tools.core.Config attribute) (ml_research_tools.core.config.Config attribute) LLMClient (class in ml_research_tools.core) (class in ml_research_tools.core.llm_tools)

LLMConfig (class in ml_research_tools.core) (class in ml_research_tools.core.config) LLMPresets (class in ml_research_tools.core) (class in ml_research_tools.core.config) load_config() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method) load_config_file() (in module ml_research_tools.core.config) load_document_with_cache() (in module ml_research_tools.doc.ask_document_tool) logging (ml_research_tools.Config attribute) (ml_research_tools.core.Config attribute) (ml_research_tools.core.config.Config attribute) LoggingConfig (class in ml_research_tools.core.config)

M

max_tokens (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute) merge_runs() (ml_research_tools.exp.wandb_store.ExperimentStore static method) Message (class in ml_research_tools.core.llm_tools) ml_research_tools module ml_research_tools.cache module ml_research_tools.cache.redis module ml_research_tools.core module ml_research_tools.core.base_tool module ml_research_tools.core.config module ml_research_tools.core.llm_tools module ml_research_tools.core.logging_tools module ml_research_tools.core.service_factories module ml_research_tools.core.service_provider module ml_research_tools.doc module ml_research_tools.doc.ask_document_tool module ml_research_tools.exp module ml_research_tools.exp.wandb_downloader_tool module ml_research_tools.exp.wandb_store module ml_research_tools.kube module ml_research_tools.kube.pod_forward_tool module ml_research_tools.tex module ml_research_tools.tex.backend module ml_research_tools.tex.backend.svg_patch module

ml_research_tools.tex.latex_grammar_tool module ml_research_tools.tex.presets module ml_research_tools.tex.presets.nice_figures module ml_research_tools.tex.presets.save module ml_research_tools.tex.presets.sizes module ml_research_tools.tex.presets.tools module model (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.llm_tools.LLMClient property) (ml_research_tools.core.LLMClient property) (ml_research_tools.core.LLMConfig attribute) module ml_research_tools ml_research_tools.cache ml_research_tools.cache.redis ml_research_tools.core ml_research_tools.core.base_tool ml_research_tools.core.config ml_research_tools.core.llm_tools ml_research_tools.core.logging_tools ml_research_tools.core.service_factories ml_research_tools.core.service_provider ml_research_tools.doc ml_research_tools.doc.ask_document_tool ml_research_tools.exp ml_research_tools.exp.wandb_downloader_tool ml_research_tools.exp.wandb_store ml_research_tools.kube ml_research_tools.kube.pod_forward_tool ml_research_tools.tex ml_research_tools.tex.backend ml_research_tools.tex.backend.svg_patch ml_research_tools.tex.latex_grammar_tool ml_research_tools.tex.presets ml_research_tools.tex.presets.nice_figures ml_research_tools.tex.presets.save ml_research_tools.tex.presets.sizes ml_research_tools.tex.presets.tools

N

name (ml_research_tools.BaseTool attribute) (ml_research_tools.core.base_tool.BaseTool attribute) (ml_research_tools.doc.ask_document_tool.AskDocumentTool attribute) (ml_research_tools.doc.AskDocumentTool attribute) (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool attribute) (ml_research_tools.kube.pod_forward_tool.PodForwardTool attribute) (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool attribute)

P

PaperSize (class in ml_research_tools.tex.presets.sizes) parse_and_execute() (ml_research_tools.BaseTool method) (ml_research_tools.core.base_tool.BaseTool method) password (ml_research_tools.core.config.RedisConfig attribute) PDFDocumentParser (class in ml_research_tools.doc.ask_document_tool) PodForwardTool (class in ml_research_tools.kube.pod_forward_tool)

port (ml_research_tools.core.config.RedisConfig attribute) post_process_chunk() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method) presets (ml_research_tools.core.config.LLMPresets attribute) (ml_research_tools.core.LLMPresets attribute) print_svg() (ml_research_tools.tex.backend.svg_patch.FigureCanvasInkSVG method) process_run() (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool method)

R

read_latex_file() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool method) recache (ml_research_tools.cache.redis.RedisCache property) (ml_research_tools.cache.RedisCache property) (ml_research_tools.core.config.RedisConfig attribute) redis (ml_research_tools.Config attribute) (ml_research_tools.core.Config attribute) (ml_research_tools.core.config.Config attribute) RedisCache (class in ml_research_tools.cache) (class in ml_research_tools.cache.redis) RedisConfig (class in ml_research_tools.core.config) register() (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) register_common_services() (in module ml_research_tools.core) (in module ml_research_tools.core.service_factories)

register_factory() (ml_research_tools.core.service_provider.ServiceProvider method) (ml_research_tools.core.ServiceProvider method) remove_think_tags() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool static method) RendererInkSVG (class in ml_research_tools.tex.backend.svg_patch) replace_texcmd() (in module ml_research_tools.tex.backend.svg_patch) retry_attempts (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute) retry_delay (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute) role (ml_research_tools.core.llm_tools.Message attribute) run_config() (ml_research_tools.exp.wandb_store.ExperimentStore static method) run_info() (ml_research_tools.exp.wandb_store.ExperimentStore static method) run_interactive_chat() (ml_research_tools.doc.ask_document_tool.DocumentChat method) run_kubectl_command() (ml_research_tools.kube.pod_forward_tool.PodForwardTool method) run_summary() (ml_research_tools.exp.wandb_store.ExperimentStore static method)

S

sanitize_filename() (ml_research_tools.exp.wandb_downloader_tool.WandbDownloaderTool static method) save_to_cache() (in module ml_research_tools.cache) (in module ml_research_tools.cache.redis) savefig() (in module ml_research_tools.tex.presets.save) search() (ml_research_tools.exp.wandb_store.ExperimentStore method) ServiceProvider (class in ml_research_tools.core) (class in ml_research_tools.core.service_provider) set() (ml_research_tools.cache.redis.RedisCache method) (ml_research_tools.cache.RedisCache method) set_plot_area_aspect() (in module ml_research_tools.tex.presets.tools) set_property() (in module ml_research_tools.tex.backend)

setup_logging() (in module ml_research_tools) (in module ml_research_tools.core) (in module ml_research_tools.core.logging_tools) setup_services() (in module ml_research_tools.core) (in module ml_research_tools.core.service_factories) should_cache() (ml_research_tools.doc.ask_document_tool.DocumentParser class method) (ml_research_tools.doc.ask_document_tool.URLParser class method) simple_call() (ml_research_tools.core.llm_tools.LLMClient method) (ml_research_tools.core.LLMClient method) split_into_chunks() (ml_research_tools.tex.latex_grammar_tool.LatexGrammarTool static method) stream_llm_response() (ml_research_tools.doc.ask_document_tool.DocumentChat method) SvgInkConfig (class in ml_research_tools.tex.backend.svg_patch)

T

temperature (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute) TextDocumentParser (class in ml_research_tools.doc.ask_document_tool) tier (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute)

top_p (ml_research_tools.core.config.LLMConfig attribute) (ml_research_tools.core.LLMConfig attribute) transform_text_between_dollars() (in module ml_research_tools.tex.backend.svg_patch) ttl (ml_research_tools.core.config.RedisConfig attribute)

U

URLParser (class in ml_research_tools.doc.ask_document_tool)

W

WandbDownloaderTool (class in ml_research_tools.exp.wandb_downloader_tool)

width (ml_research_tools.tex.presets.sizes.PaperSize attribute)