SQL-Exchange/mapping_settings.json at main · mmdrez4/SQL-Exchange · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
{
    "model": {
        "model_origin": "google",
        "model_name": "gemini-1.5-flash",
        "input_token_limit": null,
        "use_system_instruction": true,
        "temperature": 0,
        "top_p": 1,
        "google": {
            "top_k": 0
        }
    },

    "generation": {
        "copy_settings_to_output": true,
        "max_question_length_per_prompt": 10,
        "max_retry_per_prompt": 3,
        "max_fail_limit": 100,
        "validation": {
            "fields_checking": true,
            "db_id_matching": true
        },
        "method": "sql-exchange",
        "prompt_directory": "prompts",
        "base_prompt_file": "mapping_base.txt",
        "system_instruction_file": "mapping_system.txt",
        "system_instruction_file_zeroshot": "zeroshot_system.txt",
        "prompt_file": "mapping_prompt.txt",
        "output_directory": "mappings_full_analysis",
        "json_only_output_directory": "mappings",
        "json_only_output_directory_zeroshot": "mappings_zeroshot",
        "fields_to_check": [
            "source_dataset",
            "source_db_id",
            "source_query",
            "source_question",
            "tables_columns_replacement",
            "thought",
            "target_db_id",
            "target_query",
            "target_question"
        ],
        "fields_to_check_zeroshot": [
            "source_dataset",
            "source_db_id",
            "source_query",
            "source_question",
            "target_db_id",
            "target_query",
            "target_question"
        ]
    },


    "data": [
        {
            "source_dataset": "data/bird_training",
            "source_db_ids": [
                "address",
                "books"
            ],
            "source_questions_shuffle_seed": 12,
            "source_questions_limit": 20,
            "target_dataset": "data/bird_dev",
            "target_db_id": "california_schools"
        },
        {
            "source_dataset": "data/bird_training",
            "source_db_ids": [
                "beer_factory"
            ],
            "source_questions_shuffle_seed": 12,
            "source_questions_limit": 20,
            "target_dataset": "data/bird_dev",
            "target_db_id": "student_club"
        }
    ]
}