DH-4758/increasing the limit value to hundred

Dataherald · Sep 28, 2023 · 7c07709 · 7c07709
1 parent 21757ce
commit 7c07709
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py
@@ -563,7 +563,7 @@ def create_sql_agent(
         format_instructions: str = FORMAT_INSTRUCTIONS,
         input_variables: List[str] | None = None,
         max_examples: int = 20,
-        top_k: int = 13,
+        top_k: int = 100,
         max_iterations: int | None = 15,
         max_execution_time: float | None = None,
         early_stopping_method: str = "force",