diff --git a/module/ai-local/dto/output.go b/module/ai-local/dto/output.go index 58c224b3..74cc946a 100644 --- a/module/ai-local/dto/output.go +++ b/module/ai-local/dto/output.go @@ -65,13 +65,13 @@ type SimpleItem struct { } type LocalModelItem struct { - Id string `json:"id"` - Name string `json:"name"` - State LocalModelState `json:"state"` - APICount int64 `json:"api_count"` - - UpdateTime auto.TimeLabel `json:"update_time"` - CanDelete bool `json:"can_delete"` + Id string `json:"id"` + Name string `json:"name"` + State LocalModelState `json:"state"` + APICount int64 `json:"api_count"` + Provider string `json:"provider"` + UpdateTime auto.TimeLabel `json:"update_time"` + CanDelete bool `json:"can_delete"` } type LocalModelPackageItem struct { diff --git a/module/ai-local/iml.go b/module/ai-local/iml.go index f50483b7..8380a2f3 100644 --- a/module/ai-local/iml.go +++ b/module/ai-local/iml.go @@ -96,6 +96,7 @@ func (i *imlLocalModel) Search(ctx context.Context, keyword string) ([]*ai_local APICount: apiCountMap[s.Id], CanDelete: true, UpdateTime: auto.TimeLabel(s.UpdateAt), + Provider: "ollama", } }), nil }