Merge pull request #897 from chat2db/dev

Dev

README.md

@@ -104,8 +104,7 @@ Redis and MongoDB are partially supported , Hbase、Elasticsearch、openGauss、
 
 ### CONFIGURE CUSTOM AI
 
-* [Refer here to deploy your ChatGLM-6B model](https://github.com/chat2db/chat2db-chatglm-6b-deploy)
-* [Refer here to deploy your sqlcoder model](https://github.com/chat2db/chat2db-sqlcoder-deploy)
+* The rest api format for Custom AI is same as ChatGPT.
 
 ## 📦 Docker installation
 

README_CN.md

@@ -100,8 +100,7 @@ Redis和MongoDB得到部分支持，Hbase、Elasticsearch、openGauss、TiDB、I
 ### 使用Chat2DB AI 上手即用
 
 ### 使用自定义大模型
-- [参考这里部署本地ChatGLM-6B模型](https://github.com/chat2db/chat2db-chatglm-6b-deploy/blob/main/README_CN.md)
-- [参考这里部署本地sqlcoder模型](https://github.com/chat2db/chat2db-sqlcoder-deploy/blob/main/README_CN.md)
+- 使用自定义大模型，接口格式需要和open ai的接口格式保持一致
 
 ## 📦 Docker 部署
 

chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/DataSourceServiceImpl.java

@@ -95,7 +95,7 @@ public class DataSourceServiceImpl implements DataSourceService {
         dataSourceDO.setGmtCreate(DateUtil.date());
         dataSourceDO.setGmtModified(DateUtil.date());
         dataSourceDO.setUserId(ContextUtils.getUserId());
-        dataSourceDO.setExtendInfo(null);
+        //dataSourceDO.setExtendInfo(null);
 
         dataSourceMapper.insert(dataSourceDO);
         preWarmingData(dataSourceDO.getId());

chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/ConnectInfo.java

@@ -167,7 +167,15 @@ public class ConnectInfo {
 
 
     public LinkedHashMap<String,Object> getExtendMap() {
+
         if (ObjectUtils.isEmpty(extendInfo)) {
+            if(driverConfig!= null) {
+                extendInfo = driverConfig.getExtendInfo();
+            }else {
+                return new LinkedHashMap<>();
+            }
+        }
+        if(ObjectUtils.isEmpty(extendInfo)){
             return new LinkedHashMap<>();
         }
         LinkedHashMap<String,Object> map = new LinkedHashMap<>();