MCPcopy Create free account

hub / github.com/DataArcTech/ToG / functions

Functions133 in github.com/DataArcTech/ToG

⨍ Functions 133◇ Types & classes 11
↓ 11 callersFunctionget_batch_files
Returns paths to files in fdir.
Wikidata/simple_wikidata_db/db_deploy/utils.py:37
↓ 10 callersMethodquery_all
(self, method, *args)
ToG/client.py:137
↓ 9 callersFunctionrun_llm
(prompt, temperature, max_tokens, opeani_api_keys, engine="gpt-3.5-turbo")
ToG/utils.py:106
↓ 8 callersFunctionsave_2_jsonl
(question, answer, cluster_chain_of_entities, file_name)
ToG/utils.py:156
↓ 6 callersFunctionhalf_stop
(question, cluster_chain_of_entities, depth, args)
ToG/wiki_func.py:164
↓ 5 callersFunctionjsonl_generator
Returns generator for jsonl file.
Wikidata/simple_wikidata_db/db_deploy/utils.py:24
↓ 5 callersMethodwrite
(self, json_value: List[Dict[str, Any]])
Wikidata/simple_wikidata_db/preprocess_utils/writer_process.py:35
↓ 4 callersFunctionentity_search
(entity, relation, wiki_client, head)
ToG/wiki_func.py:103
↓ 4 callersFunctionexecurte_sparql
(sparql_query)
ToG/freebase_func.py:22
↓ 4 callersFunctiongenerate_without_explored_paths
(question, args)
ToG/utils.py:178
↓ 3 callersFunctionid2entity_name_or_type
(entity_id)
ToG/freebase_func.py:37
↓ 2 callersFunctionabandon_rels
(relation)
ToG/wiki_func.py:46
↓ 2 callersFunctionabandon_rels
(relation)
ToG/freebase_func.py:17
↓ 2 callersFunctioncheck_string
(string)
eval/utils.py:103
↓ 2 callersFunctionclean_relations_bm25_sent
(topn_relations, topn_scores, entity_id, head_relations)
ToG/freebase_func.py:87
↓ 2 callersFunctionclean_scores
(string, entity_candidates)
ToG/utils.py:146
↓ 2 callersMethodclose
(self)
Wikidata/simple_wikidata_db/preprocess_utils/writer_process.py:50
↓ 2 callersFunctioncompute_bm25_similarity
Computes the BM25 similarity between a question and a list of relations, and returns the topn relations with the highest similarity along wit
ToG/utils.py:39
↓ 2 callersFunctionentity_prune
(total_entities_id, total_relations, total_candidates, total_topic_entities, total_head, total_scores, args, w
ToG/wiki_func.py:178
↓ 2 callersFunctionentity_score
(question, entity_candidates_id, entity_candidates, score, relation, args)
ToG/wiki_func.py:127
↓ 2 callersFunctionexact_match
(response, answers)
eval/utils.py:121
↓ 2 callersFunctionextract_answer
(text)
ToG/utils.py:163
↓ 2 callersFunctionif_all_zero
(topn_scores)
ToG/freebase_func.py:83
↓ 2 callersFunctionif_finish_list
(lst)
ToG/utils.py:184
↓ 2 callersFunctionif_true
(prompt)
ToG/utils.py:172
↓ 2 callersFunctionmerge_list_of_list
Optimized function to merge two defaultdict(list) instances. For common keys, lists will be concatenated.
Wikidata/simple_wikidata_db/db_deploy/server.py:25
↓ 2 callersFunctionprepare_dataset
(dataset_name)
ToG/utils.py:192
↓ 2 callersFunctionprocess_mainsnak
(data, language_id)
Wikidata/simple_wikidata_db/preprocess_utils/worker_process.py:61
↓ 2 callersFunctionreasoning
(question, cluster_chain_of_entities, args)
ToG/wiki_func.py:194
↓ 2 callersFunctionrelation_search_prune
(entity_id, entity_name, pre_relations, pre_head, question, args, wiki_client)
ToG/wiki_func.py:57
↓ 2 callersFunctionreplace_relation_prefix
(relations)
ToG/freebase_func.py:30
↓ 2 callersFunctionretrieve_top_docs
Retrieve the topn most relevant documents for the given query. Parameters: - query (str): The input query. - docs (list of str): The
ToG/utils.py:11
↓ 2 callersFunctionupdate_history
(entity_candidates, entity, scores, entity_candidates_id, total_candidates, total_scores, total_relations, tot
ToG/wiki_func.py:149
↓ 1 callersMethod__init__
( self, chunk_number: int, data_dir: str, num_workers: int = 400, )
Wikidata/simple_wikidata_db/db_deploy/server.py:40
↓ 1 callersFunctionalign
(dataset_name, question_string, data, ground_truth_datas)
eval/utils.py:50
↓ 1 callersFunctionall_unknown_entity
(entity_candidates)
ToG/utils.py:135
↓ 1 callersFunctionall_zero
(topn_scores)
ToG/wiki_func.py:99
↓ 1 callersFunctioncheck_end_word
(s)
ToG/wiki_func.py:41
↓ 1 callersFunctionclean_relations
(string, entity_id, head_relations)
ToG/wiki_func.py:13
↓ 1 callersFunctionclean_relations
(string, entity_id, head_relations)
ToG/freebase_func.py:60
↓ 1 callersFunctionclean_results
(string)
eval/utils.py:106
↓ 1 callersMethodclose
(self)
Wikidata/simple_wikidata_db/preprocess_utils/writer_process.py:77
↓ 1 callersFunctionconstruct_entity_score_prompt
(question, relation, entity_candidates)
ToG/wiki_func.py:53
↓ 1 callersFunctionconstruct_entity_score_prompt
(question, relation, entity_candidates)
ToG/freebase_func.py:105
↓ 1 callersFunctionconstruct_relation_prune_prompt
(question, entity_name, total_relations, args)
ToG/wiki_func.py:37
↓ 1 callersFunctionconstruct_relation_prune_prompt
(question, entity_name, total_relations, args)
ToG/freebase_func.py:101
↓ 1 callersFunctiondel_all_unknown_entity
(entity_candidates_id, entity_candidates_name)
ToG/wiki_func.py:85
↓ 1 callersFunctiondel_unknown_entity
(entity_candidates)
ToG/utils.py:139
↓ 1 callersFunctiongenerate_answer
(question, cluster_chain_of_entities, args)
ToG/wiki_func.py:170
↓ 1 callersFunctiongenerate_answer
(question, cluster_chain_of_entities, args)
ToG/freebase_func.py:219
↓ 1 callersFunctionget_arg_parser
()
Wikidata/simple_wikidata_db/preprocess_dump.py:23
↓ 1 callersFunctionif_all_zero
(topn_scores)
ToG/utils.py:88
↓ 1 callersFunctionjsonl_to_json
(jsonl_file, json_file)
tools/jsonl2json.py:3
↓ 1 callersFunctionmain
()
Wikidata/simple_wikidata_db/preprocess_dump.py:36
↓ 1 callersFunctionmain
(args)
Wikidata/simple_wikidata_db/db_deploy/build_index.py:92
↓ 1 callersFunctionprepare_dataset
(dataset_name)
CoT/utils.py:34
↓ 1 callersFunctionprepare_dataset_for_eval
(dataset_name, output_file)
eval/utils.py:5
↓ 1 callersFunctionprocess_json
(obj, language_id="en")
Wikidata/simple_wikidata_db/preprocess_utils/worker_process.py:87
↓ 1 callersMethodquery_all
(self, method, *args)
Wikidata/simple_wikidata_db/db_deploy/client.py:97
↓ 1 callersFunctionreplace_entities_prefix
(entities)
ToG/freebase_func.py:33
↓ 1 callersFunctionrun_llm
(prompt, temperature, max_tokens, opeani_api_keys, engine="gpt-3.5-turbo")
CoT/utils.py:5
↓ 1 callersFunctionsave_result2json
(dataset_name, num_right, num_error, total_nums, method)
eval/utils.py:129
↓ 1 callersMethodserve_forever
(self)
Wikidata/simple_wikidata_db/db_deploy/server.py:202
↓ 1 callersMethodtest_connections
(self)
Wikidata/simple_wikidata_db/db_deploy/client.py:73
↓ 1 callersFunctiontransform_relation
(relation)
ToG/wiki_func.py:8
↓ 1 callersMethodwrite
(self, json_object: Dict[str, Any])
Wikidata/simple_wikidata_db/preprocess_utils/writer_process.py:64
Method__init__
(self, url: str)
ToG/client.py:12
Method__init__
(self, urls: tp.List[str])
ToG/client.py:104
Method__init__
(self, url: str)
Wikidata/simple_wikidata_db/db_deploy/client.py:12
Method__init__
(self, urls: tp.List[str])
Wikidata/simple_wikidata_db/db_deploy/client.py:64
Method__init__
(self, addr, server_args, requestHandler=RequestHandler)
Wikidata/simple_wikidata_db/db_deploy/server.py:180
Method__init__
(self, path: Path, batch_size: int, table_name: str)
Wikidata/simple_wikidata_db/preprocess_utils/writer_process.py:23
Method__init__
(self, path: Path, batch_size: int)
Wikidata/simple_wikidata_db/preprocess_utils/writer_process.py:55
Functiona_factory
()
Wikidata/simple_wikidata_db/db_deploy/utils.py:20
Functionappend_to_jsonl_file
Appends json dictionary as new line to file
Wikidata/simple_wikidata_db/utils.py:39
Functionbatch_line_generator
Returns generator for jsonl file with batched lines
Wikidata/simple_wikidata_db/utils.py:20
Functioncheck_end_word
(s)
ToG/freebase_func.py:13
Functioncheck_refuse
(string)
eval/utils.py:116
Functionclean_relations
(string, entity_id, head_relations)
ToG/utils.py:65
Functionclean_relations_bm25_sent
(topn_relations, topn_scores, entity_id, head_relations)
ToG/utils.py:92
Functioncount_lines
(input_file: Path, max_lines_to_read: int)
Wikidata/simple_wikidata_db/preprocess_utils/reader_process.py:6
Functioncreate_dir
Creates new directory if it doesn't already exist
Wikidata/simple_wikidata_db/utils.py:53
Functionentity_prune
(total_entities_id, total_relations, total_candidates, total_topic_entities, total_head, total_scores, args)
ToG/freebase_func.py:227
Functionentity_score
(question, entity_candidates_id, score, relation, args)
ToG/freebase_func.py:166
Functionentity_search
(entity, relation, head=True)
ToG/freebase_func.py:152
Functionextract_content
(s)
eval/utils.py:140
Functionfilter_value
( dict_list: tp.List[tp.Dict], key: str, )
Wikidata/simple_wikidata_db/db_deploy/build_index.py:79
Methodget_all_relations_of_an_entity
( self, entity_qid: str )
ToG/client.py:28
Methodget_all_relations_of_an_entity
( self, entity_qid: str )
Wikidata/simple_wikidata_db/db_deploy/client.py:28
Methodget_all_relations_of_an_entity
( self, entity_qid: str )
Wikidata/simple_wikidata_db/db_deploy/server.py:142
Functionget_batch_files
Returns paths to files in fdir
Wikidata/simple_wikidata_db/utils.py:46
Methodget_external_id_given_head_and_relation
( self, head_qid: str, relation_pid: str )
ToG/client.py:47
Methodget_external_id_given_head_and_relation
( self, head_qid: str, relation_pid: str )
Wikidata/simple_wikidata_db/db_deploy/client.py:47
Methodget_external_id_given_head_and_relation
( self, head_qid: str, relation_pid: str )
Wikidata/simple_wikidata_db/db_deploy/server.py:166
Methodget_tail_entities_given_head_and_relation
( self, head_qid: str, relation_pid: str )
ToG/client.py:33
Methodget_tail_entities_given_head_and_relation
( self, head_qid: str, relation_pid: str )
Wikidata/simple_wikidata_db/db_deploy/client.py:33
Methodget_tail_entities_given_head_and_relation
( self, head_qid: str, relation_pid: str )
Wikidata/simple_wikidata_db/db_deploy/server.py:150
Methodget_tail_values_given_head_and_relation
( self, head_qid: str, relation_pid: str )
ToG/client.py:40
Methodget_tail_values_given_head_and_relation
( self, head_qid: str, relation_pid: str )
Wikidata/simple_wikidata_db/db_deploy/client.py:40
Methodget_tail_values_given_head_and_relation
( self, head_qid: str, relation_pid: str )
Wikidata/simple_wikidata_db/db_deploy/server.py:158
next →1–100 of 133, ranked by callers