Default Configuration Mode (using Env Vars)
Text-Embeddings Customization
By default, the GraphRAG indexer will only export embeddings required for our query methods. However, the model has embeddings defined for all plaintext fields, and these can be generated by setting the GRAPHRAG_EMBEDDING_TARGET
environment variable to all
.
If the embedding target is all
, and you want to only embed a subset of these fields, you may specify which embeddings to skip using the GRAPHRAG_EMBEDDING_SKIP
argument described below.
Embedded Fields
text_unit.text
document.text
entity.title
entity.description
relationship.description
community.title
community.summary
community.full_content
Our pipeline can ingest .csv or .txt data from an input folder. These files can be nested within subfolders. To configure how input data is handled, what fields are mapped over, and how timestamps are parsed, look for configuration values starting with GRAPHRAG_INPUT_
below. In general, CSV-based data provides the most customizability. Each CSV should at least contain a text
field (which can be mapped with environment variables), but it's helpful if they also have title
, timestamp
, and source
fields. Additional fields can be included as well, which will land as extra fields on the Document
table.
Base LLM Settings
These are the primary settings for configuring LLM connectivity.
Parameter |
Required? |
Description |
Type |
Default Value |
GRAPHRAG_API_KEY |
Yes for OpenAI. Optional for AOAI |
The API key. (Note: OPENAI_API_KEY is also used as a fallback). If not defined when using AOAI, managed identity will be used. | str| None` |
|
|
GRAPHRAG_API_BASE |
For AOAI |
The API Base URL |
str |
None |
GRAPHRAG_API_VERSION |
For AOAI |
The AOAI API version. |
str |
None |
GRAPHRAG_API_ORGANIZATION |
|
The AOAI organization. |
str |
None |
GRAPHRAG_API_PROXY |
|
The AOAI proxy. |
str |
None |
Text Generation Settings
These settings control the text generation model used by the pipeline. Any settings with a fallback will use the base LLM settings, if available.
Parameter |
Required? |
Description |
Type |
Default Value |
GRAPHRAG_LLM_TYPE |
For AOAI |
The LLM operation type. Either openai_chat or azure_openai_chat |
str |
openai_chat |
GRAPHRAG_LLM_DEPLOYMENT_NAME |
For AOAI |
The AOAI model deployment name. |
str |
None |
GRAPHRAG_LLM_API_KEY |
Yes (uses fallback) |
The API key. If not defined when using AOAI, managed identity will be used. |
str |
None |
GRAPHRAG_LLM_API_BASE |
For AOAI (uses fallback) |
The API Base URL |
str |
None |
GRAPHRAG_LLM_API_VERSION |
For AOAI (uses fallback) |
The AOAI API version. |
str |
None |
GRAPHRAG_LLM_API_ORGANIZATION |
For AOAI (uses fallback) |
The AOAI organization. |
str |
None |
GRAPHRAG_LLM_API_PROXY |
|
The AOAI proxy. |
str |
None |
GRAPHRAG_LLM_MODEL |
|
The LLM model. |
str |
gpt-4-turbo-preview |
GRAPHRAG_LLM_MAX_TOKENS |
|
The maximum number of tokens. |
int |
4000 |
GRAPHRAG_LLM_REQUEST_TIMEOUT |
|
The maximum number of seconds to wait for a response from the chat client. |
int |
180 |
GRAPHRAG_LLM_MODEL_SUPPORTS_JSON |
|
Indicates whether the given model supports JSON output mode. True to enable. |
str |
None |
GRAPHRAG_LLM_THREAD_COUNT |
|
The number of threads to use for LLM parallelization. |
int |
50 |
GRAPHRAG_LLM_THREAD_STAGGER |
|
The time to wait (in seconds) between starting each thread. |
float |
0.3 |
GRAPHRAG_LLM_CONCURRENT_REQUESTS |
|
The number of concurrent requests to allow for the embedding client. |
int |
25 |
GRAPHRAG_LLM_TOKENS_PER_MINUTE |
|
The number of tokens per minute to allow for the LLM client. 0 = Bypass |
int |
0 |
GRAPHRAG_LLM_REQUESTS_PER_MINUTE |
|
The number of requests per minute to allow for the LLM client. 0 = Bypass |
int |
0 |
GRAPHRAG_LLM_MAX_RETRIES |
|
The maximum number of retries to attempt when a request fails. |
int |
10 |
GRAPHRAG_LLM_MAX_RETRY_WAIT |
|
The maximum number of seconds to wait between retries. |
int |
10 |
GRAPHRAG_LLM_SLEEP_ON_RATE_LIMIT_RECOMMENDATION |
|
Whether to sleep on rate limit recommendation. (Azure Only) |
bool |
True |
GRAPHRAG_LLM_TEMPERATURE |
|
The temperature to use generation. |
float |
0 |
GRAPHRAG_LLM_TOP_P |
|
The top_p to use for sampling. |
float |
1 |
GRAPHRAG_LLM_N |
|
The number of responses to generate. |
int |
1 |
Text Embedding Settings
These settings control the text embedding model used by the pipeline. Any settings with a fallback will use the base LLM settings, if available.
Parameter |
Required ? |
Description |
Type |
Default |
GRAPHRAG_EMBEDDING_TYPE |
For AOAI |
The embedding client to use. Either openai_embedding or azure_openai_embedding |
str |
openai_embedding |
GRAPHRAG_EMBEDDING_DEPLOYMENT_NAME |
For AOAI |
The AOAI deployment name. |
str |
None |
GRAPHRAG_EMBEDDING_API_KEY |
Yes (uses fallback) |
The API key to use for the embedding client. If not defined when using AOAI, managed identity will be used. |
str |
None |
GRAPHRAG_EMBEDDING_API_BASE |
For AOAI (uses fallback) |
The API base URL. |
str |
None |
GRAPHRAG_EMBEDDING_API_VERSION |
For AOAI (uses fallback) |
The AOAI API version to use for the embedding client. |
str |
None |
GRAPHRAG_EMBEDDING_API_ORGANIZATION |
For AOAI (uses fallback) |
The AOAI organization to use for the embedding client. |
str |
None |
GRAPHRAG_EMBEDDING_API_PROXY |
|
The AOAI proxy to use for the embedding client. |
str |
None |
GRAPHRAG_EMBEDDING_MODEL |
|
The model to use for the embedding client. |
str |
text-embedding-3-small |
GRAPHRAG_EMBEDDING_BATCH_SIZE |
|
The number of texts to embed at once. (Azure limit is 16) |
int |
16 |
GRAPHRAG_EMBEDDING_BATCH_MAX_TOKENS |
|
The maximum tokens per batch (Azure limit is 8191) |
int |
8191 |
GRAPHRAG_EMBEDDING_TARGET |
|
The target fields to embed. Either required or all . |
str |
required |
GRAPHRAG_EMBEDDING_SKIP |
|
A comma-separated list of fields to skip embeddings for . (e.g. 'relationship.description') |
str |
None |
GRAPHRAG_EMBEDDING_THREAD_COUNT |
|
The number of threads to use for parallelization for embeddings. |
int |
|
GRAPHRAG_EMBEDDING_THREAD_STAGGER |
|
The time to wait (in seconds) between starting each thread for embeddings. |
float |
50 |
GRAPHRAG_EMBEDDING_CONCURRENT_REQUESTS |
|
The number of concurrent requests to allow for the embedding client. |
int |
25 |
GRAPHRAG_EMBEDDING_TOKENS_PER_MINUTE |
|
The number of tokens per minute to allow for the embedding client. 0 = Bypass |
int |
0 |
GRAPHRAG_EMBEDDING_REQUESTS_PER_MINUTE |
|
The number of requests per minute to allow for the embedding client. 0 = Bypass |
int |
0 |
GRAPHRAG_EMBEDDING_MAX_RETRIES |
|
The maximum number of retries to attempt when a request fails. |
int |
10 |
GRAPHRAG_EMBEDDING_MAX_RETRY_WAIT |
|
The maximum number of seconds to wait between retries. |
int |
10 |
GRAPHRAG_EMBEDDING_SLEEP_ON_RATE_LIMIT_RECOMMENDATION |
|
Whether to sleep on rate limit recommendation. (Azure Only) |
bool |
True |
These settings control the data input used by the pipeline. Any settings with a fallback will use the base LLM settings, if available.
Plaintext Input Data (GRAPHRAG_INPUT_FILE_TYPE
=text)
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_INPUT_FILE_PATTERN |
The file pattern regexp to use when reading input files from the input directory. |
str |
optional |
.*\.txt$ |
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_INPUT_TYPE |
The input storage type to use when reading files. (file or blob ) |
str |
optional |
file |
GRAPHRAG_INPUT_FILE_PATTERN |
The file pattern regexp to use when reading input files from the input directory. |
str |
optional |
.*\.txt$ |
GRAPHRAG_INPUT_SOURCE_COLUMN |
The 'source' column to use when reading CSV input files. |
str |
optional |
source |
GRAPHRAG_INPUT_TIMESTAMP_COLUMN |
The 'timestamp' column to use when reading CSV input files. |
str |
optional |
None |
GRAPHRAG_INPUT_TIMESTAMP_FORMAT |
The timestamp format to use when parsing timestamps in the timestamp column. |
str |
optional |
None |
GRAPHRAG_INPUT_TEXT_COLUMN |
The 'text' column to use when reading CSV input files. |
str |
optional |
text |
GRAPHRAG_INPUT_DOCUMENT_ATTRIBUTE_COLUMNS |
A list of CSV columns, comma-separated, to incorporate as document fields. |
str |
optional |
id |
GRAPHRAG_INPUT_TITLE_COLUMN |
The 'title' column to use when reading CSV input files. |
str |
optional |
title |
GRAPHRAG_INPUT_STORAGE_ACCOUNT_BLOB_URL |
The Azure Storage blob endpoint to use when in blob mode and using managed identity. Will have the format https://<storage_account_name>.blob.core.windows.net |
str |
optional |
None |
GRAPHRAG_INPUT_CONNECTION_STRING |
The connection string to use when reading CSV input files from Azure Blob Storage. |
str |
optional |
None |
GRAPHRAG_INPUT_CONTAINER_NAME |
The container name to use when reading CSV input files from Azure Blob Storage. |
str |
optional |
None |
GRAPHRAG_INPUT_BASE_DIR |
The base directory to read input files from. |
str |
optional |
None |
Data Mapping Settings
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_INPUT_FILE_TYPE |
The type of input data, csv or text |
str |
optional |
text |
GRAPHRAG_INPUT_ENCODING |
The encoding to apply when reading CSV/text input files. |
str |
optional |
utf-8 |
Data Chunking
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_CHUNK_SIZE |
The chunk size in tokens for text-chunk analysis windows. |
str |
optional |
1200 |
GRAPHRAG_CHUNK_OVERLAP |
The chunk overlap in tokens for text-chunk analysis windows. |
str |
optional |
100 |
GRAPHRAG_CHUNK_BY_COLUMNS |
A comma-separated list of document attributes to groupby when performing TextUnit chunking. |
str |
optional |
id |
GRAPHRAG_CHUNK_ENCODING_MODEL |
The encoding model to use for chunking. |
str |
optional |
The top-level encoding model. |
Prompting Overrides
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_ENTITY_EXTRACTION_PROMPT_FILE |
The path (relative to the root) of an entity extraction prompt template text file. |
str |
optional |
None |
GRAPHRAG_ENTITY_EXTRACTION_MAX_GLEANINGS |
The maximum number of redrives (gleanings) to invoke when extracting entities in a loop. |
int |
optional |
1 |
GRAPHRAG_ENTITY_EXTRACTION_ENTITY_TYPES |
A comma-separated list of entity types to extract. |
str |
optional |
organization,person,event,geo |
GRAPHRAG_ENTITY_EXTRACTION_ENCODING_MODEL |
The encoding model to use for entity extraction. |
str |
optional |
The top-level encoding model. |
GRAPHRAG_SUMMARIZE_DESCRIPTIONS_PROMPT_FILE |
The path (relative to the root) of an description summarization prompt template text file. |
str |
optional |
None |
GRAPHRAG_SUMMARIZE_DESCRIPTIONS_MAX_LENGTH |
The maximum number of tokens to generate per description summarization. |
int |
optional |
500 |
GRAPHRAG_CLAIM_EXTRACTION_ENABLED |
Whether claim extraction is enabled for this pipeline. |
bool |
optional |
False |
GRAPHRAG_CLAIM_EXTRACTION_DESCRIPTION |
The claim_description prompting argument to utilize. |
string |
optional |
"Any claims or facts that could be relevant to threat analysis." |
GRAPHRAG_CLAIM_EXTRACTION_PROMPT_FILE |
The claim extraction prompt to utilize. |
string |
optional |
None |
GRAPHRAG_CLAIM_EXTRACTION_MAX_GLEANINGS |
The maximum number of redrives (gleanings) to invoke when extracting claims in a loop. |
int |
optional |
1 |
GRAPHRAG_CLAIM_EXTRACTION_ENCODING_MODEL |
The encoding model to use for claim extraction. |
str |
optional |
The top-level encoding model |
GRAPHRAG_COMMUNITY_REPORTS_PROMPT_FILE |
The community reports extraction prompt to utilize. |
string |
optional |
None |
GRAPHRAG_COMMUNITY_REPORTS_MAX_LENGTH |
The maximum number of tokens to generate per community reports. |
int |
optional |
1500 |
Storage
This section controls the storage mechanism used by the pipeline used for exporting output tables.
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_STORAGE_TYPE |
The type of reporter to use. Options are file , memory , or blob |
str |
optional |
file |
GRAPHRAG_STORAGE_STORAGE_ACCOUNT_BLOB_URL |
The Azure Storage blob endpoint to use when in blob mode and using managed identity. Will have the format https://<storage_account_name>.blob.core.windows.net |
str |
optional |
None |
GRAPHRAG_STORAGE_CONNECTION_STRING |
The Azure Storage connection string to use when in blob mode. |
str |
optional |
None |
GRAPHRAG_STORAGE_CONTAINER_NAME |
The Azure Storage container name to use when in blob mode. |
str |
optional |
None |
GRAPHRAG_STORAGE_BASE_DIR |
The base path to data outputs outputs. |
str |
optional |
None |
Cache
This section controls the cache mechanism used by the pipeline. This is used to cache LLM invocation results.
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_CACHE_TYPE |
The type of cache to use. Options are file , memory , none or blob |
str |
optional |
file |
GRAPHRAG_CACHE_STORAGE_ACCOUNT_BLOB_URL |
The Azure Storage blob endpoint to use when in blob mode and using managed identity. Will have the format https://<storage_account_name>.blob.core.windows.net |
str |
optional |
None |
GRAPHRAG_CACHE_CONNECTION_STRING |
The Azure Storage connection string to use when in blob mode. |
str |
optional |
None |
GRAPHRAG_CACHE_CONTAINER_NAME |
The Azure Storage container name to use when in blob mode. |
str |
optional |
None |
GRAPHRAG_CACHE_BASE_DIR |
The base path to the cache files. |
str |
optional |
None |
Reporting
This section controls the reporting mechanism used by the pipeline, for common events and error messages. The default is to write reports to a file in the output directory. However, you can also choose to write reports to the console or to an Azure Blob Storage container.
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_REPORTING_TYPE |
The type of reporter to use. Options are file , console , or blob |
str |
optional |
file |
GRAPHRAG_REPORTING_STORAGE_ACCOUNT_BLOB_URL |
The Azure Storage blob endpoint to use when in blob mode and using managed identity. Will have the format https://<storage_account_name>.blob.core.windows.net |
str |
optional |
None |
GRAPHRAG_REPORTING_CONNECTION_STRING |
The Azure Storage connection string to use when in blob mode. |
str |
optional |
None |
GRAPHRAG_REPORTING_CONTAINER_NAME |
The Azure Storage container name to use when in blob mode. |
str |
optional |
None |
GRAPHRAG_REPORTING_BASE_DIR |
The base path to the reporting outputs. |
str |
optional |
None |
Node2Vec Parameters
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_NODE2VEC_ENABLED |
Whether to enable Node2Vec |
bool |
optional |
False |
GRAPHRAG_NODE2VEC_NUM_WALKS |
The Node2Vec number of walks to perform |
int |
optional |
10 |
GRAPHRAG_NODE2VEC_WALK_LENGTH |
The Node2Vec walk length |
int |
optional |
40 |
GRAPHRAG_NODE2VEC_WINDOW_SIZE |
The Node2Vec window size |
int |
optional |
2 |
GRAPHRAG_NODE2VEC_ITERATIONS |
The number of iterations to run node2vec |
int |
optional |
3 |
GRAPHRAG_NODE2VEC_RANDOM_SEED |
The random seed to use for node2vec |
int |
optional |
597832 |
Data Snapshotting
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_SNAPSHOT_EMBEDDINGS |
Whether to enable embeddings snapshots. |
bool |
optional |
False |
GRAPHRAG_SNAPSHOT_GRAPHML |
Whether to enable GraphML snapshots. |
bool |
optional |
False |
GRAPHRAG_SNAPSHOT_RAW_ENTITIES |
Whether to enable raw entity snapshots. |
bool |
optional |
False |
GRAPHRAG_SNAPSHOT_TOP_LEVEL_NODES |
Whether to enable top-level node snapshots. |
bool |
optional |
False |
GRAPHRAG_SNAPSHOT_TRANSIENT |
Whether to enable transient table snapshots. |
bool |
optional |
False |
Miscellaneous Settings
Parameter |
Description |
Type |
Required or Optional |
Default |
GRAPHRAG_ASYNC_MODE |
Which async mode to use. Either asyncio or threaded . |
str |
optional |
asyncio |
GRAPHRAG_ENCODING_MODEL |
The text encoding model, used in tiktoken, to encode text. |
str |
optional |
cl100k_base |
GRAPHRAG_MAX_CLUSTER_SIZE |
The maximum number of entities to include in a single Leiden cluster. |
int |
optional |
10 |
GRAPHRAG_SKIP_WORKFLOWS |
A comma-separated list of workflow names to skip. |
str |
optional |
None |
GRAPHRAG_UMAP_ENABLED |
Whether to enable UMAP layouts |
bool |
optional |
False |