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

(fix)[headless]change viewName to datasetName

Browse Source
This commit is contained in:
jerryjzhang
2024-04-09 22:08:53 +08:00
parent d6b81249d7
commit d754bd50b0
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
assembly/build/build.xml
View File

@@ -21,7 +21,7 @@
</includes>
</fileSet>
<fileSet>
<directory>${project.basedir}/../../chat/python</directory>
<directory>${project.basedir}/../../headless/python</directory>
<outputDirectory>pyllm</outputDirectory>
<fileMode>0777</fileMode>
<directoryMode>0755</directoryMode>

4
headless/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']
viewName = schema['viewName']
dataset_name = schema['dataSetName']
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=viewName, fields_list=fields_list,
resp = await text2sql_agent_router.async_query2sql(question=query_text, filter_condition=filter_condition, model_name=dataset_name, 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