MCPcopy Create free account

hub / github.com/JingbiaoMei/ATM-Bench / functions

Functions815 in github.com/JingbiaoMei/ATM-Bench

↓ 79 callersFunction_env
(key: str, default: str)
agent_systems/config.py:32
↓ 46 callersFunctionload_json
(path: Path)
memqa/retrieve/utils.py:94
↓ 43 callersFunctionwrite_json
(path: Path, payload: Any)
memqa/retrieve/utils.py:116
↓ 38 callersFunction_as_int
(value: Any)
agent_systems/extract_usage.py:103
↓ 29 callersMethodload
(self, state_path: str, embeddings_path: str)
memqa/qa_agent_baselines/A-Mem/memory_layer.py:381
↓ 19 callersFunction_repo_path_str
(path: str)
agent_systems/config.py:37
↓ 17 callersMethodencode
Encode texts (alternative interface name). Args: texts: Single text or list of texts **kwargs: Ignored, for
memqa/qa_agent_baselines/HippoRag2/local_embeddings.py:255
↓ 17 callersMethodsearch
(self, query: str, k: int)
memqa/qa_agent_baselines/A-Mem/memory_layer.py:327
↓ 16 callersFunctionresolve_repo_path
(path: str)
agent_systems/config.py:49
↓ 13 callersFunctionextract_evidence_ids
(qa: Dict[str, Any])
memqa/retrieve/utils.py:140
↓ 13 callersMethodupdate
Update progress.
memqa/mem_processor/image/utils.py:293
↓ 10 callersMethodchat_with_usage
( self, messages: List[Dict[str, Any]] )
memqa/qa_agent_baselines/MMRag/llm_utils.py:195
↓ 9 callersFunctionbuild_retrieval_items
( email_entries: List[Dict[str, Any]], image_entries: List[Dict[str, Any]], video_entries: List[Di
memqa/retrieve/utils.py:282
↓ 9 callersFunctiontokenize
(text: str)
memqa/utils/evaluator/normalizer.py:590
↓ 8 callersFunction_content_to_text
(value: Any)
agent_systems/extract_answer.py:312
↓ 8 callersMethod_get_cache_path
Get cache file path for a given image and cache type.
memqa/mem_processor/image/image_processor.py:103
↓ 8 callersMethod_get_cache_path
Get cache file path for a given video and cache type.
memqa/mem_processor/video/video_processor.py:114
↓ 8 callersMethod_load_from_cache
Load data from cache if it exists.
memqa/mem_processor/image/image_processor.py:108
↓ 8 callersMethod_load_from_cache
Load data from cache if it exists.
memqa/mem_processor/video/video_processor.py:119
↓ 8 callersMethod_save_to_cache
Save data to cache.
memqa/mem_processor/image/image_processor.py:118
↓ 8 callersMethod_save_to_cache
Save data to cache.
memqa/mem_processor/video/video_processor.py:129
↓ 8 callersFunctionbuild_cache_key
(config: Dict[str, Any])
memqa/retrieve/utils.py:370
↓ 8 callersFunctionload_qa_list
(qa_data: Any)
memqa/retrieve/utils.py:104
↓ 8 callersFunctionnormalize_metadata_value
( metadata: Dict[str, Any], key: str, default: str = "Unknown" )
memqa/qa_agent_baselines/MemoryOS/memoryos_baseline.py:588
↓ 8 callersFunctionnormalize_metadata_value
( metadata: Dict[str, Any], key: str, default: str = "Unknown" )
memqa/qa_agent_baselines/SimpleMem/simplemem_baseline.py:400
↓ 8 callersMethodupdate
Update progress.
memqa/mem_processor/video/utils.py:600
↓ 8 callersFunctionwrite_jsonl
(path: Path, rows: Iterable[Dict[str, Any]])
memqa/retrieve/utils.py:122
↓ 7 callersFunction_maybe_model
(obj: Any)
agent_systems/extract_usage.py:747
↓ 7 callersFunction_sum_opt
(values: list[Optional[float]])
agent_systems/collect_usage.py:36
↓ 7 callersFunctionbatched
(values: List[Any], batch_size: int)
memqa/retrieve/retrievers.py:34
↓ 7 callersFunctionbuild_checkpoint_record
( result: Optional[Dict[str, Any]], qa_id: str, error: Optional[str] = None, )
memqa/qa_agent_baselines/mem0/mem0_baseline.py:105
↓ 7 callersFunctionbuild_text_evidence
(items: List[RetrievalItem])
memqa/qa_agent_baselines/MMRag/mmrag_utils.py:24
↓ 7 callersFunctionget_vllm_api_key
Get VLLM API key.
memqa/global_config.py:84
↓ 7 callersFunctionminimal_email_text_config
()
memqa/retrieve/utils.py:273
↓ 7 callersFunctionminimal_media_text_config
()
memqa/retrieve/utils.py:260
↓ 6 callersMethod__init__
( self, cache_dir: Path, batch_size: int = 16, device: Optional[str] = None,
memqa/retrieve/retrievers.py:68
↓ 6 callersMethod_extract_and_encode_frames
Extract frames from video and encode them as base64. Returns: Tuple of (list of base64 encoded frames, path to f
memqa/mem_processor/video/video_processor.py:179
↓ 6 callersFunction_extract_cost_usd
(d: Dict[str, Any])
agent_systems/extract_usage.py:213
↓ 6 callersFunctionextract_dates
(text: str)
memqa/utils/evaluator/normalizer.py:369
↓ 6 callersFunctionget_qa_id
(qa: Dict[str, Any])
memqa/qa_agent_baselines/mem0/mem0_baseline.py:98
↓ 6 callersFunctionnormalize_evidence_id
(value: str)
memqa/qa_agent_baselines/HippoRag2/hipporag2_baseline.py:145
↓ 6 callersFunctionnormalize_text
(text: str)
memqa/utils/evaluator/normalizer.py:220
↓ 6 callersMethodquery_vision_model
Query vision model (VLLM, OpenAI, or none) with image and prompt.
memqa/mem_processor/image/image_processor.py:245
↓ 6 callersMethodquery_vision_model_multi_frame
Query vision model with multiple frames and prompt.
memqa/mem_processor/video/video_processor.py:251
↓ 6 callersFunctionwrite_incremental_result
( path: Path, new_result: Dict[str, Any], existing_results: Dict[str, Dict[str, Any]] )
memqa/utils/evaluator/evaluate_qa.py:130
↓ 5 callersFunction_extract_usage_fields
(d: Dict[str, Any])
agent_systems/extract_usage.py:272
↓ 5 callersFunction_maybe_text
(value: Any)
agent_systems/extract_answer.py:304
↓ 5 callersFunction_pick_id
(event: Dict[str, Any])
agent_systems/extract_usage.py:240
↓ 5 callersFunction_records_from_json
(obj: Any)
agent_systems/extract_usage.py:378
↓ 5 callersFunctionclassify_evidence_id
(evidence_id: str)
memqa/qa_agent_baselines/oracle/oracle_baseline.py:420
↓ 5 callersFunctioncontent_to_text
(content: Any)
agent_systems/extract_answer.py:257
↓ 5 callersFunctiondump_json
(path: Path, payload: Any)
agent_systems/memory_variants.py:42
↓ 5 callersFunctionextract_currency_amounts
(text: str)
memqa/utils/evaluator/normalizer.py:504
↓ 5 callersFunctionextract_times
(text: str)
memqa/utils/evaluator/normalizer.py:426
↓ 5 callersFunctionget_openai_api_key
Get OpenAI API key.
memqa/global_config.py:80
↓ 5 callersFunctionis_vl_retriever
(retriever_name: str)
memqa/qa_agent_baselines/MMRag/mmrag_retrieve_rerank_answer.py:52
↓ 5 callersFunctionload_jsonl
(path: Path)
agent_systems/extract_usage.py:94
↓ 5 callersFunctionremove_date_time_text
(text: str)
memqa/utils/evaluator/normalizer.py:470
↓ 5 callersMethodsave
(self, state_path: str, embeddings_path: str)
memqa/qa_agent_baselines/A-Mem/memory_layer.py:369
↓ 4 callersFunction_as_float
(value: Any)
agent_systems/extract_usage.py:125
↓ 4 callersFunction_atm_write_checkpoint
(builder: Any, status: str = "in_progress")
memqa/qa_agent_baselines/SimpleMem/simplemem_baseline.py:1039
↓ 4 callersFunction_extract_reasoning_tokens
(usage: Any)
memqa/qa_agent_baselines/MMRag/llm_utils.py:138
↓ 4 callersFunction_extract_response_text
(response: Any)
memqa/qa_agent_baselines/MMRag/llm_utils.py:99
↓ 4 callersFunction_format_date
(year: Optional[int], month: int, day: int)
memqa/utils/evaluator/normalizer.py:353
↓ 4 callersFunction_messages_to_responses_input
( messages: List[Dict[str, Any]] )
memqa/qa_agent_baselines/MMRag/llm_utils.py:43
↓ 4 callersFunction_normalize_time
(hour: int, minute: int, ampm: Optional[str])
memqa/utils/evaluator/normalizer.py:416
↓ 4 callersFunction_records_from_jsonl
(events: Iterable[Dict[str, Any]])
agent_systems/extract_usage.py:399
↓ 4 callersFunction_responses_supports_temperature
(model: str | None)
memqa/qa_agent_baselines/MMRag/llm_utils.py:132
↓ 4 callersFunction_save_construction_cache
( constructed_notes: List[Optional[Any]], construction_cache: Path, )
memqa/qa_agent_baselines/A-Mem/amem_baseline.py:424
↓ 4 callersFunction_should_use_responses
(model: str | None, config: Dict[str, Any])
memqa/qa_agent_baselines/MMRag/llm_utils.py:124
↓ 4 callersMethodadd_passage
Add a passage and update the index mapping.
memqa/qa_agent_baselines/HippoRag2/data_adapter.py:69
↓ 4 callersFunctionanswer_question_with_memoryos
Answer a question using MemoryOS. Args: qa: QA dict with 'id' and 'question' memoryos_instance: Initialized MemoryOS instanc
memqa/qa_agent_baselines/MemoryOS/memoryos_baseline.py:1557
↓ 4 callersFunctionappend_jsonl
(path: Path, row: Dict[str, Any])
memqa/qa_agent_baselines/MMRag/mmrag_utils.py:145
↓ 4 callersFunctionbuild_text_messages
( question: str, evidence_chunks: List[str] )
memqa/qa_agent_baselines/MMRag/mmrag_utils.py:28
↓ 4 callersFunctioncollect_refusal
(value: Any)
memqa/utils/evaluator/evaluate_qa.py:389
↓ 4 callersFunctioncompute_recall
(gt_ids: List[str], retrieved_ids: List[str])
memqa/qa_agent_baselines/MMRag/mmrag_retrieve_rerank_answer.py:371
↓ 4 callersFunctioncreate_memoryos_instance
Create a MemoryOS instance with proper configuration. Args: args: Parsed arguments user_id: Unique user ID for this instance
memqa/qa_agent_baselines/MemoryOS/memoryos_baseline.py:1313
↓ 4 callersFunctioncreate_simplemem_system
( args: argparse.Namespace, api_key: str, base_url: Optional[str], clear_db: bool = True,
memqa/qa_agent_baselines/SimpleMem/simplemem_baseline.py:1232
↓ 4 callersFunctiondedupe_tokens
(tokens: Iterable)
memqa/utils/evaluator/normalizer.py:736
↓ 4 callersMethodembed
Embed texts (HippoRAG interface). Args: texts: Single text or list of texts **kwargs: Ignored, for compatibi
memqa/qa_agent_baselines/HippoRag2/local_embeddings.py:238
↓ 4 callersFunctionformat_pct
(value: float)
scripts/QA_Agent/NIAH/summarize_niah_results.py:131
↓ 4 callersFunctionindex_memories_into_memoryos
Index retrieval items into MemoryOS as dialogue pairs. Args: memoryos_instance: MemoryOS instance items: List of retrieval i
memqa/qa_agent_baselines/MemoryOS/memoryos_baseline.py:1509
↓ 4 callersFunctionis_failed_record
(record: Optional[Dict[str, Any]])
memqa/qa_agent_baselines/MMRag/mmrag_utils.py:166
↓ 4 callersMethodload
Load corpus from JSON file.
memqa/qa_agent_baselines/HippoRag2/data_adapter.py:98
↓ 4 callersFunctionload_config
()
agent_systems/config.py:239
↓ 4 callersFunctionload_existing_results
(path: Path)
memqa/utils/evaluator/evaluate_qa.py:118
↓ 4 callersFunctionload_json
(path: Path)
memqa/qa_agent_baselines/oracle/oracle_baseline.py:361
↓ 4 callersFunctionnormalize_openai_base_url
(endpoint: Optional[str])
memqa/qa_agent_baselines/SimpleMem/simplemem_baseline.py:347
↓ 4 callersMethodretrieve
Retrieve top-k items for a query. Args: query: Query string top_k: Number of top results to return Returns:
memqa/retrieve/retrievers.py:158
↓ 4 callersFunctionsafe_mtime
(path: str)
memqa/qa_agent_baselines/MMRag/mmrag_retrieve_rerank_answer.py:56
↓ 4 callersFunctionsafe_mtime
(path: str)
memqa/qa_agent_baselines/MMRag/mmrag_retrieve_answer.py:296
↓ 4 callersFunctionsave_json_file
Save data to JSON file with proper error handling.
memqa/mem_processor/email/utils.py:343
↓ 4 callersFunctionset_memoryos_answer_llm
(memoryos_instance: Any, llm_model: str)
memqa/qa_agent_baselines/MemoryOS/memoryos_baseline.py:681
↓ 4 callersFunctionsummarize_recalls
(details: List[Dict[str, Any]], key: str)
memqa/qa_agent_baselines/MMRag/mmrag_retrieve_rerank_answer.py:382
↓ 4 callersFunctionsuppress_output
(enabled: bool)
memqa/qa_agent_baselines/SimpleMem/simplemem_baseline.py:66
↓ 4 callersFunctiontokens_match
(expected: List, actual: List)
memqa/utils/evaluator/evaluate_qa.py:177
↓ 4 callersMethodupdate
Update progress counter.
memqa/mem_processor/email/utils.py:426
↓ 3 callersMethod__init__
(self, provider: str, config: Dict[str, Any])
memqa/utils/evaluator/evaluate_qa.py:449
↓ 3 callersFunction_atm_persist_entry_mapping
(builder: Any)
memqa/qa_agent_baselines/SimpleMem/simplemem_baseline.py:1064
↓ 3 callersFunction_deterministic_accuracy_core
( ground_truth: str, prediction: str, question: Optional[str] = None )
memqa/utils/evaluator/evaluate_qa.py:181
next →1–100 of 815, ranked by callers