MCPcopy Create free account

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

Functions554 in github.com/Snowflake-Labs/ReFoRCE

↓ 1 callersFunctionget_example_format_cls
(example_format: str)
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:52
↓ 1 callersMethodget_example_prefix
(self)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:8
↓ 1 callersMethodget_example_quality
(self)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:20
↓ 1 callersFunctionget_example_selector
(selector_type: str)
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:70
↓ 1 callersMethodget_examples
(self, question, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:16
↓ 1 callersFunctionget_exec_output
( sql: str, db: str, instance_id: str, plug_value: bool = False, keep
spider2-lite/baselines/dailsql/utils/post_process.py:210
↓ 1 callersMethodget_exploration_prompt
(self, api, table_struct)
methods/ReFoRCE/prompt.py:83
↓ 1 callersMethodget_exploration_refine_prompt
(self, sql, corrected_sql, sqls, res)
methods/ReFoRCE/prompt.py:106
↓ 1 callersMethodget_exploration_self_correct_prompt
(self, sql, error)
methods/ReFoRCE/prompt.py:109
↓ 1 callersFunctionget_external
(text)
methods/ReFoRCE/utils.py:270
↓ 1 callersFunctionget_foreign_key
(table_name, path_db=None, cur=None)
spider2-lite/baselines/dailsql/utils/utils.py:150
↓ 1 callersMethodget_format_prompt
(self)
methods/ReFoRCE/prompt.py:76
↓ 1 callersFunctionget_matched_entries
( s: str, field_values: List[str], m_theta: float = 0.85, s_theta: float = 0.85 )
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:121
↓ 1 callersFunctionget_metrics
(ce_json)
methods/SL/gen_sl/utils.py:74
↓ 1 callersMethodget_pattern_similarity
(self)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:38
↓ 1 callersFunctionget_primary_key
(table_name, path_db=None, cur=None)
spider2-lite/baselines/dailsql/utils/utils.py:156
↓ 1 callersMethodget_prompt_dialect_nested
(self, api)
methods/ReFoRCE/prompt.py:48
↓ 1 callersFunctionget_questionTok_and_gold_query_for_dev_json
(data)
spider2-lite/baselines/dailsql/preprocessed_data/spider2_preprocess.py:80
↓ 1 callersFunctionget_question_pattern_with_schema_linking
(data_jsons)
spider2-lite/baselines/dailsql/utils/linking_utils/application.py:38
↓ 1 callersFunctionget_repr_cls
(repr_type: str)
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:10
↓ 1 callersFunctionget_sample_rows_for_database_from_tables_json
(db_id, tables_json)
spider2-lite/baselines/dailsql/utils/utils.py:209
↓ 1 callersFunctionget_score
(final_results)
spider2-snow/evaluation_suite/evaluate_yx.py:188
↓ 1 callersMethodget_self_consistency_prompt
(self, task, format_csv)
methods/ReFoRCE/prompt.py:149
↓ 1 callersFunctionget_sequence_length
(text, tables_and_columns, tokenizer)
spider2-lite/baselines/codes/schema_item_filter.py:80
↓ 1 callersFunctionget_snowflake_sql_result
is_save = True, output a 'result.csv' if_save = False, output a string
spider2-lite/evaluation_suite/evaluate.py:165
↓ 1 callersFunctionget_snowflake_sql_result
(sql_query, database_id, is_save, save_dir=None, file_name="result.csv", timeout=30)
spider2-snow/evaluation_suite/evaluate.py:132
↓ 1 callersFunctionget_sqlite_data
(path, entry, add_description=False, add_sample_rows=False, gold_table_names=None, gold_column_names=None)
methods/ReFoRCE/reconstruct_data.py:323
↓ 1 callersFunctionget_sqlite_result
(db_path, query, save_dir=None, file_name="result.csv", chunksize=500)
spider2-lite/evaluation_suite/evaluate.py:194
↓ 1 callersFunctionget_sqlite_result
(db_path, query, save_dir=None, file_name="result.csv", chunksize=500)
spider2-snow/evaluation_suite/evaluate_utils.py:189
↓ 1 callersFunctionget_table_info
(test_path, sql_data, api, clear_des=False, full_tb_info=None)
methods/ReFoRCE/utils.py:150
↓ 1 callersFunctionget_tables
(path_db)
spider2-lite/baselines/dailsql/utils/utils.py:18
↓ 1 callersFunctionget_tables_from_tables_json
(db_id, tables_json)
spider2-lite/baselines/dailsql/utils/utils.py:54
↓ 1 callersFunctionget_tb_info
(text)
methods/ReFoRCE/utils.py:262
↓ 1 callersMethodget_test_json
(self, mini_set=False)
spider2-lite/baselines/dailsql/utils/data_builder.py:110
↓ 1 callersMethodget_test_schema_linking
(self, mini_set=False)
spider2-lite/baselines/dailsql/utils/data_builder.py:137
↓ 1 callersFunctionget_tokenizer
(tokenizer_type: str)
spider2-lite/baselines/dailsql/utils/utils.py:314
↓ 1 callersMethodget_train_questions
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:65
↓ 1 callersMethodget_train_schema_linking
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:150
↓ 1 callersFunctionhard_prompt_maker
(row, schema_links, sub_questions, spider_schema, spider_foreign, args)
spider2-lite/baselines/dinsql/DIN-SQL.py:502
↓ 1 callersFunctioninit_chatgpt
(OPENAI_API_KEY, OPENAI_GROUP_ID, model)
spider2-lite/baselines/dailsql/llm/chatgpt.py:12
↓ 1 callersMethodinit_messages
(self)
methods/ReFoRCE/chat.py:51
↓ 1 callersFunctioninitialize_logger
(log_path, logger_name=None)
methods/ReFoRCE/utils.py:61
↓ 1 callersFunctionisFloat
(string)
spider2-lite/baselines/dailsql/utils/utils.py:545
↓ 1 callersFunctionisNegativeInt
(string)
spider2-lite/baselines/dailsql/utils/utils.py:538
↓ 1 callersFunctionis_common_db_term
(s: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:53
↓ 1 callersFunctionis_csv_empty
(path)
methods/ReFoRCE/utils.py:332
↓ 1 callersFunctionis_file
(filepath, suffix)
methods/ReFoRCE/utils.py:179
↓ 1 callersFunctionis_same_schema
(repre_js_pth, js_pth)
methods/ReFoRCE/utils.py:411
↓ 1 callersFunctionis_valid_result
(df_csv, do_column_exploration)
methods/ReFoRCE/utils.py:308
↓ 1 callersFunctionisnumber
(word)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:85
↓ 1 callersFunctionkeyset_in_list
(k1, k2, k_list)
spider2-lite/baselines/dailsql/utils/datasets/spider.py:28
↓ 1 callersMethodload
Load state from disk.
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:40
↓ 1 callersFunctionload_data
(DATASET)
spider2-lite/baselines/dinsql/DIN-SQL.py:499
↓ 1 callersFunctionload_jsonl_to_dict
(jsonl_file)
methods/ReFoRCE/eval.py:15
↓ 1 callersFunctionload_tables
(paths)
spider2-lite/baselines/dailsql/utils/datasets/spider.py:99
↓ 1 callersFunctionlower
(s)
spider2-lite/baselines/dailsql/utils/utils.py:338
↓ 1 callersFunctionmain
(args)
methods/SL/gen_sl/parse_json.py:44
↓ 1 callersFunctionmain
(args)
methods/SL/gen_sl/mk_prompt_spider2.py:111
↓ 1 callersFunctionmain
()
methods/SL/gen_sl/infer_api.py:10
↓ 1 callersFunctionmain
(args)
methods/SL/gen_sl/parse_logs.py:8
↓ 1 callersFunctionmain
(args)
methods/ReFoRCE/run.py:79
↓ 1 callersFunctionmake_folder
(args)
methods/ReFoRCE/reconstruct_data.py:104
↓ 1 callersFunctionmatch_shift
(q_col_match, q_tab_match, cell_match)
spider2-lite/baselines/dailsql/utils/linking_utils/spider_match_utils.py:165
↓ 1 callersFunctionmedium_prompt_maker
(row, schema_links, spider_schema, spider_foreign, args)
spider2-lite/baselines/dinsql/DIN-SQL.py:541
↓ 1 callersFunctionmerge_pred_results
(sample, pred_results)
spider2-lite/baselines/codes/schema_item_filter.py:147
↓ 1 callersFunctionmerge_predictions
(parsed, output_path)
methods/SL/gen_sl/parse_json.py:9
↓ 1 callersFunctionmultiset_eq
(l1: List, l2: List)
spider2-lite/baselines/dailsql/utils/post_process.py:55
↓ 1 callersFunctionparse_db
Parse the sql file and extract primary and foreign keys :param path_file: :return:
spider2-lite/baselines/dailsql/utils/utils.py:92
↓ 1 callersFunctionparse_option
()
spider2-lite/baselines/codes/evaluate_ex.py:6
↓ 1 callersFunctionparse_option
()
spider2-lite/baselines/codes/text2sql_few_shot.py:16
↓ 1 callersFunctionparse_response
(response)
methods/SL/gen_sl/utils.py:9
↓ 1 callersFunctionpermute_tuple
(element: Tuple, perm: Tuple)
spider2-lite/baselines/dailsql/utils/post_process.py:32
↓ 1 callersFunctionplot_statistics
(new_data)
spider2-lite/baselines/dailsql/preprocessed_data/spider2_preprocess.py:24
↓ 1 callersFunctionpost_process
(sql, schema_items, instance_id)
spider2-lite/baselines/codes/text2sql_zero_shot.py:15
↓ 1 callersFunctionpost_process
(sql, schema_items)
spider2-lite/baselines/codes/text2sql_few_shot.py:34
↓ 1 callersFunctionpostprocess
(query: str)
spider2-lite/baselines/dailsql/utils/post_process.py:202
↓ 1 callersFunctionpostprocess_sql_by_dialect
(sql_content, selected_tables_to_projDataset, file_name)
spider2-lite/baselines/utils/post_utils.py:20
↓ 1 callersMethodpredict_one
(self, sample)
spider2-lite/baselines/codes/schema_item_filter.py:288
↓ 1 callersFunctionprefix_match
(s1: str, s2: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:71
↓ 1 callersFunctionprepare_cross_domain_input_seq
(opt, eval_data, demonstration_set, similarity)
spider2-lite/baselines/codes/text2sql_few_shot.py:96
↓ 1 callersFunctionprepare_input_ids_and_attention_mask
(tokenizer, input_seq, max_input_length, device)
spider2-lite/baselines/codes/text2sql_few_shot.py:75
↓ 1 callersFunctionprepare_inputs
(prefix_seq, tokenizer, max_prefix_length)
spider2-lite/baselines/codes/utils/load_sft_dataset.py:86
↓ 1 callersFunctionprepare_inputs_and_labels
(sample, tokenizer)
spider2-lite/baselines/codes/schema_item_filter.py:10
↓ 1 callersFunctionprepare_inputs_and_labels
(prefix_seq, target_seq, tokenizer, max_tokens)
spider2-lite/baselines/codes/utils/load_sft_dataset.py:57
↓ 1 callersFunctionprepare_text2sql_prefix_sequence
(data, args)
spider2-lite/baselines/codes/utils/load_sft_dataset.py:13
↓ 1 callersFunctionpreprocess_dev_json
(args)
spider2-lite/baselines/codes/preprocessed_data/spider2_preprocess.py:109
↓ 1 callersMethodpreprocess_item
(self, item, schema, validation_info)
spider2-lite/baselines/dailsql/utils/linking_process.py:155
↓ 1 callersFunctionpreprocess_schema_uncached
If it's bert, we also cache the normalized version of question/column/table for schema linking
spider2-lite/baselines/dailsql/utils/linking_process.py:33
↓ 1 callersFunctionpreprocess_table_json
(args)
spider2-lite/baselines/codes/preprocessed_data/spider2_preprocess.py:17
↓ 1 callersFunctionprocess_ddl
(ddl_file, db_name_path)
methods/ReFoRCE/reconstruct_data.py:12
↓ 1 callersFunctionprocess_ddl_gold
(ddl_file, gold_table_names, entry=None)
methods/ReFoRCE/reconstruct_data.py:42
↓ 1 callersFunctionprocess_ddl_gold_schema
(ddl_file, full_table_names_with_omit, entry)
methods/ReFoRCE/reconstruct_data.py:69
↓ 1 callersFunctionprompt_factory
(repr_type:str, k_shot: int, example_format: str, selector_type: str)
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:96
↓ 1 callersFunctionquery_data
(sql_query, is_save)
spider2-lite/resource/interface/bigquery.py:5
↓ 1 callersFunctionquery_data
(sql_query, is_save)
spider2-lite/resource/interface/snowflake.py:8
↓ 1 callersFunctionquery_database_pandas
(db_path, query)
spider2-lite/resource/interface/sqlite.py:4
↓ 1 callersFunctionquick_rej
(result1: List[Tuple], result2: List[Tuple], order_matters: bool)
spider2-lite/baselines/dailsql/utils/post_process.py:45
↓ 1 callersFunctionread_jsonl
(file_path)
spider2-lite/baselines/utils/utils.py:13
↓ 1 callersFunctionread_jsonl
(file_path)
spider2-lite/baselines/dailsql/preprocessed_data/spider2_preprocess.py:20
↓ 1 callersMethodrecord_example_quality
(self, examples, target)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:14
← previousnext →201–300 of 554, ranked by callers