|
124 | 124 | " \"PLACE\": [\"HAS\", \"PART_OF\", \"WORKED_AT\"],\n",
|
125 | 125 | " \"ORGANIZATION\": [\"HAS\", \"PART_OF\", \"WORKED_WITH\"],\n",
|
126 | 126 | "}\n",
|
127 |
| - "validation_schema = [\n", |
128 |
| - " (\"ORGANIZATION\", \"HAS\", \"PERSON\"),\n", |
129 |
| - " (\"PERSON\", \"WORKED_AT\", \"ORGANIZATION\"),\n", |
130 |
| - " (\"PERSON\", \"WORKED_WITH\", \"PERSON\"),\n", |
131 |
| - " (\"PERSON\", \"WORKED_ON\", \"ORGANIZATION\"),\n", |
132 |
| - " (\"PERSON\", \"PART_OF\", \"ORGANIZATION\"),\n", |
133 |
| - " (\"ORGANIZATION\", \"PART_OF\", \"ORGANIZATION\"),\n", |
134 |
| - " (\"PERSON\", \"WORKED_AT\", \"PLACE\"),\n", |
135 |
| - "]\n", |
136 | 127 | "\n",
|
137 | 128 | "kg_extractor = SchemaLLMPathExtractor(\n",
|
138 | 129 | " llm=Ollama(model=\"llama3\", json_mode=True, request_timeout=3600),\n",
|
|
0 commit comments