├── .gitignore ├── LICENSE ├── README.md ├── data ├── datasets │ ├── CrossNER │ │ ├── ai │ │ │ ├── class.json │ │ │ ├── test.json │ │ │ └── train.json │ │ ├── literature │ │ │ ├── class.json │ │ │ ├── test.json │ │ │ └── train.json │ │ ├── music │ │ │ ├── class.json │ │ │ ├── test.json │ │ │ └── train.json │ │ ├── politics │ │ │ ├── class.json │ │ │ ├── test.json │ │ │ └── train.json │ │ ├── process.py │ │ └── science │ │ │ ├── class.json │ │ │ ├── test.json │ │ │ └── train.json │ └── NYT11 │ │ ├── class.json │ │ ├── test.json │ │ └── train.json └── input_files │ ├── Artificial_Intelligence_Wikipedia.txt │ ├── ChineseNewsExample.json │ ├── Harry_Potter_Chapter1.pdf │ └── Tulsi_Gabbard_News.html ├── examples ├── config │ ├── BookExtraction.yaml │ ├── EE.yaml │ ├── NER.yaml │ ├── NewsExtraction.yaml │ ├── RE.yaml │ └── Triple2KG.yaml ├── example.py └── results │ ├── BookExtraction.json │ ├── EE.json │ ├── NER.json │ ├── NewsExtraction.json │ ├── RE.json │ └── TripleExtraction.json ├── experiments ├── dataset_def.py ├── run_ner.py └── run_re.py ├── figs ├── logo.png ├── main.png └── webdemo │ ├── demo000.png │ ├── demo001.png │ ├── demo010.png │ ├── demo011.png │ ├── demo020.png │ ├── demo021.png │ ├── demo030.png │ ├── demo031.png │ ├── demo040.png │ ├── demo041.png │ ├── demo050.png │ ├── demo051.png │ └── demo_kg.png ├── frontend ├── app.py ├── app_addi.py ├── components │ ├── proxy_config.py │ ├── results.py │ └── sidebar.py ├── config │ └── settings.py ├── start.sh └── tools │ └── examples.py ├── requirements.txt └── src ├── config.yaml ├── construct ├── __init__.py └── convert.py ├── models ├── __init__.py ├── llm_def.py ├── prompt_example.py ├── prompt_template.py └── vllm_serve.py ├── modules ├── __init__.py ├── extraction_agent.py ├── knowledge_base │ ├── case_repository.json │ ├── case_repository.py │ └── schema_repository.py ├── reflection_agent.py └── schema_agent.py ├── pipeline.py ├── run.py └── utils ├── __init__.py ├── data_def.py └── process.py /.gitignore: -------------------------------------------------------------------------------- 1 | **/__pycache__ 2 | *.pyc 3 | dev -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/README.md -------------------------------------------------------------------------------- /data/datasets/CrossNER/ai/class.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/ai/class.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/ai/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/ai/test.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/ai/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/ai/train.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/literature/class.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/literature/class.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/literature/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/literature/test.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/literature/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/literature/train.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/music/class.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/music/class.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/music/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/music/test.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/music/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/music/train.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/politics/class.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/politics/class.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/politics/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/politics/test.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/politics/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/politics/train.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/process.py -------------------------------------------------------------------------------- /data/datasets/CrossNER/science/class.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/science/class.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/science/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/science/test.json -------------------------------------------------------------------------------- /data/datasets/CrossNER/science/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/CrossNER/science/train.json -------------------------------------------------------------------------------- /data/datasets/NYT11/class.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/NYT11/class.json -------------------------------------------------------------------------------- /data/datasets/NYT11/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/NYT11/test.json -------------------------------------------------------------------------------- /data/datasets/NYT11/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/datasets/NYT11/train.json -------------------------------------------------------------------------------- /data/input_files/Artificial_Intelligence_Wikipedia.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/input_files/Artificial_Intelligence_Wikipedia.txt -------------------------------------------------------------------------------- /data/input_files/ChineseNewsExample.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/input_files/ChineseNewsExample.json -------------------------------------------------------------------------------- /data/input_files/Harry_Potter_Chapter1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/input_files/Harry_Potter_Chapter1.pdf -------------------------------------------------------------------------------- /data/input_files/Tulsi_Gabbard_News.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/data/input_files/Tulsi_Gabbard_News.html -------------------------------------------------------------------------------- /examples/config/BookExtraction.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/config/BookExtraction.yaml -------------------------------------------------------------------------------- /examples/config/EE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/config/EE.yaml -------------------------------------------------------------------------------- /examples/config/NER.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/config/NER.yaml -------------------------------------------------------------------------------- /examples/config/NewsExtraction.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/config/NewsExtraction.yaml -------------------------------------------------------------------------------- /examples/config/RE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/config/RE.yaml -------------------------------------------------------------------------------- /examples/config/Triple2KG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/config/Triple2KG.yaml -------------------------------------------------------------------------------- /examples/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/example.py -------------------------------------------------------------------------------- /examples/results/BookExtraction.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/results/BookExtraction.json -------------------------------------------------------------------------------- /examples/results/EE.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/results/EE.json -------------------------------------------------------------------------------- /examples/results/NER.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/results/NER.json -------------------------------------------------------------------------------- /examples/results/NewsExtraction.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/results/NewsExtraction.json -------------------------------------------------------------------------------- /examples/results/RE.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/results/RE.json -------------------------------------------------------------------------------- /examples/results/TripleExtraction.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/examples/results/TripleExtraction.json -------------------------------------------------------------------------------- /experiments/dataset_def.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/experiments/dataset_def.py -------------------------------------------------------------------------------- /experiments/run_ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/experiments/run_ner.py -------------------------------------------------------------------------------- /experiments/run_re.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/experiments/run_re.py -------------------------------------------------------------------------------- /figs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/logo.png -------------------------------------------------------------------------------- /figs/main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/main.png -------------------------------------------------------------------------------- /figs/webdemo/demo000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo000.png -------------------------------------------------------------------------------- /figs/webdemo/demo001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo001.png -------------------------------------------------------------------------------- /figs/webdemo/demo010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo010.png -------------------------------------------------------------------------------- /figs/webdemo/demo011.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo011.png -------------------------------------------------------------------------------- /figs/webdemo/demo020.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo020.png -------------------------------------------------------------------------------- /figs/webdemo/demo021.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo021.png -------------------------------------------------------------------------------- /figs/webdemo/demo030.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo030.png -------------------------------------------------------------------------------- /figs/webdemo/demo031.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo031.png -------------------------------------------------------------------------------- /figs/webdemo/demo040.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo040.png -------------------------------------------------------------------------------- /figs/webdemo/demo041.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo041.png -------------------------------------------------------------------------------- /figs/webdemo/demo050.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo050.png -------------------------------------------------------------------------------- /figs/webdemo/demo051.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo051.png -------------------------------------------------------------------------------- /figs/webdemo/demo_kg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/figs/webdemo/demo_kg.png -------------------------------------------------------------------------------- /frontend/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/app.py -------------------------------------------------------------------------------- /frontend/app_addi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/app_addi.py -------------------------------------------------------------------------------- /frontend/components/proxy_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/components/proxy_config.py -------------------------------------------------------------------------------- /frontend/components/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/components/results.py -------------------------------------------------------------------------------- /frontend/components/sidebar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/components/sidebar.py -------------------------------------------------------------------------------- /frontend/config/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/config/settings.py -------------------------------------------------------------------------------- /frontend/start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/start.sh -------------------------------------------------------------------------------- /frontend/tools/examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/frontend/tools/examples.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/config.yaml -------------------------------------------------------------------------------- /src/construct/__init__.py: -------------------------------------------------------------------------------- 1 | from .convert import * 2 | -------------------------------------------------------------------------------- /src/construct/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/construct/convert.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/models/__init__.py -------------------------------------------------------------------------------- /src/models/llm_def.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/models/llm_def.py -------------------------------------------------------------------------------- /src/models/prompt_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/models/prompt_example.py -------------------------------------------------------------------------------- /src/models/prompt_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/models/prompt_template.py -------------------------------------------------------------------------------- /src/models/vllm_serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/models/vllm_serve.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/__init__.py -------------------------------------------------------------------------------- /src/modules/extraction_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/extraction_agent.py -------------------------------------------------------------------------------- /src/modules/knowledge_base/case_repository.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/knowledge_base/case_repository.json -------------------------------------------------------------------------------- /src/modules/knowledge_base/case_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/knowledge_base/case_repository.py -------------------------------------------------------------------------------- /src/modules/knowledge_base/schema_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/knowledge_base/schema_repository.py -------------------------------------------------------------------------------- /src/modules/reflection_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/reflection_agent.py -------------------------------------------------------------------------------- /src/modules/schema_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/modules/schema_agent.py -------------------------------------------------------------------------------- /src/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/pipeline.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/run.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/data_def.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/utils/data_def.py -------------------------------------------------------------------------------- /src/utils/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/OneKE/HEAD/src/utils/process.py --------------------------------------------------------------------------------