mirror of
https://github.com/APIParkLab/APIPark.git
synced 2026-06-12 18:11:34 +08:00
Liujian
30 lines
986 B
Go
30 lines
986 B
Go
package ai_local
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
|
|
"github.com/eolinker/go-common/autowire"
|
|
|
|
ai_provider_local "github.com/APIParkLab/APIPark/ai-provider/local"
|
|
|
|
ai_local_dto "github.com/APIParkLab/APIPark/module/ai-local/dto"
|
|
)
|
|
|
|
type ILocalModelModule interface {
|
|
Search(ctx context.Context, keyword string) ([]*ai_local_dto.LocalModelItem, error)
|
|
ListCanInstall(ctx context.Context, keyword string) ([]*ai_local_dto.LocalModelPackageItem, error)
|
|
Deploy(ctx context.Context, model string, session string) (*ai_provider_local.Pipeline, error)
|
|
CancelDeploy(ctx context.Context, model string) error
|
|
RemoveModel(ctx context.Context, model string) error
|
|
Enable(ctx context.Context, model string) error
|
|
Disable(ctx context.Context, model string) error
|
|
ModelState(ctx context.Context, model string) (*ai_local_dto.DeployState, *ai_local_dto.ModelInfo, error)
|
|
}
|
|
|
|
func init() {
|
|
autowire.Auto[ILocalModelModule](func() reflect.Value {
|
|
return reflect.ValueOf(&imlLocalModel{})
|
|
})
|
|
}
|