Browse Source

fix: fix baidu's embedding api (#398)

* fix:judge baidu embeddings input type

* chore: add back update all channels balance

* chore: update default configuration for sql connection

---------

Co-authored-by: JustSong <[email protected]>
igophper 2 years ago
parent
commit
4ef5e2020c
4 changed files with 13 additions and 9 deletions
  1. 2 2
      README.md
  2. 6 2
      controller/relay-baidu.go
  3. 2 2
      model/main.go
  4. 3 3
      web/src/components/ChannelsTable.js

+ 2 - 2
README.md

@@ -284,8 +284,8 @@ graph LR
    + 如果使用本地数据库:部署命令可添加 `--network="host"` 以使得容器内的程序可以访问到宿主机上的 MySQL。
    + 如果使用本地数据库:部署命令可添加 `--network="host"` 以使得容器内的程序可以访问到宿主机上的 MySQL。
    + 如果使用云数据库:如果云服务器需要验证身份,需要在连接参数中添加 `?tls=skip-verify`。
    + 如果使用云数据库:如果云服务器需要验证身份,需要在连接参数中添加 `?tls=skip-verify`。
    + 请根据你的数据库配置修改下列参数(或者保持默认值):
    + 请根据你的数据库配置修改下列参数(或者保持默认值):
-     + `SQL_MAX_IDLE_CONNS`:最大空闲连接数,默认为 `10`。
-     + `SQL_MAX_OPEN_CONNS`:最大打开连接数,默认为 `100`。
+     + `SQL_MAX_IDLE_CONNS`:最大空闲连接数,默认为 `100`。
+     + `SQL_MAX_OPEN_CONNS`:最大打开连接数,默认为 `1000`。
        + 如果报错 `Error 1040: Too many connections`,请适当减小该值。
        + 如果报错 `Error 1040: Too many connections`,请适当减小该值。
      + `SQL_CONN_MAX_LIFETIME`:连接的最大生命周期,默认为 `60`,单位分钟。
      + `SQL_CONN_MAX_LIFETIME`:连接的最大生命周期,默认为 `60`,单位分钟。
 4. `FRONTEND_BASE_URL`:设置之后将重定向页面请求到指定的地址,仅限从服务器设置。
 4. `FRONTEND_BASE_URL`:设置之后将重定向页面请求到指定的地址,仅限从服务器设置。

+ 6 - 2
controller/relay-baidu.go

@@ -150,8 +150,12 @@ func embeddingRequestOpenAI2Baidu(request GeneralOpenAIRequest) *BaiduEmbeddingR
 	switch request.Input.(type) {
 	switch request.Input.(type) {
 	case string:
 	case string:
 		baiduEmbeddingRequest.Input = []string{request.Input.(string)}
 		baiduEmbeddingRequest.Input = []string{request.Input.(string)}
-	case []string:
-		baiduEmbeddingRequest.Input = request.Input.([]string)
+	case []any:
+		for _, item := range request.Input.([]any) {
+			if str, isStr := item.(string); isStr {
+				baiduEmbeddingRequest.Input = append(baiduEmbeddingRequest.Input, str)
+			}
+		}
 	}
 	}
 	return &baiduEmbeddingRequest
 	return &baiduEmbeddingRequest
 }
 }

+ 2 - 2
model/main.go

@@ -74,8 +74,8 @@ func InitDB() (err error) {
 		if err != nil {
 		if err != nil {
 			return err
 			return err
 		}
 		}
-		sqlDB.SetMaxIdleConns(common.GetOrDefault("SQL_MAX_IDLE_CONNS", 10))
-		sqlDB.SetMaxOpenConns(common.GetOrDefault("SQL_MAX_OPEN_CONNS", 100))
+		sqlDB.SetMaxIdleConns(common.GetOrDefault("SQL_MAX_IDLE_CONNS", 100))
+		sqlDB.SetMaxOpenConns(common.GetOrDefault("SQL_MAX_OPEN_CONNS", 1000))
 		sqlDB.SetConnMaxLifetime(time.Second * time.Duration(common.GetOrDefault("SQL_MAX_LIFETIME", 60)))
 		sqlDB.SetConnMaxLifetime(time.Second * time.Duration(common.GetOrDefault("SQL_MAX_LIFETIME", 60)))
 
 
 		if !common.IsMasterNode {
 		if !common.IsMasterNode {

+ 3 - 3
web/src/components/ChannelsTable.js

@@ -368,7 +368,7 @@ const ChannelsTable = () => {
                       }} style={{ cursor: 'pointer' }}>
                       }} style={{ cursor: 'pointer' }}>
                       {renderBalance(channel.type, channel.balance)}
                       {renderBalance(channel.type, channel.balance)}
                     </span>}
                     </span>}
-                      content="点击更新"
+                      content='点击更新'
                       basic
                       basic
                     />
                     />
                   </Table.Cell>
                   </Table.Cell>
@@ -447,8 +447,8 @@ const ChannelsTable = () => {
               <Button size='small' loading={loading} onClick={testAllChannels}>
               <Button size='small' loading={loading} onClick={testAllChannels}>
                 测试所有已启用通道
                 测试所有已启用通道
               </Button>
               </Button>
-              {/* <Button size='small' onClick={updateAllChannelsBalance}
-                      loading={loading || updatingBalance}>更新所有已启用通道余额</Button> */}
+              <Button size='small' onClick={updateAllChannelsBalance}
+                      loading={loading || updatingBalance}>更新所有已启用通道余额</Button>
               <Pagination
               <Pagination
                 floated='right'
                 floated='right'
                 activePage={activePage}
                 activePage={activePage}