From 2df0567f5c8c80ed10b1a657a6f8f227a537e5bb Mon Sep 17 00:00:00 2001 From: Julian Whiting Date: Tue, 8 Oct 2024 11:56:32 -0400 Subject: [PATCH] add defs for drift config --- graphrag/config/defaults.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) 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