fix: longer questions

aymenfurter · Aug 7, 2024 · 5a27747 · 5a27747
1 parent 981724e
commit 5a27747
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/query/graphrag_query.py b/app/query/graphrag_query.py
@@ -75,8 +75,8 @@ async def global_query(self, query: str):
             context_builder=context_builder,
             token_encoder=token_encoder,
             max_data_tokens=3000,
-            map_llm_params={"max_tokens": 500, "temperature": 0.0},
-            reduce_llm_params={"max_tokens": 500, "temperature": 0.0},
+            map_llm_params={"max_tokens": 1000, "temperature": 0.0},
+            reduce_llm_params={"max_tokens": 2000, "temperature": 0.0},
             context_builder_params={
                 "use_community_summary": False,
                 "shuffle_data": True,