├── ag_src ├── __init__.py ├── agent_utils │ ├── __init__.py │ ├── ag_utils.py │ ├── config.py │ ├── prompt_list.py │ ├── run_exp.py │ └── simlarity_search.py ├── grail_src │ ├── GRAIL.py │ ├── sparql_generator.py │ └── wikienv.py ├── graphq_src │ ├── GRAPHQ.py │ ├── sparql_generator.py │ └── wikienv_graphq.py ├── meta_src │ ├── META.py │ └── wikienv_meta.py ├── webqsp_src │ ├── WebQSP.py │ ├── sparql_generator.py │ └── wikienv_webqsp.py └── wikisql_src │ ├── WIKISQL.py │ ├── sql_generator_for_wikisql.py │ └── wikienv_wikisql.py ├── data ├── GrailQA_v1.0 │ ├── grail_pangu_tiara.json │ └── grailqa_dev_pyql.json ├── GraphQ │ ├── graphq_test.json │ └── graphquestions_v1_fb15_test_091420.json ├── WebQSP │ ├── WebQSP_test_processed.json │ └── webqsp_test_el.json ├── fb_relation.txt ├── metaQA │ ├── kb.txt │ ├── kb_entity_dict.txt │ └── qa_test_3hop.txt └── wikisql │ ├── test.db │ └── wikisql_test.json ├── environment.yml ├── readme.md └── requirements.txt /ag_src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/__init__.py -------------------------------------------------------------------------------- /ag_src/agent_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/agent_utils/__init__.py -------------------------------------------------------------------------------- /ag_src/agent_utils/ag_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/agent_utils/ag_utils.py -------------------------------------------------------------------------------- /ag_src/agent_utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/agent_utils/config.py -------------------------------------------------------------------------------- /ag_src/agent_utils/prompt_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/agent_utils/prompt_list.py -------------------------------------------------------------------------------- /ag_src/agent_utils/run_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/agent_utils/run_exp.py -------------------------------------------------------------------------------- /ag_src/agent_utils/simlarity_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/agent_utils/simlarity_search.py -------------------------------------------------------------------------------- /ag_src/grail_src/GRAIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/grail_src/GRAIL.py -------------------------------------------------------------------------------- /ag_src/grail_src/sparql_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/grail_src/sparql_generator.py -------------------------------------------------------------------------------- /ag_src/grail_src/wikienv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/grail_src/wikienv.py -------------------------------------------------------------------------------- /ag_src/graphq_src/GRAPHQ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/graphq_src/GRAPHQ.py -------------------------------------------------------------------------------- /ag_src/graphq_src/sparql_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/graphq_src/sparql_generator.py -------------------------------------------------------------------------------- /ag_src/graphq_src/wikienv_graphq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/graphq_src/wikienv_graphq.py -------------------------------------------------------------------------------- /ag_src/meta_src/META.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/meta_src/META.py -------------------------------------------------------------------------------- /ag_src/meta_src/wikienv_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/meta_src/wikienv_meta.py -------------------------------------------------------------------------------- /ag_src/webqsp_src/WebQSP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/webqsp_src/WebQSP.py -------------------------------------------------------------------------------- /ag_src/webqsp_src/sparql_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/webqsp_src/sparql_generator.py -------------------------------------------------------------------------------- /ag_src/webqsp_src/wikienv_webqsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/webqsp_src/wikienv_webqsp.py -------------------------------------------------------------------------------- /ag_src/wikisql_src/WIKISQL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/wikisql_src/WIKISQL.py -------------------------------------------------------------------------------- /ag_src/wikisql_src/sql_generator_for_wikisql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/wikisql_src/sql_generator_for_wikisql.py -------------------------------------------------------------------------------- /ag_src/wikisql_src/wikienv_wikisql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/ag_src/wikisql_src/wikienv_wikisql.py -------------------------------------------------------------------------------- /data/GrailQA_v1.0/grail_pangu_tiara.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/GrailQA_v1.0/grail_pangu_tiara.json -------------------------------------------------------------------------------- /data/GrailQA_v1.0/grailqa_dev_pyql.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/GrailQA_v1.0/grailqa_dev_pyql.json -------------------------------------------------------------------------------- /data/GraphQ/graphq_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/GraphQ/graphq_test.json -------------------------------------------------------------------------------- /data/GraphQ/graphquestions_v1_fb15_test_091420.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/GraphQ/graphquestions_v1_fb15_test_091420.json -------------------------------------------------------------------------------- /data/WebQSP/WebQSP_test_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/WebQSP/WebQSP_test_processed.json -------------------------------------------------------------------------------- /data/WebQSP/webqsp_test_el.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/WebQSP/webqsp_test_el.json -------------------------------------------------------------------------------- /data/fb_relation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/fb_relation.txt -------------------------------------------------------------------------------- /data/metaQA/kb.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/metaQA/kb.txt -------------------------------------------------------------------------------- /data/metaQA/kb_entity_dict.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/metaQA/kb_entity_dict.txt -------------------------------------------------------------------------------- /data/metaQA/qa_test_3hop.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/metaQA/qa_test_3hop.txt -------------------------------------------------------------------------------- /data/wikisql/test.db: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/wikisql/test.db -------------------------------------------------------------------------------- /data/wikisql/wikisql_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/data/wikisql/wikisql_test.json -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/environment.yml -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cdhx/QueryAgent/HEAD/requirements.txt --------------------------------------------------------------------------------