Skip to content

Commit 4034562

Browse files
committed
chore: simplify passing options to ModelOptions
Signed-off-by: Ettore Di Giacinto <[email protected]>
1 parent 650134d commit 4034562

File tree

12 files changed

+13
-23
lines changed

12 files changed

+13
-23
lines changed

Diff for: core/backend/embeddings.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ func ModelEmbedding(s string, tokens []int, loader *model.ModelLoader, backendCo
1414
var inferenceModel interface{}
1515
var err error
1616

17-
opts := ModelOptions(backendConfig, appConfig, []model.Option{})
17+
opts := ModelOptions(backendConfig, appConfig)
1818

1919
if backendConfig.Backend == "" {
2020
inferenceModel, err = loader.GreedyLoader(opts...)

Diff for: core/backend/image.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ import (
99

1010
func ImageGeneration(height, width, mode, step, seed int, positive_prompt, negative_prompt, src, dst string, loader *model.ModelLoader, backendConfig config.BackendConfig, appConfig *config.ApplicationConfig) (func() error, error) {
1111

12-
opts := ModelOptions(backendConfig, appConfig, []model.Option{})
12+
opts := ModelOptions(backendConfig, appConfig)
1313

1414
inferenceModel, err := loader.BackendLoader(
1515
opts...,

Diff for: core/backend/llm.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ func ModelInference(ctx context.Context, s string, messages []schema.Message, im
3939
var inferenceModel grpc.Backend
4040
var err error
4141

42-
opts := ModelOptions(c, o, []model.Option{})
42+
opts := ModelOptions(c, o)
4343

4444
if c.Backend != "" {
4545
opts = append(opts, model.WithBackendString(c.Backend))

Diff for: core/backend/options.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import (
1111
"github.com/rs/zerolog/log"
1212
)
1313

14-
func ModelOptions(c config.BackendConfig, so *config.ApplicationConfig, opts []model.Option) []model.Option {
14+
func ModelOptions(c config.BackendConfig, so *config.ApplicationConfig, opts ...model.Option) []model.Option {
1515
name := c.Name
1616
if name == "" {
1717
name = c.Model

Diff for: core/backend/rerank.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import (
1111

1212
func Rerank(modelFile string, request *proto.RerankRequest, loader *model.ModelLoader, appConfig *config.ApplicationConfig, backendConfig config.BackendConfig) (*proto.RerankResult, error) {
1313

14-
opts := ModelOptions(backendConfig, appConfig, []model.Option{model.WithModel(modelFile)})
14+
opts := ModelOptions(backendConfig, appConfig, model.WithModel(modelFile))
1515
rerankModel, err := loader.BackendLoader(opts...)
1616
if err != nil {
1717
return nil, err

Diff for: core/backend/soundgeneration.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ func SoundGeneration(
2525
backendConfig config.BackendConfig,
2626
) (string, *proto.Result, error) {
2727

28-
opts := ModelOptions(backendConfig, appConfig, []model.Option{model.WithModel(modelFile)})
28+
opts := ModelOptions(backendConfig, appConfig, model.WithModel(modelFile))
2929

3030
soundGenModel, err := loader.BackendLoader(opts...)
3131
if err != nil {

Diff for: core/backend/token_metrics.go

+1-3
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,7 @@ func TokenMetrics(
1515
appConfig *config.ApplicationConfig,
1616
backendConfig config.BackendConfig) (*proto.MetricsResponse, error) {
1717

18-
opts := ModelOptions(backendConfig, appConfig, []model.Option{
19-
model.WithModel(modelFile),
20-
})
18+
opts := ModelOptions(backendConfig, appConfig, model.WithModel(modelFile))
2119
model, err := loader.BackendLoader(opts...)
2220
if err != nil {
2321
return nil, err

Diff for: core/backend/tokenize.go

+1-3
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,7 @@ func ModelTokenize(s string, loader *model.ModelLoader, backendConfig config.Bac
1414
var inferenceModel grpc.Backend
1515
var err error
1616

17-
opts := ModelOptions(backendConfig, appConfig, []model.Option{
18-
model.WithModel(modelFile),
19-
})
17+
opts := ModelOptions(backendConfig, appConfig, model.WithModel(modelFile))
2018

2119
if backendConfig.Backend == "" {
2220
inferenceModel, err = loader.GreedyLoader(opts...)

Diff for: core/backend/transcript.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ func ModelTranscription(audio, language string, translate bool, ml *model.ModelL
1818
backendConfig.Backend = model.WhisperBackend
1919
}
2020

21-
opts := ModelOptions(backendConfig, appConfig, []model.Option{})
21+
opts := ModelOptions(backendConfig, appConfig)
2222

2323
transcriptionModel, err := ml.BackendLoader(opts...)
2424
if err != nil {

Diff for: core/backend/tts.go

+1-5
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,7 @@ func ModelTTS(
2828
bb = model.PiperBackend
2929
}
3030

31-
opts := ModelOptions(backendConfig, appConfig, []model.Option{
32-
model.WithBackendString(bb),
33-
model.WithModel(modelFile),
34-
})
35-
31+
opts := ModelOptions(backendConfig, appConfig, model.WithBackendString(bb), model.WithModel(modelFile))
3632
ttsModel, err := loader.BackendLoader(opts...)
3733
if err != nil {
3834
return "", nil, err

Diff for: core/http/endpoints/openai/realtime.go

+2-4
Original file line numberDiff line numberDiff line change
@@ -129,10 +129,8 @@ func newModel(cl *config.BackendConfigLoader, ml *model.ModelLoader, appConfig *
129129

130130
if cfg.Pipeline.LLM == "" || cfg.Pipeline.TTS == "" || cfg.Pipeline.Transcription == "" {
131131
// If we don't have Wrapped model definitions, just return a standard model
132-
opts := backend.ModelOptions(*cfg, appConfig, []model.Option{
133-
model.WithBackendString(cfg.Backend),
134-
model.WithModel(cfg.Model),
135-
})
132+
opts := backend.ModelOptions(*cfg, appConfig, model.WithBackendString(cfg.Backend),
133+
model.WithModel(cfg.Model))
136134
return ml.BackendLoader(opts...)
137135
}
138136

Diff for: core/startup/startup.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ func Startup(opts ...config.AppOption) (*config.BackendConfigLoader, *model.Mode
160160

161161
log.Debug().Msgf("Auto loading model %s into memory from file: %s", m, cfg.Model)
162162

163-
o := backend.ModelOptions(*cfg, options, []model.Option{})
163+
o := backend.ModelOptions(*cfg, options)
164164

165165
var backendErr error
166166
if cfg.Backend != "" {

0 commit comments

Comments
 (0)