(fix)[headless]change viewName to datasetName

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

View File

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

View File

@@ -50,11 +50,11 @@ async def query2sql(query_body: Mapping[str, Any]):
else: else:
sql_generation_mode = query_body['sqlGenerationMode'] sql_generation_mode = query_body['sqlGenerationMode']
viewName = schema['viewName'] dataset_name = schema['dataSetName']
fields_list = schema['fieldNameList'] fields_list = schema['fieldNameList']
prior_schema_links = {item['fieldValue']:item['fieldName'] for item in linking} 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) data_date=current_date, prior_schema_links=prior_schema_links, prior_exts=prior_exts, sql_generation_mode=sql_generation_mode)
return resp return resp