add quotes for the fallback option when sql breakdown fails (#480)

* add quotes

* fix failed test
This commit is contained in:
Chih-Yu Yeh
2024-07-04 10:53:48 +08:00
committed by GitHub
parent db83edbcf4
commit 445e4bbfee
2 changed files with 3 additions and 2 deletions
@@ -4,6 +4,7 @@ from typing import List, Literal, Optional
from haystack import Pipeline
from pydantic import BaseModel
from src.core.engine import add_quotes
from src.utils import async_timer
logger = logging.getLogger("wren-ai-service")
@@ -93,7 +94,7 @@ class AskDetailsService:
if not ask_details_result["steps"]:
ask_details_result["steps"] = [
{
"sql": ask_details_request.sql,
"sql": add_quotes(ask_details_request.sql),
"summary": ask_details_request.summary,
"cte_name": "",
}
@@ -70,7 +70,7 @@ def test_ask_details_with_successful_sql(ask_details_service: AskDetailsService)
def test_ask_details_with_failed_sql(ask_details_service: AskDetailsService):
# asking details
query_id = str(uuid.uuid4())
sql = "SELECT * FROM xxx"
sql = 'SELECT * FROM "xxx"'
summary = "This is a summary"
ask_details_request = AskDetailsRequest(
query="How many books are there?'",