chore: get model also from query (#3716)

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Oct 2, 2024 · e5586e8 · e5586e8
1 parent 3acd767
commit e5586e8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/http/ctx/fiber.go b/core/http/ctx/fiber.go
@@ -19,7 +19,9 @@ func ModelFromContext(ctx *fiber.Ctx, cl *config.BackendConfigLoader, loader *mo
 	if ctx.Params("model") != "" {
 		modelInput = ctx.Params("model")
 	}
-
+	if ctx.Query("model") != "" {
+		modelInput = ctx.Query("model")
+	}
 	// Set model from bearer token, if available
 	bearer := strings.TrimLeft(ctx.Get("authorization"), "Bear ") // Reduced duplicate characters of Bearer
 	bearerExists := bearer != "" && loader.ExistsInModelPath(bearer)