Spaces:

nolanzandi
/

virtual-data-analyst

Running

nolanzandi commited on Mar 12

Commit

82636e0

verified ·

1 Parent(s): c2276e3

Upload sqlite_functions.py

Files changed (1) hide show

functions/sqlite_functions.py CHANGED Viewed

@@ -33,7 +33,7 @@ def sqlite_query_func(queries: List[str], session_hash):
     sql_query = SQLiteQuery(f'{dir_path}/data_source.db')
     try:
       result = sql_query.run(queries, session_hash)
-      if len(result["results"][0]) > 500:
         print("QUERY TOO LARGE")
         return {"reply": "query result too large to be processed by llm, the query results are in our query.csv file"}
       else:

     sql_query = SQLiteQuery(f'{dir_path}/data_source.db')
     try:
       result = sql_query.run(queries, session_hash)
+      if len(result["results"][0]) > 1000:
         print("QUERY TOO LARGE")
         return {"reply": "query result too large to be processed by llm, the query results are in our query.csv file"}
       else: