mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-01 05:23:47 +08:00
Go CLI: update list supported models (#15845)
### 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:
@@ -402,7 +402,7 @@ func (m *ModelScopeModel) ParseFile(modelName *string, content []byte, url *stri
|
||||
|
||||
// ListModels returns the model IDs exposed by ModelScope's OpenAI-compatible
|
||||
// /v1/models endpoint.
|
||||
func (m *ModelScopeModel) ListModels(apiConfig *APIConfig) ([]string, error) {
|
||||
func (m *ModelScopeModel) ListModels(apiConfig *APIConfig) ([]ListModelResponse, error) {
|
||||
if err := m.baseModel.APIConfigCheck(apiConfig); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -446,10 +446,10 @@ func (m *ModelScopeModel) ListModels(apiConfig *APIConfig) ([]string, error) {
|
||||
return nil, fmt.Errorf("failed to parse response: %w", err)
|
||||
}
|
||||
|
||||
models := make([]string, 0, len(result.Data))
|
||||
models := make([]ListModelResponse, 0, len(result.Data))
|
||||
for _, model := range result.Data {
|
||||
if model.ID != "" {
|
||||
models = append(models, model.ID)
|
||||
models = append(models, ListModelResponse{Name: model.ID})
|
||||
}
|
||||
}
|
||||
return models, nil
|
||||
|
||||
Reference in New Issue
Block a user