[improvement](chat) Support sqlGenerationMode in python interface (#521)

This commit is contained in:
lexluo09
2023-12-16 17:24:24 +08:00
committed by GitHub
parent f03da53d6f
commit 276b224c13
4 changed files with 6 additions and 4 deletions

View File

@@ -48,7 +48,7 @@ async def query2sql(query_body: Mapping[str, Any]):
if 'sqlGenerationMode' not in query_body:
raise HTTPException(status_code=400, detail="sql_generation_mode is not in query_body")
else:
sql_generation_mode = query_body['sql_generation_mode']
sql_generation_mode = query_body['sqlGenerationMode']
model_name = schema['modelName']
fields_list = schema['fieldNameList']