zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-17 05:07:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go CLI: fix list provider instance tasks (#16345)

Browse Source
This commit is contained in:
Jin Hai
2026-06-25 15:49:31 +08:00
committed by GitHub
parent 3f3a2ece3d
commit edfa9be67f
6 changed files with 183 additions and 234 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
internal/cli/common_command.go
View File

@@ -1056,7 +1056,7 @@ func (c *CLI) ListDefaultModels(cmd *Command) (ResponseIf, error) {
return &result, nil
}
func (c *CLI) CommonShowCurrent(cmd *Command) (ResponseIf, error) {
func (c *CLI) CommonShowCurrentCommand(cmd *Command) (ResponseIf, error) {
var result *CommonDataResponse
switch c.Config.CLIMode {
@@ -1117,7 +1117,7 @@ func (c *CLI) ShowAPIServer(cmd *Command) (ResponseIf, error) {
return result, nil
}
func (c *CLI) ListAPIServer(cmd *Command) (ResponseIf, error) {
func (c *CLI) CommonListAPIServers(cmd *Command) (ResponseIf, error) {
var result CommonResponse
result.Data = make([]map[string]interface{}, 0)
@@ -1317,7 +1317,7 @@ func (c *CLI) SaveServerConfig(cmd *Command) (ResponseIf, error) {
return nil, fmt.Errorf("invalid server type")
}
return nil, nil
return nil, fmt.Errorf("save server config isn't implemented")
}
func (c *CLI) GetAdminServerInfo() (ResponseIf, error) {
@@ -1363,12 +1363,17 @@ func (c *CLI) GetAPIServerInfo(serverName string) (ResponseIf, error) {
if apiServerConfig.UserPassword != nil {
result.Data["user_password"] = strings.Repeat("*", len(*apiServerConfig.UserPassword))
}
if c.APIServerClientMap[c.Config.APIClientConfig.CurrentAPIServer].LoginToken != nil {
result.Data["auth"] = "login"
} else if c.APIServerClientMap[c.Config.APIClientConfig.CurrentAPIServer].APIKey != nil {
result.Data["auth"] = "api key"
if c.Config.APIClientConfig.CurrentAPIServer == "" {
result.Data["auth"] = "unknown"
} else {
result.Data["auth"] = "no auth"
if c.APIServerClientMap[c.Config.APIClientConfig.CurrentAPIServer].LoginToken != nil {
result.Data["auth"] = "login"
} else if c.APIServerClientMap[c.Config.APIClientConfig.CurrentAPIServer].APIKey != nil {
result.Data["auth"] = "api key"
} else {
result.Data["auth"] = "no auth"
}
}
}
return &result, nil
@@ -1524,7 +1529,7 @@ func FlattenMap(data map[string]interface{}, prefix string, result *[]map[string
}
}
func (c *CLI) UseAPIServer(cmd *Command) (ResponseIf, error) {
func (c *CLI) CommonUseAPIServerCommand(cmd *Command) (ResponseIf, error) {
serverName, ok := cmd.Params["server_name"].(string)
if !ok {
return nil, fmt.Errorf("server_name not provided")
@@ -1564,7 +1569,7 @@ func (c *CLI) UseAPIServer(cmd *Command) (ResponseIf, error) {
}
func (c *CLI) UseAdminServer(cmd *Command) (ResponseIf, error) {
func (c *CLI) CommonUseAdminServerCommand(cmd *Command) (ResponseIf, error) {
if c.Config.CLIMode == AdminMode {
return nil, fmt.Errorf("already in admin mode")
@@ -1589,7 +1594,10 @@ func (c *CLI) getDatasetIDByName(datasetName string) (string, error) {
if err != nil {
return "", err
}
commonResponse := response.(*CommonResponse)
commonResponse, ok := response.(*CommonResponse)
if !ok {
return "", fmt.Errorf("invalid response")
}
for _, dataset := range commonResponse.Data {
if dataset["name"] == datasetName {
return dataset["id"].(string), nil