chore(wren-ai-service): minor updates (#854)

wren-ai-service/src/pipelines/common.py

@@ -268,6 +268,7 @@ async def _task(result: Dict[str, str]):
 - YOU MUST USE "JOIN" if you choose columns from multiple tables!
 - YOU MUST USE "lower(<column_name>) = lower(<value>)" function for case-insensitive comparison!
 - DON'T USE "DATE_ADD" or "DATE_SUB" functions for date operations, instead use syntax like this "current_date - INTERVAL '7' DAY"!
+- DON'T USE "COUNT(*) FILTER(WHERE <condition>)"
 - ALWAYS ADD "timestamp" to the front of the timestamp literal, ex. "timestamp '2024-02-20 12:00:00'"
 - USE THE VIEW TO SIMPLIFY THE QUERY.
 - DON'T MISUSE THE VIEW NAME. THE ACTUAL NAME IS FOLLOWING THE CREATE VIEW STATEMENT.

wren-ai-service/tests/pytest/test_usecases.py

@@ -7,6 +7,7 @@
 import aiohttp
 import orjson
 import requests
+import toml
 
 from demo.utils import _get_connection_info, _replace_wren_engine_env_variables
 
@@ -24,11 +25,11 @@ def test_load_mdl_and_questions():
         with open("tests/data/hubspot/mdl.json", "r") as f:
             mdl_str = orjson.dumps(json.load(f)).decode("utf-8")
 
-        with open("tests/data/hubspot/questions.json", "r") as f:
-            questions = json.load(f)["questions"]
+        data = toml.load("tests/data/hubspot/eval_dataset.toml")
+        questions = [item["question"] for item in data.get("eval_dataset", [])]
     except FileNotFoundError:
         raise Exception(
-            "tests/data/hubspot/mdl.json or tests/data/hubspot/questions.json not found"
+            "tests/data/hubspot/mdl.json or tests/data/hubspot/eval_dataset.toml not found"
         )
 
     return mdl_str, questions