Files changed (1) hide show
  1. functions/query_functions.py +1 -0
functions/query_functions.py CHANGED
@@ -163,6 +163,7 @@ def query_func(queries:List[str], session_hash, session_folder, args, **kwargs):
163
  graphql_object = GraphQLQuery()
164
  result = graphql_object.run(queries, args[0], args[1], args[2], session_hash)
165
  print("RESULT")
 
166
  if len(result["results"][0]) > 1000:
167
  print("QUERY TOO LARGE")
168
  return {"reply": f"""query result too large to be processed by llm, the query results are in our query.csv file.
 
163
  graphql_object = GraphQLQuery()
164
  result = graphql_object.run(queries, args[0], args[1], args[2], session_hash)
165
  print("RESULT")
166
+ result["csv_columns"]
167
  if len(result["results"][0]) > 1000:
168
  print("QUERY TOO LARGE")
169
  return {"reply": f"""query result too large to be processed by llm, the query results are in our query.csv file.