fix

Signed-off-by: Dave Lee <dave@gray101.com>

Author: dave-gray101

core/http/endpoints/openai/image.go

@@ -68,7 +68,6 @@ func downloadFile(url string) (string, error) {
 func ImageEndpoint(cl *config.BackendConfigLoader, ml *model.ModelLoader, appConfig *config.ApplicationConfig) func(c *fiber.Ctx) error {
 	return func(c *fiber.Ctx) error {
 		input, ok := c.Locals(middleware.CONTEXT_LOCALS_KEY_LOCALAI_REQUEST).(*schema.OpenAIRequest)
-		log.Warn().Str("input", fmt.Sprintf("%+v", *input)).Msg("TEMP IMAGE TRACE")
 		if !ok || input.Model == "" {
 			log.Error().Msg("Image Endpoint - Invalid Input")
 			return fiber.ErrBadRequest

core/http/routes/openai.go

@@ -109,6 +109,7 @@ func RegisterOpenAIRoutes(app *fiber.App,
 	// images
 	app.Post("/v1/images/generations",
 		re.BuildConstantDefaultModelNameMiddleware(model.StableDiffusionBackend),
+		re.SetModelAndConfig(func() schema.LocalAIRequest { return new(schema.OpenAIRequest) }),
 		re.SetOpenAIRequest,
 		openai.ImageEndpoint(cl, ml, appConfig))
 

tests/e2e-aio/e2e_test.go

@@ -235,7 +235,9 @@ var _ = Describe("E2E test", func() {
 				modelName := "jina-reranker-v1-base-en"
 
 				req := schema.JINARerankRequest{
-					Model: modelName,
+					BasicModelRequest: schema.BasicModelRequest{
+						Model: modelName,
+					},
 					Query: "Organic skincare products for sensitive skin",
 					Documents: []string{
 						"Eco-friendly kitchenware for modern homes",