mirror of
https://github.com/microsoft/graphrag.git
synced 2026-02-21 10:16:29 +08:00
2de302ff0d
* Setup basic verb test runner * Replace join_text_units_to_entity_ids with subflow * Update comments * Replace join_text_units_to_relationship_ids subflow * Roll in final select * Reuse assertion util * Small fix + format * Format/typing * Semver * Format/typing * Semver * Revert format changes * Fix smoke test subworkflow count * Edit subworkflows for another smoke test
182 lines
4.5 KiB
JSON
182 lines
4.5 KiB
JSON
{
|
|
"input_path": "./tests/fixtures/text",
|
|
"input_file_type": "text",
|
|
"workflow_config": {
|
|
"create_base_text_units": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 11,
|
|
"max_runtime": 10
|
|
},
|
|
"create_base_extracted_entities": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 2,
|
|
"max_runtime": 300
|
|
},
|
|
"create_final_covariates": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"nan_allowed_columns": [
|
|
"type",
|
|
"description",
|
|
"subject_type",
|
|
"object_id",
|
|
"object_type",
|
|
"status",
|
|
"start_date",
|
|
"end_date",
|
|
"source_text"
|
|
],
|
|
"subworkflows": 6,
|
|
"max_runtime": 300
|
|
},
|
|
"create_summarized_entities": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 1,
|
|
"max_runtime": 300
|
|
},
|
|
"join_text_units_to_covariate_ids": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 2,
|
|
"max_runtime": 10
|
|
},
|
|
"create_base_entity_graph": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 2,
|
|
"max_runtime": 10
|
|
},
|
|
"create_final_entities": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"nan_allowed_columns": [
|
|
"type",
|
|
"description",
|
|
"graph_embedding"
|
|
],
|
|
"subworkflows": 11,
|
|
"max_runtime": 300
|
|
},
|
|
"create_final_relationships": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 8,
|
|
"max_runtime": 100
|
|
},
|
|
"create_final_nodes": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"nan_allowed_columns": [
|
|
"entity_type",
|
|
"description",
|
|
"graph_embedding",
|
|
"community",
|
|
"level"
|
|
],
|
|
"subworkflows": 10,
|
|
"max_runtime": 10
|
|
},
|
|
"create_final_communities": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 14,
|
|
"max_runtime": 10
|
|
},
|
|
"create_final_community_reports": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"nan_allowed_columns": [
|
|
"community_id",
|
|
"title",
|
|
"summary",
|
|
"full_content",
|
|
"full_content_json",
|
|
"rank",
|
|
"rank_explanation",
|
|
"findings"
|
|
],
|
|
"subworkflows": 7,
|
|
"max_runtime": 300
|
|
},
|
|
"join_text_units_to_entity_ids": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 1,
|
|
"max_runtime": 10
|
|
},
|
|
"join_text_units_to_relationship_ids": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 1,
|
|
"max_runtime": 10
|
|
},
|
|
"create_final_text_units": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"nan_allowed_columns": [
|
|
"relationship_ids",
|
|
"entity_ids"
|
|
],
|
|
"subworkflows": 7,
|
|
"max_runtime": 100
|
|
},
|
|
"create_base_documents": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 8,
|
|
"max_runtime": 10
|
|
},
|
|
"create_final_documents": {
|
|
"row_range": [
|
|
1,
|
|
2000
|
|
],
|
|
"subworkflows": 1,
|
|
"max_runtime": 100
|
|
}
|
|
},
|
|
"query_config": [
|
|
{
|
|
"query": "Who is Agent Alex Mercer and what are his goals?",
|
|
"method": "local"
|
|
},
|
|
{
|
|
"query": "What is the major conflict in this story and who are the protagonist and antagonist?",
|
|
"method": "global"
|
|
}
|
|
],
|
|
"slow": false
|
|
} |