mirror of
https://github.com/APIParkLab/APIPark.git
synced 2026-06-04 10:13:53 +08:00
sunanzhi
43 lines
1.6 KiB
Go
43 lines
1.6 KiB
Go
package ai
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
|
|
"github.com/APIParkLab/APIPark/gateway"
|
|
ai_dto "github.com/APIParkLab/APIPark/module/ai/dto"
|
|
"github.com/eolinker/go-common/autowire"
|
|
)
|
|
|
|
type IProviderModule interface {
|
|
ConfiguredProviders(ctx context.Context, keyword string) ([]*ai_dto.ConfiguredProviderItem, error)
|
|
UnConfiguredProviders(ctx context.Context) ([]*ai_dto.ProviderItem, error)
|
|
SimpleProviders(ctx context.Context) ([]*ai_dto.SimpleProviderItem, error)
|
|
SimpleConfiguredProviders(ctx context.Context, all bool) ([]*ai_dto.SimpleProviderItem, *ai_dto.BackupProvider, error)
|
|
Provider(ctx context.Context, id string) (*ai_dto.Provider, error)
|
|
SimpleProvider(ctx context.Context, id string) (*ai_dto.SimpleProvider, error)
|
|
LLMs(ctx context.Context, driver string) ([]*ai_dto.LLMItem, *ai_dto.ProviderItem, error)
|
|
UpdateProviderConfig(ctx context.Context, id string, input *ai_dto.UpdateConfig) error
|
|
Delete(ctx context.Context, id string) error
|
|
AddProvider(ctx context.Context, input *ai_dto.NewProvider) (*ai_dto.SimpleProvider, error)
|
|
}
|
|
|
|
type IAIAPIModule interface {
|
|
APIs(ctx context.Context, keyword string, providerId string, start int64, end int64, page int, pageSize int, sortCondition string, asc bool, models []string, services []string) ([]*ai_dto.APIItem, *ai_dto.Condition, int64, error)
|
|
}
|
|
|
|
type ILocalModelModule interface {
|
|
}
|
|
|
|
func init() {
|
|
autowire.Auto[IProviderModule](func() reflect.Value {
|
|
module := new(imlProviderModule)
|
|
gateway.RegisterInitHandleFunc(module.initGateway)
|
|
return reflect.ValueOf(module)
|
|
})
|
|
|
|
autowire.Auto[IAIAPIModule](func() reflect.Value {
|
|
return reflect.ValueOf(new(imlAIApiModule))
|
|
})
|
|
}
|