diff --git a/graphrag/config/defaults.py b/graphrag/config/defaults.py index c7880616..08903630 100644 --- a/graphrag/config/defaults.py +++ b/graphrag/config/defaults.py @@ -104,3 +104,28 @@ GLOBAL_SEARCH_DATA_MAX_TOKENS = 12_000 GLOBAL_SEARCH_MAP_MAX_TOKENS = 1000 GLOBAL_SEARCH_REDUCE_MAX_TOKENS = 2_000 GLOBAL_SEARCH_CONCURRENCY = 32 + +# DRIFT Search + +DRIFT_SEARCH_LLM_TEMPERATURE = 0 +DRIFT_SEARCH_LLM_TOP_P = 1 +DRIFT_SEARCH_LLM_N = 3 +DRIFT_SEARCH_MAX_TOKENS = 12_000 +DRIFT_SEARCH_DATA_MAX_TOKENS = 12_000 +DRIFT_SEARCH_CONCURRENCY = 32 + +DRIFT_SEARCH_PRIMER_K = 20 +DRIFT_SEARCH_PRIMER_FOLDS = 5 +DRIFT_SEARCH_PRIMER_MAX_TOKENS = 12_000 + +DRIFT_LOCAL_SEARCH_TEXT_UNIT_PROP = 0.9 +DRIFT_LOCAL_SEARCH_COMMUNITY_PROP = 0.1 +DRIFT_LOCAL_SEARCH_TOP_K_MAPPED_ENTITIES = 10 +DRIFT_LOCAL_SEARCH_TOP_K_RELATIONSHIPS = 10 +DRIFT_LOCAL_SEARCH_MAX_TOKENS = 12_000 +DRIFT_LOCAL_SEARCH_LLM_TEMPERATURE = 0 +DRIFT_LOCAL_SEARCH_LLM_TOP_P = 1 +DRIFT_LOCAL_SEARCH_LLM_N = 1 +DRIFT_LOCAL_SEARCH_LLM_MAX_TOKENS = 2000 + +DRIFT_SEARCH_STEPS = 3