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

Go CLI: update list supported models (#15845)

Browse Source 
### What problem does this PR solve?

Now list supported models will show more info.

```
RAGFlow(api/default)> list supported models from 'gitee' 'test';
+-----------+------------+-------------+----------------------------------------------------------+---------------------------------------------+
| dimension | max_tokens | model_types | name                                                     | thinking                                    |
+-----------+------------+-------------+----------------------------------------------------------+---------------------------------------------+
|           |            |             | Wan2.7                                                   |                                             |
|           |            |             | HappyHorse-1.0                                           |                                             |
|           |            |             | Qwen3.6-27B@Qwen                                         |                                             |
|           |            |             | Qwen3.6-35B-A3B@Qwen                                     |                                             |
|           | 1048576    | [chat]      | DeepSeek-V4-Flash@deepseek-ai                            | map[clear_thinking:true default_value:true] |
|           | 1048576    | [chat]      | DeepSeek-V4-Pro@deepseek-ai                              | map[clear_thinking:true default_value:true] |
+-----------+------------+-------------+----------------------------------------------------------+---------------------------------------------+
```

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-06-09 19:01:00 +08:00
committed by GitHub
parent f0efa63bf2
commit 719ce15c95
68 changed files with 357 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/entity/models/replicate.go
View File

@@ -504,7 +504,7 @@ func dispatchReplicateSSEEvent(event replicateSSEEvent, sender func(*string, *st
}
}
func (r *ReplicateModel) ListModels(apiConfig *APIConfig) ([]string, error) {
func (r *ReplicateModel) ListModels(apiConfig *APIConfig) ([]ListModelResponse, error) {
if err := r.baseModel.APIConfigCheck(apiConfig); err != nil {
return nil, err
}
@@ -543,10 +543,10 @@ func (r *ReplicateModel) ListModels(apiConfig *APIConfig) ([]string, error) {
return nil, fmt.Errorf("failed to parse response: %w", err)
}
models := make([]string, 0, len(result.Results))
models := make([]ListModelResponse, 0, len(result.Results))
for _, model := range result.Results {
if model.Owner != "" && model.Name != "" {
models = append(models, fmt.Sprintf("%s/%s", model.Owner, model.Name))
models = append(models, ListModelResponse{Name: model.Owner + "/" + model.Name})
}
}
return models, nil