diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java index 1cdbcd6f3..6ceeb1127 100644 --- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java +++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java @@ -71,8 +71,9 @@ public class MemoryServiceImpl implements MemoryService, CommandLineRunner { chatMemoryDO.setS2sql(chatMemoryUpdateReq.getS2sql()); chatMemoryDO.setDbSchema(chatMemoryUpdateReq.getDbSchema()); enableMemory(chatMemoryDO); - } else if ((MemoryStatus.DISABLED.equals(chatMemoryUpdateReq.getStatus())||MemoryStatus.PENDING.equals(chatMemoryUpdateReq.getStatus())) && hadEnabled) { - // Remove from vector DB when transitioning: launched→disabled OR enabled→pending + } else if ((MemoryStatus.DISABLED.equals(chatMemoryUpdateReq.getStatus()) + || MemoryStatus.PENDING.equals(chatMemoryUpdateReq.getStatus())) && hadEnabled) { + // Remove from vector DB when transitioning: launched→disabled OR enabled→pending disableMemory(chatMemoryDO); } LambdaUpdateWrapper updateWrapper = new LambdaUpdateWrapper<>(); diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java index c5d935b3b..a319b8491 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java @@ -52,7 +52,8 @@ public class PromptHelper { for (int i = 0; i < selfConsistencyNumber; i++) { List shuffledList = new ArrayList<>(exemplars); // only shuffle the exemplars from config - List subList=shuffledList.subList(llmReq.getDynamicExemplars().size(),shuffledList.size()); + List subList = + shuffledList.subList(llmReq.getDynamicExemplars().size(), shuffledList.size()); Collections.shuffle(subList); results.add(shuffledList.subList(0, Math.min(shuffledList.size(), fewShotNumber))); } diff --git a/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/BaseDbAdaptor.java b/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/BaseDbAdaptor.java index 2637a1a7f..4eac4b3c2 100644 --- a/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/BaseDbAdaptor.java +++ b/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/BaseDbAdaptor.java @@ -5,7 +5,6 @@ import com.tencent.supersonic.headless.api.pojo.DBColumn; import com.tencent.supersonic.headless.api.pojo.enums.FieldType; import com.tencent.supersonic.headless.core.pojo.ConnectInfo; import lombok.extern.slf4j.Slf4j; -import org.apache.commons.lang3.StringUtils; import java.sql.*; import java.util.ArrayList; @@ -148,7 +147,8 @@ public abstract class BaseDbAdaptor implements DbAdaptor { String url = connectionInfo.getUrl().toLowerCase(); // 设置通用属性 - properties.setProperty("user", connectionInfo.getUserName()); + String userName = Optional.ofNullable(connectionInfo.getUserName()).orElse(""); + properties.setProperty("user", userName); String password = Optional.ofNullable(connectionInfo.getPassword()).orElse(""); diff --git a/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/DuckdbAdaptor.java b/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/DuckdbAdaptor.java index 0d3481faf..dfa8d8ed7 100644 --- a/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/DuckdbAdaptor.java +++ b/headless/core/src/main/java/com/tencent/supersonic/headless/core/adaptor/db/DuckdbAdaptor.java @@ -10,6 +10,7 @@ import java.sql.DatabaseMetaData; import java.sql.ResultSet; import java.sql.SQLException; import java.util.List; +import java.util.Properties; @Slf4j public class DuckdbAdaptor extends DefaultDbAdaptor { @@ -23,7 +24,7 @@ public class DuckdbAdaptor extends DefaultDbAdaptor { String tableName) throws SQLException { List dbColumns = Lists.newArrayList(); DatabaseMetaData metaData = getDatabaseMetaData(connectInfo); - ResultSet columns = metaData.getColumns(schemaName, null, tableName, null); + ResultSet columns = metaData.getColumns(null, schemaName, tableName, null); while (columns.next()) { String columnName = columns.getString("COLUMN_NAME"); String dataType = columns.getString("TYPE_NAME"); @@ -42,4 +43,9 @@ public class DuckdbAdaptor extends DefaultDbAdaptor { return sql.replaceAll("`", ""); } + @Override + public Properties getProperties(ConnectInfo connectionInfo) { + return new Properties(); + } + } diff --git a/headless/core/src/main/java/com/tencent/supersonic/headless/core/translator/parser/calcite/DataModelNode.java b/headless/core/src/main/java/com/tencent/supersonic/headless/core/translator/parser/calcite/DataModelNode.java index f401ed500..1b1146e1b 100644 --- a/headless/core/src/main/java/com/tencent/supersonic/headless/core/translator/parser/calcite/DataModelNode.java +++ b/headless/core/src/main/java/com/tencent/supersonic/headless/core/translator/parser/calcite/DataModelNode.java @@ -36,8 +36,8 @@ public class DataModelNode extends SemanticNode { && !dataModel.getModelDetail().getSqlQuery().isEmpty()) { sqlTable = dataModel.getModelDetail().getSqlQuery(); // if model has sqlVariables, parse sqlVariables - if (Objects.nonNull(dataModel.getModelDetail().getSqlVariables()) && - !(CollectionUtils.isEmpty(dataModel.getModelDetail().getSqlVariables()))) { + if (Objects.nonNull(dataModel.getModelDetail().getSqlVariables()) + && !(CollectionUtils.isEmpty(dataModel.getModelDetail().getSqlVariables()))) { sqlTable = SqlVariableParseUtils.parse(sqlTable, dataModel.getModelDetail().getSqlVariables(), Lists.newArrayList()); } diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelConverter.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelConverter.java index 5f4a6cdf6..2ef90fb66 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelConverter.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelConverter.java @@ -156,7 +156,11 @@ public class ModelConverter { modelDetail.setSqlQuery(modelBuildReq.getSql()); } else { modelDetail.setQueryType(ModelDefineType.TABLE_QUERY.getName()); - modelDetail.setTableQuery(String.format("%s.%s", modelBuildReq.getDb(), tableName)); + if (modelBuildReq.getDb() != null) { + modelDetail.setTableQuery(String.format("%s.%s", modelBuildReq.getDb(), tableName)); + } else { + modelDetail.setTableQuery(tableName); + } } List fields = new ArrayList<>(); for (SemanticColumn semanticColumn : modelSchema.getSemanticColumns()) {