Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/aiming-lab/MDocAgent
/ functions
Functions
79 in github.com/aiming-lab/MDocAgent
⨍
Functions
79
◇
Types & classes
16
↓ 7 callers
Method
load_data
(self, use_retreival=True)
mydatasets/base_dataset.py:26
↓ 4 callers
Method
clean_up
(self)
models/base_model.py:13
↓ 4 callers
Method
load_processed_content
(self, sample: dict, disable_load_image=True)
mydatasets/base_dataset.py:125
↓ 4 callers
Method
predict
(self, sample, question, texts, images)
agents/ablations.py:8
↓ 4 callers
Method
predict_dataset
(self, dataset:BaseDataset, resume_path = None)
agents/multi_agent_system.py:51
↓ 4 callers
Method
sum
(self, sum_question)
agents/multi_agent_system.py:39
↓ 3 callers
Method
dump_data
(self, samples, use_retreival=True)
mydatasets/base_dataset.py:41
↓ 3 callers
Method
process_message
(self, question, texts, images, history)
models/base_model.py:16
↓ 3 callers
Method
self_reflect
(self, prompt=None, add_to_message = True)
agents/base_agent.py:41
↓ 2 callers
Method
__init__
(self, config)
agents/ablations.py:5
↓ 2 callers
Method
_predict
(self, question, texts=None, images=None, add_to_message = False)
agents/base_agent.py:26
↓ 2 callers
Method
dump_reults
(self, samples)
mydatasets/base_dataset.py:60
↓ 2 callers
Method
eval
(self, question, answer, gt)
agents/base_agent.py:53
↓ 2 callers
Method
load_sample_retrieval_data
(self, sample)
mydatasets/base_dataset.py:75
↓ 2 callers
Method
predict
Implement the method in the subclass
agents/multi_agent_system.py:35
↓ 2 callers
Method
predict
(self, question, texts=None, images=None, with_sys_prompt=True)
agents/base_agent.py:36
↓ 1 callers
Method
_extract_content
(self, sample, resolution=144)
mydatasets/base_dataset.py:156
↓ 1 callers
Method
add_agent
(self, agent_config, model)
agents/multi_agent_system.py:29
↓ 1 callers
Method
clean_messages
(self)
agents/multi_agent_system.py:85
↓ 1 callers
Method
create_image_message
(self, images, question)
models/qwen.py:38
↓ 1 callers
Method
create_text_message
(self, texts, question)
models/llama.py:24
↓ 1 callers
Function
encode_image
(image_path)
models/openai.py:5
↓ 1 callers
Method
eval_dataset
(self, dataset: BaseDataset)
agents/base_agent.py:63
↓ 1 callers
Method
extract_content
(self, resolution=144)
mydatasets/base_dataset.py:151
↓ 1 callers
Function
extract_evaluation_metrics
(eval_str: str)
agents/base_agent.py:92
↓ 1 callers
Function
find_latest_json
(result_dir)
mydatasets/base_dataset.py:185
↓ 1 callers
Method
find_sample_top_k
(self, sample, document_embed, top_k: int, page_id_key: str)
retrieval/image_retrieval.py:61
↓ 1 callers
Method
find_sample_top_k
(self, sample, top_k: int, page_id_key: str)
retrieval/text_retrieval.py:43
↓ 1 callers
Method
find_top_k
(self, dataset: BaseDataset)
retrieval/base_retrieval.py:10
↓ 1 callers
Method
is_valid_history
(self, history)
models/base_model.py:33
↓ 1 callers
Method
load_document_embeds
(self, dataset: BaseDataset, force_prepare=False)
retrieval/image_retrieval.py:104
↓ 1 callers
Method
load_image
(self, file)
mydatasets/base_dataset.py:140
↓ 1 callers
Method
load_latest_results
(self)
mydatasets/base_dataset.py:53
↓ 1 callers
Method
load_sample_full_data
(self, sample)
mydatasets/base_dataset.py:107
↓ 1 callers
Method
load_txt
(self, file)
mydatasets/base_dataset.py:144
↓ 1 callers
Function
main
(cfg)
scripts/predict.py:9
↓ 1 callers
Function
main
(cfg)
scripts/eval.py:9
↓ 1 callers
Function
main
(cfg)
scripts/retrieve.py:10
↓ 1 callers
Function
main
(cfg)
scripts/extract.py:8
↓ 1 callers
Function
main
(cfg)
scripts/ablations/mmds.py:9
↓ 1 callers
Function
main
(cfg)
scripts/ablations/mmdi.py:9
↓ 1 callers
Function
main
(cfg)
scripts/ablations/mmdt.py:9
↓ 1 callers
Method
prepare
(self, dataset: BaseDataset)
retrieval/image_retrieval.py:23
↓ 1 callers
Method
prepare
(self, dataset: BaseDataset)
retrieval/text_retrieval.py:13
Method
__init__
(self, config)
mydatasets/base_dataset.py:18
Method
__init__
Base model constructor to initialize common attributes. :param config: A dictionary containing model configuration parameters.
models/base_model.py:3
Method
__init__
(self, config)
models/llama.py:6
Method
__init__
(self, config)
models/openai.py:10
Method
__init__
(self, config)
models/qwen.py:7
Method
__init__
(self, config)
models/qwen.py:97
Method
__init__
(self, config)
retrieval/image_retrieval.py:16
Method
__init__
(self, config)
retrieval/base_retrieval.py:4
Method
__init__
(self, config)
retrieval/text_retrieval.py:10
Method
__init__
(self, config)
agents/ablations.py:38
Method
__init__
(self, config)
agents/ablations.py:70
Method
__init__
(self, config)
agents/mdoc_agent.py:12
Method
__init__
(self, config)
agents/multi_agent_system.py:11
Method
__init__
(self, config, model=None)
agents/base_agent.py:12
Method
clean_messages
(self)
agents/base_agent.py:23
Method
create_image_message
(self, images, question)
models/openai.py:40
Method
create_text_message
(self, texts, question)
models/openai.py:29
Method
create_text_message
(self, texts, question)
models/qwen.py:27
Method
extract_final_answer
(agent_response)
agents/multi_agent_system.py:41
Function
extract_time
(file_path)
mydatasets/base_dataset.py:180
Method
find_top_k
(self, dataset: BaseDataset, prepare=False)
retrieval/image_retrieval.py:90
Method
find_top_k
(self, dataset: BaseDataset, force_prepare=False)
retrieval/text_retrieval.py:73
Method
is_valid_history
(self, history)
models/llama.py:46
Method
is_valid_history
(self, history)
models/openai.py:63
Method
is_valid_history
(self, history)
models/qwen.py:77
Method
load_full_data
(self)
mydatasets/base_dataset.py:101
Method
load_retrieval_data
(self)
mydatasets/base_dataset.py:67
Method
predict
(self, question, texts = None, images = None, history = None)
models/base_model.py:10
Method
predict
(self, question, texts = None, images = None, history = None)
models/llama.py:35
Method
predict
(self, question, texts = None, images = None, history = None)
models/openai.py:51
Method
predict
(self, question, texts = None, images = None, history = None)
models/qwen.py:50
Method
predict
(self, sample, question, texts, images)
agents/ablations.py:41
Method
predict
(self, sample, question, texts, images)
agents/ablations.py:73
Method
predict
(self, question, texts, images)
agents/mdoc_agent.py:15
Method
prepare
(self, dataset: BaseDataset)
retrieval/base_retrieval.py:7