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

(improvement)(Headless) Fix python llm proxy sqlWeight npe (#902) (#903)

Browse Source 
* (improvement)(Headless) fix python llm proxy sqlWeight npe

---------

Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-04-10 17:48:06 +08:00
committed by GitHub
parent d754bd50b0
commit f2b9b61be6
3 changed files with 54 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
headless/core/src/main/java/com/tencent/supersonic/headless/core/chat/parser/llm/OutputFormat.java
View File

@@ -113,6 +113,9 @@ public class OutputFormat {
public static Map<String, LLMSqlResp> buildSqlRespMap(List<Map<String, String>> sqlExamples,
Map<String, Double> sqlMap) {
if (sqlMap == null) {
return new HashMap<>();
}
return sqlMap.entrySet().stream()
.collect(Collectors.toMap(
Map.Entry::getKey,

2
headless/python/services/s2sql/run.py
View File

@@ -51,7 +51,7 @@ text2sql_agent.reload_setting(sql_ids, sql_exemplars, TEXT2DSL_EXAMPLE_NUM, TEXT
if text2sql_agent_autoCoT.count_examples()==0:
source_dir_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
example_dir_path = os.path.join(source_dir_path, 'few_shot_example')
example_json_file = os.path.join(example_dir_path, 's2sql_exemplar3_transformed.json.json')
example_json_file = os.path.join(example_dir_path, 's2sql_exemplar3_transformed.json')
with open(example_json_file, 'r', encoding='utf-8') as f:
transformed_sql_examplar_list = json.load(f)