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

Fix mysql keyword conflict (#1245)

Browse Source
This commit is contained in:
二毛
2024-06-27 15:02:19 +08:00
committed by GitHub
parent 90f7a79380
commit d1ec6badce
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/executor/SqlExecutor.java
View File

@@ -35,7 +35,7 @@ public class SqlExecutor implements ChatExecutor {
.status(ChatMemoryDO.Status.PENDING) .status(ChatMemoryDO.Status.PENDING)
.question(chatExecuteContext.getQueryText()) .question(chatExecuteContext.getQueryText())
.s2sql(chatExecuteContext.getParseInfo().getSqlInfo().getS2SQL()) .s2sql(chatExecuteContext.getParseInfo().getSqlInfo().getS2SQL())
.schema(buildSchemaStr(chatExecuteContext.getParseInfo())) .dbSchema(buildSchemaStr(chatExecuteContext.getParseInfo()))
.build()); .build());
} }
} }

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/memory/MemoryReviewTask.java
View File

@@ -52,7 +52,7 @@ public class MemoryReviewTask {
.filter(c -> c.getStatus() == ChatMemoryDO.Status.PENDING) .filter(c -> c.getStatus() == ChatMemoryDO.Status.PENDING)
.forEach(m -> { .forEach(m -> {
Agent chatAgent = agentService.getAgent(m.getAgentId()); Agent chatAgent = agentService.getAgent(m.getAgentId());
String promptStr = String.format(INSTRUCTION, m.getQuestion(), m.getSchema(), m.getS2sql()); String promptStr = String.format(INSTRUCTION, m.getQuestion(), m.getDbSchema(), m.getS2sql());
Prompt prompt = PromptTemplate.from(promptStr).apply(Collections.EMPTY_MAP); Prompt prompt = PromptTemplate.from(promptStr).apply(Collections.EMPTY_MAP);
keyPipelineLog.info("MemoryReviewTask reqPrompt:{}", promptStr); keyPipelineLog.info("MemoryReviewTask reqPrompt:{}", promptStr);

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/persistence/dataobject/ChatMemoryDO.java
View File

@@ -25,7 +25,7 @@ public class ChatMemoryDO {
private Integer agentId; private Integer agentId;
@TableField("db_schema") @TableField("db_schema")
private String schema; private String dbSchema;
@TableField("s2_sql") @TableField("s2_sql")
private String s2sql; private String s2sql;

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java
View File

@@ -47,7 +47,7 @@ public class MemoryServiceImpl implements MemoryService {
exemplarService.storeExemplar(memory.getAgentId().toString(), exemplarService.storeExemplar(memory.getAgentId().toString(),
SqlExemplar.builder() SqlExemplar.builder()
.question(memory.getQuestion()) .question(memory.getQuestion())
.dbSchema(memory.getSchema()) .dbSchema(memory.getDbSchema())
.sql(memory.getS2sql()) .sql(memory.getS2sql())
.build()); .build());
memory.setStatus(Status.ENABLED); memory.setStatus(Status.ENABLED);