From 5be2caec9eba93aa3ab10502008785cf5d385247 Mon Sep 17 00:00:00 2001
From: oravidov <or@singular.net>
Date: Thu, 23 Sep 2021 15:30:18 +0300
Subject: [PATCH] Increased information schema result limit

---
 lineage/snowflake_query_history.py | 3 ++-
 setup.py                           | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/lineage/snowflake_query_history.py b/lineage/snowflake_query_history.py
index f53d01b4b..b93531ea9 100644
--- a/lineage/snowflake_query_history.py
+++ b/lineage/snowflake_query_history.py
@@ -15,7 +15,8 @@ class SnowflakeQueryHistory(QueryHistory):
     select query_text, database_name, schema_name
       from table(information_schema.query_history(
         end_time_range_start=>to_timestamp_ltz(:2),
-        {end_time_range_end_expr})) 
+        {end_time_range_end_expr},
+        result_limit=>10000)) 
         where execution_status = 'SUCCESS' and query_type not in 
         ('SHOW', 'COPY', 'COMMIT', 'DESCRIBE', 'ROLLBACK', 'CREATE_STREAM', 'DROP_STREAM', 'PUT_FILES', 
         'BEGIN_TRANSACTION', 'GRANT', 'ALTER_SESSION', 'USE') and
diff --git a/setup.py b/setup.py
index 450897446..6fa16a56b 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@
 setup(
     name='elementary-lineage',
     description='Presenting data lineage based on your data warehouse query history',
-    version='0.0.5',
+    version='0.0.6',
     packages=find_packages(),
     include_package_data=True,
     python_requires='>=3.6.2',