Skip to content

Commit 7ec3b0b

Browse files
authored
Merge pull request spring-ai-alibaba#24 from zxuexingzhijie/main
fix: fix the SQL generation logic when semantic consistency fails
2 parents cac3864 + a9a28e8 commit 7ec3b0b

File tree

2 files changed

+2
-72
lines changed

2 files changed

+2
-72
lines changed

spring-ai-alibaba-data-agent-chat/src/main/java/com/alibaba/cloud/ai/node/SqlGenerateNode.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,8 @@ public Map<String, Object> apply(OverAllState state) throws Exception {
100100
else if (isSemanticConsistencyFailed(state)) {
101101
displayMessage = "语义一致性校验未通过,开始重新生成SQL...";
102102
String newSql = handleSemanticConsistencyFailure(state, toolParameters);
103-
result = Map.of(SQL_GENERATE_OUTPUT, newSql, RESULT, newSql);
103+
toolParameters.setSqlQuery(newSql);
104+
result = Map.of(SQL_GENERATE_OUTPUT, SQL_EXECUTE_NODE, PLANNER_NODE_OUTPUT, plan.toJsonStr());
104105
logger.info("[{}] Regenerated SQL due to semantic consistency failure: {}", this.getClass().getSimpleName(),
105106
newSql);
106107
}

spring-ai-alibaba-data-agent-management/PromptConfig.http

Lines changed: 0 additions & 71 deletions
This file was deleted.

0 commit comments

Comments
 (0)