MCPcopy Create free account

hub / github.com/Snowflake-Labs/ReFoRCE / functions

Functions554 in github.com/Snowflake-Labs/ReFoRCE

⨍ Functions 554◇ Types & classes 83↳ Endpoints 1
↓ 72 callersMethodformat
(self, target: dict, max_seq_len: int, max_ans_len: int, scope_factor: int, cross_domain=False, args=None)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:44
↓ 56 callersMethodload
(self, sections)
spider2-lite/baselines/dailsql/utils/linking_process.py:221
↓ 35 callersFunctionremove_digits
(s)
methods/ReFoRCE/utils.py:176
↓ 23 callersMethodwrite
(self, message)
spider2-lite/evaluation_suite/evaluate.py:25
↓ 18 callersMethodclose
(self)
spider2-lite/evaluation_suite/evaluate.py:33
↓ 18 callersMethodget_extra_info
(self, db_id)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:24
↓ 14 callersFunctionclear_tb
(tb)
methods/SL/gen_sl/utils.py:31
↓ 13 callersMethodto
(self, device)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:105
↓ 11 callersFunctioncheck_length
(existing_prompt, new, future_prompt)
spider2-lite/baselines/dinsql/DIN-SQL.py:496
↓ 10 callersMethod__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:9
↓ 10 callersMethodget_model_response
(self, prompt, code_format=None)
methods/ReFoRCE/chat.py:15
↓ 10 callersFunctionjaccard_similarity
(skeleton1, skeleton2)
spider2-lite/baselines/dailsql/utils/utils.py:559
↓ 9 callersMethodexecute_sql_api
(self, sql_query, ex_id, save_path=None, api="sqlite", max_len=30000, sqlite_path=None, timeout=300)
methods/ReFoRCE/sql.py:133
↓ 8 callersMethod__init__
(self, intermediate_size, config)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:363
↓ 8 callersFunctionmask_question_with_schema_linking
(data_jsons, mask_tag, value_tag)
spider2-lite/baselines/dailsql/utils/linking_utils/application.py:3
↓ 7 callersFunctiondetect_special_char
(name)
spider2-lite/baselines/codes/utils/db_utils.py:57
↓ 7 callersFunctionhard_cut
(str_e, length=0)
methods/ReFoRCE/utils.py:34
↓ 6 callersFunctioncompare_pandas_table
_summary_ Args: pred (Dataframe): _description_ gold (Dataframe): _description_ condition_cols (list, optional): _descrip
spider2-snow/evaluation_suite/evaluate_utils.py:75
↓ 6 callersFunctionfind_fields_MYSQL_like
(db_name, spider_schema)
spider2-lite/baselines/dinsql/DIN-SQL.py:633
↓ 6 callersFunctionis_span_separator
(c: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:63
↓ 5 callersFunctionadd_quotation_mark
(s)
spider2-lite/baselines/codes/utils/db_utils.py:64
↓ 5 callersFunctioncompare_pandas_table
_summary_ Args: pred (Dataframe): _description_ gold (Dataframe): _description_ condition_cols (list, optional): _de
spider2-lite/evaluation_suite/evaluate.py:77
↓ 5 callersFunctionfind_foreign_keys_MYSQL_like
(db_name, spider_foreign)
spider2-lite/baselines/dinsql/DIN-SQL.py:652
↓ 5 callersMethodget_mini_index
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:69
↓ 4 callersFunctionGPT4o_generation
(prompt, n=1, temperature=0)
spider2-lite/baselines/dinsql/DIN-SQL.py:728
↓ 4 callersMethodclose
(self)
spider2-snow/evaluation_suite/evaluate.py:39
↓ 4 callersFunctioncompare_multi_pandas_table
(pred, multi_gold, multi_condition_cols=[], multi_ignore_order=False)
spider2-lite/evaluation_suite/evaluate.py:60
↓ 4 callersFunctionexecute_query
Execute queries and return results. Reuse cur if it's not None.
spider2-lite/baselines/dailsql/utils/utils.py:112
↓ 4 callersFunctionexecute_sql
(cursor, sql)
spider2-lite/baselines/codes/utils/db_utils.py:21
↓ 4 callersFunctionfilter_schema
(dataset, dataset_type, sic, num_top_k_tables = 5, num_top_k_columns = 5)
spider2-lite/baselines/codes/schema_item_filter.py:180
↓ 4 callersFunctionget_api_name
(sql_data)
methods/ReFoRCE/utils.py:166
↓ 4 callersFunctionget_sqlite_path
(db_path="", sql_data=None, db_id=None, task=None)
methods/ReFoRCE/utils.py:215
↓ 4 callersMethodget_table_json
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:59
↓ 4 callersFunctionload_json
(file_path)
spider2-lite/baselines/utils/post_utils.py:5
↓ 4 callersFunctionmask
(question_toks, mask_ids, tag)
spider2-lite/baselines/dailsql/utils/linking_utils/application.py:15
↓ 3 callersMethodannotate
(self, text, annotators=None, output_format=None, properties=None)
spider2-lite/baselines/dailsql/utils/linking_utils/corenlp.py:47
↓ 3 callersFunctionclear_name
(table_names, do_remove_digits=True)
methods/ReFoRCE/utils.py:353
↓ 3 callersFunctionclear_tb
(tb)
methods/ReFoRCE/utils.py:376
↓ 3 callersFunctioncompare_multi_pandas_table
(pred, multi_gold, multi_condition_cols=[], multi_ignore_order=False)
spider2-snow/evaluation_suite/evaluate_utils.py:60
↓ 3 callersFunctioncompare_pandas_table
_summary_ Args: pred (Dataframe): _description_ gold (Dataframe): _description_ condition_cols (list, optional): _descrip
spider2-snow/evaluation_suite/evaluate.py:81
↓ 3 callersFunctionget_db_schema_sequence
(schema)
spider2-lite/baselines/codes/utils/db_utils.py:76
↓ 3 callersFunctionget_matched_content_sequence
(matched_contents)
spider2-lite/baselines/codes/utils/db_utils.py:120
↓ 3 callersFunctionget_sql_for_database_from_tables_json
(db_id, tables_json, use_column_desc=False)
spider2-lite/baselines/dailsql/utils/utils.py:238
↓ 3 callersFunctionget_table_names
Get names of all tables within the database, and reuse cur if it's not None
spider2-lite/baselines/dailsql/utils/utils.py:165
↓ 3 callersMethodget_train_json
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:98
↓ 3 callersFunctionis_commonword
(s: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:49
↓ 3 callersFunctionis_stopword
(s: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:45
↓ 3 callersFunctionlista_contains_listb
(lista, listb)
spider2-lite/baselines/codes/schema_item_filter.py:268
↓ 3 callersFunctionmain_postprocess
(root_path, dev_json_path, table_json_path, method)
spider2-lite/baselines/utils/post_utils.py:74
↓ 3 callersFunctionreplace_table_names
(sql_content, selected_tables_to_projDataset)
spider2-lite/baselines/utils/post_utils.py:9
↓ 3 callersFunctionsearch_file
(directory, target_file)
methods/ReFoRCE/utils.py:43
↓ 2 callersMethod_get_mask_value
(self, device, dtype)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:208
↓ 2 callersFunctionclear_description
(table_info)
methods/ReFoRCE/utils.py:147
↓ 2 callersMethodclose_db
(self)
methods/ReFoRCE/sql.py:48
↓ 2 callersFunctioncompare_multi_pandas_table
(pred, multi_gold, multi_condition_cols=[], multi_ignore_order=False)
spider2-snow/evaluation_suite/evaluate.py:66
↓ 2 callersFunctioncompare_pandas_table
_summary_ Args: pred (Dataframe): _description_ gold (Dataframe): _description_ condition_cols (list, optional): _descrip
methods/ReFoRCE/eval.py:42
↓ 2 callersFunctioncompress_ddl
(example_folder, add_description=False, add_sample_rows=False, rm_digits=False, schema_linked=False, clear_lon
methods/ReFoRCE/reconstruct_data.py:137
↓ 2 callersMethodcompute_batch_loss
(self, batch_logits, batch_labels, batch_size)
spider2-lite/baselines/codes/utils/classifier_loss.py:44
↓ 2 callersMethodcompute_metrics
(self, gen_set, gold_set)
methods/SL/gen_sl/naive_parser.py:193
↓ 2 callersFunctioncost_estimate
(n_tokens: int, model)
spider2-lite/baselines/dailsql/utils/utils.py:186
↓ 2 callersFunctioncount_tokens
(string: str, tokenizer_type: str=None, tokenizer=None)
spider2-lite/baselines/dailsql/utils/utils.py:320
↓ 2 callersMethoddata_pre_process
(self, datas, linking_infos=None, pre_queries=None)
spider2-lite/baselines/dailsql/utils/data_builder.py:190
↓ 2 callersFunctionevaluate_spider2sql
(args)
spider2-lite/evaluation_suite/evaluate.py:223
↓ 2 callersFunctionevaluate_spider2sql
(args)
spider2-snow/evaluation_suite/evaluate.py:319
↓ 2 callersFunctionexact_match
(x_list, y_list)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:30
↓ 2 callersFunctionexecute_sql
(cursor, sql)
spider2-lite/baselines/codes/evaluate_ex.py:18
↓ 2 callersFunctionextract_code_blocks
(text: str, tag: str)
methods/SL/gen_sl/utils.py:22
↓ 2 callersFunctionextract_skeleton
(text)
spider2-lite/baselines/codes/text2sql_few_shot.py:49
↓ 2 callersMethodget_all_tables
(self, ex_pth, db_id)
methods/SL/gen_sl/naive_parser.py:62
↓ 2 callersFunctionget_bigquery_sql_result
is_save = True, output a 'result.csv' if_save = False, output a string
spider2-snow/evaluation_suite/evaluate_utils.py:153
↓ 2 callersMethodget_columns
(self, sql, all_cols, dialect="snowflake")
methods/SL/gen_sl/naive_parser.py:37
↓ 2 callersMethodget_condition_onmit_tables
(self)
methods/ReFoRCE/prompt.py:31
↓ 2 callersMethodget_csv
(self, columns, rows)
methods/ReFoRCE/sql.py:27
↓ 2 callersFunctionget_cursor_from_path
(sqlite_path)
spider2-lite/baselines/codes/utils/db_utils.py:7
↓ 2 callersFunctionget_dictionary
(db_path, task)
methods/ReFoRCE/utils.py:187
↓ 2 callersMethodget_message_len
(self)
methods/ReFoRCE/chat.py:44
↓ 2 callersMethodget_pre_skeleton
(self, queries=None, schemas=None, mini_set=False)
spider2-lite/baselines/dailsql/utils/data_builder.py:83
↓ 2 callersMethodget_prompt_decimal_places
(self)
methods/ReFoRCE/prompt.py:42
↓ 2 callersMethodget_prompt_dialect_basic
(self, api)
methods/ReFoRCE/prompt.py:57
↓ 2 callersMethodget_prompt_dialect_list_all_tables
(self, table_struct, api)
methods/ReFoRCE/prompt.py:33
↓ 2 callersMethodget_response
(self, prompt)
methods/ReFoRCE/chat.py:12
↓ 2 callersMethodget_rows
(self, cursor, max_len)
methods/ReFoRCE/sql.py:16
↓ 2 callersFunctionget_schema
(tables_and_columns)
spider2-lite/baselines/codes/schema_item_filter.py:67
↓ 2 callersMethodget_self_refine_prompt
(self, table_info, task, pre_info, question, api, format_csv, table_struct, omnisql_format_pth=None)
methods/ReFoRCE/prompt.py:112
↓ 2 callersFunctionget_snowflake_sql_result
is_save = True, output a 'result.csv' if_save = False, output a string
spider2-snow/evaluation_suite/evaluate_utils.py:125
↓ 2 callersFunctionget_special_function_summary
(data)
spider2-lite/baselines/utils/utils.py:17
↓ 2 callersFunctionget_sqls
(result, select_number, db_dir, instance_id)
spider2-lite/baselines/dailsql/utils/post_process.py:242
↓ 2 callersMethodget_tables
(self, db_id)
spider2-lite/baselines/dailsql/utils/data_builder.py:46
↓ 2 callersMethodget_tables
(self, sql, dialect="snowflake")
methods/SL/gen_sl/naive_parser.py:18
↓ 2 callersFunctionget_tuple
(csv_str)
methods/ReFoRCE/eval.py:114
↓ 2 callersFunctionget_values_from_table
(csv_data_str)
methods/ReFoRCE/utils.py:40
↓ 2 callersFunctionlist_to_dict
(tokens)
spider2-lite/baselines/dailsql/utils/utils.py:564
↓ 2 callersFunctionload_jsonl_to_dict
(jsonl_file)
spider2-lite/evaluation_suite/evaluate.py:44
↓ 2 callersFunctionload_jsonl_to_dict
(jsonl_file)
spider2-snow/evaluation_suite/evaluate_utils.py:44
↓ 2 callersFunctionload_jsonl_to_dict
(jsonl_file)
spider2-snow/evaluation_suite/evaluate.py:50
↓ 2 callersMethodmodel_vote
(self, result, sql_paths, search_directory, args, table_info, task)
methods/ReFoRCE/agent.py:289
↓ 2 callersMethodparse_cols
(self, sqls, parsed, all_cols, dialect="snowflake")
methods/SL/gen_sl/naive_parser.py:159
↓ 2 callersMethodparse_tbs
(self, parsed, sqls, all_tbs, dialect="snowflake", db_id=None)
methods/SL/gen_sl/naive_parser.py:138
↓ 2 callersFunctionpartial_match
(x_list, y_list)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:19
↓ 2 callersMethodpredict
(self, test_sample)
spider2-lite/baselines/codes/schema_item_filter.py:324
next →1–100 of 554, ranked by callers