mirror of
https://github.com/Canner/WrenAI.git
synced 2026-09-24 23:29:49 +08:00
add quotes for the fallback option when sql breakdown fails (#480)
* add quotes * fix failed test
This commit is contained in:
@@ -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?'",
|
||||
|
||||
Reference in New Issue
Block a user