topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 03:58:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(Chat) Fix modelName to viewName in python (#786)

Browse Source 
Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-03-05 21:43:33 +08:00
committed by GitHub
parent 0b9bfebacc
commit eaa4cc5a1b
3 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
chat/python/services_router/query2sql_service.py
View File

@@ -50,11 +50,11 @@ async def query2sql(query_body: Mapping[str, Any]):
else:
sql_generation_mode = query_body['sqlGenerationMode']
model_name = schema['modelName']
viewName = schema['viewName']
fields_list = schema['fieldNameList']
prior_schema_links = {item['fieldValue']:item['fieldName'] for item in linking}
resp = await text2sql_agent_router.async_query2sql(question=query_text, filter_condition=filter_condition, model_name=model_name, fields_list=fields_list,
resp = await text2sql_agent_router.async_query2sql(question=query_text, filter_condition=filter_condition, model_name=viewName, fields_list=fields_list,
data_date=current_date, prior_schema_links=prior_schema_links, prior_exts=prior_exts, sql_generation_mode=sql_generation_mode)
return resp