mirror of
https://github.com/microsoft/graphrag.git
synced 2026-02-12 14:04:30 +08:00
20a96cb35e
Some checks failed
Python Build and Type Check / python-ci (ubuntu-latest, 3.11) (push) Has been cancelled
Python Build and Type Check / python-ci (ubuntu-latest, 3.12) (push) Has been cancelled
Python Build and Type Check / python-ci (windows-latest, 3.11) (push) Has been cancelled
Python Build and Type Check / python-ci (windows-latest, 3.12) (push) Has been cancelled
Python Integration Tests / python-ci (ubuntu-latest, 3.12) (push) Has been cancelled
Python Integration Tests / python-ci (windows-latest, 3.12) (push) Has been cancelled
Python Notebook Tests / python-ci (ubuntu-latest, 3.12) (push) Has been cancelled
Python Notebook Tests / python-ci (windows-latest, 3.12) (push) Has been cancelled
Python Smoke Tests / python-ci (ubuntu-latest, 3.12) (push) Has been cancelled
Python Smoke Tests / python-ci (windows-latest, 3.12) (push) Has been cancelled
Python Unit Tests / python-ci (ubuntu-latest, 3.12) (push) Has been cancelled
Python Unit Tests / python-ci (windows-latest, 3.12) (push) Has been cancelled
* Add init prompting for models * Remove hard-coded model config validation * Switch to typer option prompt for full CLI use with models * Update getting started for init model input * Bump request timeout and overall smoke test timeout
109 lines
2.9 KiB
JSON
109 lines
2.9 KiB
JSON
{
|
|
"input_path": "./tests/fixtures/text",
|
|
"input_file_type": "text",
|
|
"index_method": "fast",
|
|
"workflow_config": {
|
|
"load_input_documents": {
|
|
"max_runtime": 30
|
|
},
|
|
"create_base_text_units": {
|
|
"max_runtime": 30
|
|
},
|
|
"extract_graph_nlp": {
|
|
"max_runtime": 30
|
|
},
|
|
"prune_graph": {
|
|
"max_runtime": 30
|
|
},
|
|
"finalize_graph": {
|
|
"row_range": [
|
|
10,
|
|
300
|
|
],
|
|
"max_runtime": 30,
|
|
"expected_artifacts": [
|
|
"entities.parquet",
|
|
"relationships.parquet"
|
|
]
|
|
},
|
|
"create_communities": {
|
|
"row_range": [
|
|
1,
|
|
30
|
|
],
|
|
"max_runtime": 30,
|
|
"expected_artifacts": ["communities.parquet"]
|
|
},
|
|
"create_community_reports_text": {
|
|
"row_range": [
|
|
1,
|
|
30
|
|
],
|
|
"nan_allowed_columns": [
|
|
"title",
|
|
"summary",
|
|
"full_content",
|
|
"full_content_json",
|
|
"rank",
|
|
"rank_explanation",
|
|
"findings",
|
|
"period",
|
|
"size"
|
|
],
|
|
"max_runtime": 2000,
|
|
"expected_artifacts": ["community_reports.parquet"]
|
|
},
|
|
"create_final_text_units": {
|
|
"row_range": [
|
|
1,
|
|
10
|
|
],
|
|
"nan_allowed_columns": [
|
|
"relationship_ids",
|
|
"entity_ids",
|
|
"covariate_ids"
|
|
],
|
|
"max_runtime": 30,
|
|
"expected_artifacts": ["text_units.parquet"]
|
|
},
|
|
"create_final_documents": {
|
|
"row_range": [
|
|
1,
|
|
1
|
|
],
|
|
"nan_allowed_columns": [
|
|
"metadata"
|
|
],
|
|
"max_runtime": 30,
|
|
"expected_artifacts": ["documents.parquet"]
|
|
},
|
|
"generate_text_embeddings": {
|
|
"row_range": [
|
|
1,
|
|
100
|
|
],
|
|
"max_runtime": 150,
|
|
"expected_artifacts": [
|
|
"embeddings.text_unit_text.parquet",
|
|
"embeddings.entity_description.parquet",
|
|
"embeddings.community_full_content.parquet"
|
|
]
|
|
}
|
|
},
|
|
"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"
|
|
},
|
|
{
|
|
"query": "Who is Jordan Hayes?",
|
|
"method": "basic"
|
|
}
|
|
],
|
|
"slow": false
|
|
}
|