Log csv headers (#43)
Browse files- Log csv headers (3d536a5418ce09384ace9174a28ba42a2ae658a0)
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.
|