refac

backend/main.py

@@ -314,10 +314,12 @@ async def get_all_models():
                     model["name"] = custom_model.name
                     model["info"] = custom_model.model_dump()
         else:
-
             owned_by = "openai"
             for model in models:
-                if custom_model.base_model_id == model["id"]:
+                if (
+                    custom_model.base_model_id == model["id"]
+                    or custom_model.base_model_id == model["id"].split(":")[0]
+                ):
                     owned_by = model["owned_by"]
                     break
 
@@ -329,6 +331,7 @@ async def get_all_models():
                     "created": custom_model.created_at,
                     "owned_by": owned_by,
                     "info": custom_model.model_dump(),
+                    "preset": True,
                 }
             )