| { |
| "api_key": { |
| "qwen": "your_qwen_api_key_here", |
| "deepseek": "your_deepseek_api_key_here", |
| "zhipu": "your_zhipu_api_key_here" |
| }, |
| "llm": { |
| "use": "qwen", |
| "model_name": "qwen-turbo", |
| "context_window": 120000, |
| "max_token": 8000, |
| "top_p": 0.9, |
| "temperature": 0.75, |
| "time_out": 300.0 |
| }, |
| "text_embed": { |
| "embed_model_source": "HuggingFace", |
| "embed_model_name": "BAAI/bge-large-en-v1.5" |
| }, |
| "router": { |
| "use_demo": true |
| }, |
| "dataloader": { |
| "is_prepare_data": true |
| }, |
| "dataset": { |
| "data_source": "spider:dev", |
| "data_source_dir": "../files/data_source", |
| "overwrite_exist_file": true, |
| "need_few_shot": false, |
| "few_shot_num": 3, |
| "few_shot_save_dir": "../files/reasoning_examples/user", |
| "few_shot_range": [], |
| "need_external": false, |
| "external_range": [], |
| "external_save_dir": "../files/external", |
| "db_path": null |
| }, |
| "database": { |
| "skip_schema_init": false, |
| "schema_source": "spider:dev", |
| "multi_database": false, |
| "vector_store": "../vector_store", |
| "schema_source_dir": "../files/schema_source", |
| "need_build_index": false, |
| "index_range": false |
| }, |
| "reducer": { |
| "reduce_type": "LinkAlign", |
| "is_save_reduce": true, |
| "reduce_save_dir": "../files/instance_schemas", |
| "reduce_output_format": "dataframe" |
| }, |
| "parser": { |
| "parse_type": "LinkAlign", |
| "is_save_parse": true, |
| "parse_save_dir": "../files/schema_links", |
| "parse_output_format": "dataframe" |
| }, |
| "generator": { |
| "generate_type": "LinkAlign", |
| "is_save_generate": true, |
| "generate_save_dir": "../files/pred_sql" |
| }, |
| "task": { |
| "task_meta": [ |
| { |
| "task_id": "demo_task", |
| "task_type": "generate", |
| "data_source": "spider:dev", |
| "schema_source": "spider:dev", |
| "meta": { |
| "dataset": { |
| "random_size": null, |
| "filter_by": null |
| } |
| } |
| } |
| ], |
| "cpx_task_meta": [], |
| "is_save_dataset": true, |
| "open_parallel": true, |
| "max_workers": 5 |
| }, |
| "engine": { |
| "exec_process": ["demo_task"] |
| }, |
| "credential": { |
| "big_query": "path/to/big_query_credential.json", |
| "snowflake": "path/to/snowflake_credential.json" |
| } |
| } |
