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
↓ 2 callersFunctionprocess_dev_json
(args, proj_dir)
spider2-lite/baselines/dailsql/preprocessed_data/spider2_preprocess.py:79
↓ 2 callersFunctionprocess_duplication
(sql)
spider2-lite/baselines/dailsql/utils/post_process.py:23
↓ 2 callersFunctionprocess_table_json
(args, proj_dir)
spider2-lite/baselines/dailsql/preprocessed_data/spider2_preprocess.py:60
↓ 2 callersFunctionsplit
(s: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:67
↓ 2 callersFunctionsplit_sql
(sql: str)
methods/ReFoRCE/utils.py:235
↓ 2 callersFunctionsql_split
(s)
spider2-lite/baselines/dailsql/utils/utils.py:373
↓ 2 callersFunctionstrip_char
(s)
methods/SL/gen_sl/utils.py:53
↓ 2 callersMethodtable_back
(self, parsed)
methods/SL/gen_sl/naive_parser.py:174
↓ 2 callersFunctionunorder_row
(row: Tuple)
spider2-lite/baselines/dailsql/utils/post_process.py:37
↓ 2 callersMethodupdate_cols
(self, used_tbs, all_cols)
methods/SL/gen_sl/naive_parser.py:150
↓ 2 callersFunctionupdate_db_id
(full_name, db_id)
methods/SL/gen_sl/naive_parser.py:8
↓ 2 callersFunctionvectors_match
(v1, v2, tol=tolerance, ignore_order_=False)
spider2-lite/evaluation_suite/evaluate_utils.py:89
↓ 2 callersFunctionvectors_match
(v1, v2, tol=tolerance, ignore_order_=False)
spider2-lite/evaluation_suite/evaluate.py:91
↓ 2 callersFunctionvectors_match
(v1, v2, tol=tolerance, ignore_order_=False)
spider2-snow/evaluation_suite/evaluate_utils.py:89
↓ 2 callersFunctionvectors_match
(v1, v2, tol=tolerance, ignore_order_=False)
spider2-snow/evaluation_suite/evaluate.py:95
↓ 2 callersFunctionvectors_match
(v1, v2, tol=tolerance, ignore_order_=False)
methods/ReFoRCE/eval.py:56
↓ 2 callersFunctionvectors_match
(v1, v2, tol=tolerance, ignore_order_=False)
methods/ReFoRCE/utils.py:112
↓ 2 callersFunctionwalk_metadata
(dev)
spider2-lite/baselines/utils/utils.py:134
↓ 1 callersFunctionGPT4o_debug
(prompt, n=1, temperature=0)
spider2-lite/baselines/dinsql/DIN-SQL.py:758
↓ 1 callersMethod__init__
(self, path_data, args)
spider2-lite/baselines/dailsql/utils/data_builder.py:12
↓ 1 callersMethod_attn
(self, query, key, value, attention_mask=None, head_mask=None)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:214
↓ 1 callersMethod_preprocess_schema
(self, schema)
spider2-lite/baselines/dailsql/utils/linking_process.py:190
↓ 1 callersMethod_tokenize_for_copying
(self, presplit, unsplit)
spider2-lite/baselines/dailsql/utils/linking_process.py:203
↓ 1 callersFunctionadd_agent_setting
()
methods/ReFoRCE/spider_agent_setup_lite.py:85
↓ 1 callersFunctionadd_asc
(s)
spider2-lite/baselines/dailsql/utils/utils.py:365
↓ 1 callersMethodadd_item
(self, item, schema, section, validation_info)
spider2-lite/baselines/dailsql/utils/linking_process.py:148
↓ 1 callersFunctionadd_snowflake_agent_setting
()
methods/ReFoRCE/spider_agent_setup_snow.py:66
↓ 1 callersMethodadd_sqls
(self, sqls, CE_PTHS, example_id)
methods/SL/gen_sl/naive_parser.py:98
↓ 1 callersMethodadd_sqls_from_json_omni
(self, sqls, responses, question_id)
methods/SL/gen_sl/naive_parser.py:129
↓ 1 callersFunctionask_chat
(model, messages: list, temperature, n, max_tokens)
spider2-lite/baselines/dailsql/llm/chatgpt.py:41
↓ 1 callersFunctionask_llm
(model: str, batch: list, temperature: float, n:int, max_tokens)
spider2-lite/baselines/dailsql/llm/chatgpt.py:98
↓ 1 callersFunctionask_model_sl
(example_path, json_save_pth)
methods/ReFoRCE/schema_linking.py:180
↓ 1 callersFunctionask_model_sl_
(tb_info, task, chat_session)
methods/ReFoRCE/schema_linking.py:207
↓ 1 callersFunctioncheck_sql_executability
(generated_sql, db)
spider2-lite/baselines/codes/utils/db_utils.py:33
↓ 1 callersFunctioncheck_table_names
(ddl_path)
methods/ReFoRCE/reconstruct_data.py:97
↓ 1 callersFunctionclassification_prompt_maker
(test_sample_text, db_names, schema_links, spider_schema, spider_foreign)
spider2-lite/baselines/dinsql/DIN-SQL.py:594
↓ 1 callersFunctionclear_byte
(rows)
methods/ReFoRCE/utils.py:369
↓ 1 callersFunctionclear_folder
(folder_path)
methods/ReFoRCE/spider_agent_setup_lite.py:14
↓ 1 callersFunctionclear_folder
(folder_path)
methods/ReFoRCE/spider_agent_setup_snow.py:12
↓ 1 callersFunctionclear_sample_rows
(text, byte_limit=1000)
methods/ReFoRCE/utils.py:238
↓ 1 callersFunctioncolumn_description_length_histogram
1106 status: deprecated. should be compatible with the new metadata format.
spider2-lite/baselines/utils/utils.py:44
↓ 1 callersFunctioncombine_table_dev_json
(args)
spider2-lite/baselines/codes/preprocessed_data/spider2_preprocess.py:141
↓ 1 callersFunctioncompare_multi_pandas_table
(pred, multi_gold, multi_condition_cols=[], multi_ignore_order=False)
methods/ReFoRCE/eval.py:30
↓ 1 callersFunctioncompare_pandas_table
_summary_ Args: pred (Dataframe): _description_ gold (Dataframe): _description_ condition_cols (list, optional): _descrip
spider2-lite/evaluation_suite/evaluate_utils.py:75
↓ 1 callersFunctioncompare_pandas_table
_summary_ Args: pred (Dataframe): _description_ gold (Dataframe): _description_ condition_cols (list, optional): _descrip
methods/ReFoRCE/utils.py:100
↓ 1 callersFunctioncompare_sql
(predicted_sql, ground_truth, db_path)
spider2-lite/baselines/codes/evaluate_ex.py:24
↓ 1 callersFunctioncompute_cell_value_linking
(tokens, schema)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:84
↓ 1 callersFunctioncompute_metrics
(gen_set, gold_set)
methods/SL/gen_sl/utils.py:4
↓ 1 callersFunctioncompute_metrics_sl
(file_pth, db_path, threshold)
methods/ReFoRCE/schema_linking.py:236
↓ 1 callersFunctioncompute_precision_recall
(predicted: set, ground_truth: set)
methods/ReFoRCE/utils.py:276
↓ 1 callersFunctioncompute_schema_linking
(question, column, table)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:18
↓ 1 callersFunctioncreating_schema
(DATASET_JSON)
spider2-lite/baselines/dinsql/DIN-SQL.py:659
↓ 1 callersFunctiondb_stats
(db_stats_list)
spider2-lite/baselines/utils/utils.py:78
↓ 1 callersFunctiondb_stats_bar_chart
(db_stats_list)
spider2-lite/baselines/utils/utils.py:95
↓ 1 callersFunctiondb_word_exact_match
(word, column, table, db_conn)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:107
↓ 1 callersFunctiondb_word_partial_match
(word, column, table, db_conn)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:92
↓ 1 callersFunctiondebuger
(test_sample_text, db_names, sql, spider_schema, spider_primary, spider_foreign)
spider2-lite/baselines/dinsql/DIN-SQL.py:695
↓ 1 callersFunctiondouble2single
(s)
spider2-lite/baselines/dailsql/utils/utils.py:362
↓ 1 callersMethodevaluate_coverage
(self, dataset)
spider2-lite/baselines/codes/schema_item_filter.py:332
↓ 1 callersFunctionevaluate_passk
(pth, task, update_res=False)
methods/ReFoRCE/eval.py:151
↓ 1 callersFunctionevaluate_spider2sql
(args)
spider2-snow/evaluation_suite/evaluate_yx.py:235
↓ 1 callersFunctionexec_on_bigquery_
(query: str, instance_id: str)
spider2-lite/baselines/dailsql/utils/post_process.py:154
↓ 1 callersFunctionexec_on_db
( query: str, sqlite_path: str, instance_id: str, process_id: str = "", timeout: int = TIMEOUT )
spider2-lite/baselines/dailsql/utils/post_process.py:184
↓ 1 callersFunctionexec_on_db_
(query: str, sqlite_path: str)
spider2-lite/baselines/dailsql/utils/post_process.py:169
↓ 1 callersMethodexec_sql_bq
(self, sql_query, save_path, max_len)
methods/ReFoRCE/sql.py:106
↓ 1 callersMethodexec_sql_sf
(self, sql_query, save_path, max_len, ex_id)
methods/ReFoRCE/sql.py:85
↓ 1 callersFunctionexecute
(question, table_info, args, csv_save_path, log_save_path, sql_save_path, search_directory, format_csv, sql_da
methods/ReFoRCE/run.py:13
↓ 1 callersFunctionexecute_sql_long_time_limitation
(cursor, sql)
spider2-lite/baselines/codes/utils/db_utils.py:28
↓ 1 callersMethodexecute_sqlite_with_timeout
(self, sql_query, save_path, max_len, sqlite_path, timeout=300)
methods/ReFoRCE/sql.py:151
↓ 1 callersMethodexecute_sqls
(self, sqls, logger)
methods/ReFoRCE/agent.py:41
↓ 1 callersMethodexploration
(self, task, table_struct, table_info, logger)
methods/ReFoRCE/agent.py:129
↓ 1 callersFunctionextract_all_blocks
(main_content, code_format)
methods/ReFoRCE/utils.py:12
↓ 1 callersFunctionextract_column_names
(sql: str)
methods/ReFoRCE/utils.py:296
↓ 1 callersFunctionextract_real_table_names
(sql: str, dialect: str = "bigquery")
methods/ReFoRCE/utils.py:339
↓ 1 callersFunctionextract_sql_query
(pred_sql_query)
spider2-snow/evaluation_suite/evaluate.py:174
↓ 1 callersFunctionfilter_bijection_like_dict
(d)
methods/ReFoRCE/utils.py:321
↓ 1 callersFunctionfind_primary_keys_MYSQL_like
(db_name, spider_primary)
spider2-lite/baselines/dinsql/DIN-SQL.py:644
↓ 1 callersMethodformat_answer
(self, task, chat_session)
methods/ReFoRCE/agent.py:123
↓ 1 callersMethodformat_example
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:11
↓ 1 callersFunctionformat_foreign_key
(table_name: str, res: list)
spider2-lite/baselines/dailsql/utils/utils.py:140
↓ 1 callersMethodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:21
↓ 1 callersMethodformat_question
(self, example: dict, args)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:46
↓ 1 callersMethodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:179
↓ 1 callersMethodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:300
↓ 1 callersMethodformat_question
(self, example: dict, args)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:450
↓ 1 callersMethodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:475
↓ 1 callersMethodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:502
↓ 1 callersMethodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:529
↓ 1 callersMethodformat_target
(self, example: dict, args)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:16
↓ 1 callersMethodgen
(self, args, logger, question, format_csv, table_struct, table_info, response_pre_txt, pre_info, csv_save_path
methods/ReFoRCE/agent.py:267
↓ 1 callersFunctionget_bigquery_sql_result
is_save = True, output a 'result.csv' if_save = False, output a string
spider2-lite/evaluation_suite/evaluate.py:127
↓ 1 callersFunctionget_column_contents
(column_name, table_name, cursor)
spider2-lite/baselines/codes/utils/db_utils.py:67
↓ 1 callersFunctionget_column_picklist
(table_name: str, column_name: str, db_path: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:194
↓ 1 callersFunctionget_constraint_permutation
(tab1_sets_by_columns: List[Set], result2: List[Tuple])
spider2-lite/baselines/dailsql/utils/post_process.py:68
↓ 1 callersFunctionget_csv_from_dic
(folder_names, output_dic, file_type)
methods/ReFoRCE/get_metadata.py:12
↓ 1 callersFunctionget_cursor_from_path
(sqlite_path: str)
spider2-lite/baselines/dailsql/utils/post_process.py:141
↓ 1 callersMethodget_databases
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:36
↓ 1 callersFunctionget_db_id
(db_path, ex_id)
methods/ReFoRCE/utils.py:204
↓ 1 callersFunctionget_dictionary
(db_path, task, json_path)
methods/SL/gen_sl/mk_prompt_spider2.py:69
↓ 1 callersFunctionget_effective_match_source
(s: str, start: int, end: int)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:87
← previousnext →101–200 of 554, ranked by callers