diff --git a/src/components/models/ModelBadges.tsx b/src/components/models/ModelBadges.tsx index 8b3c751dfebbc0..f259f1fb1fffd9 100644 --- a/src/components/models/ModelBadges.tsx +++ b/src/components/models/ModelBadges.tsx @@ -23,6 +23,20 @@ const ModelBadges = ({ model }: { model: WorkersAIModelsSchema }) => { }; } + if (property_id === "partner" && value === "true") { + return { + variant: "note", + text: "Partner", + }; + } + + if (property_id === "realtime" && value === "true") { + return { + variant: "note", + text: "Real-time", + }; + } + if (property_id === "planned_deprecation_date") { const timestamp = Math.floor(new Date(value as string).getTime() / 1000); diff --git a/src/components/models/ModelFeatures.tsx b/src/components/models/ModelFeatures.tsx index a5ffe34acd6602..3c7071fa96e78f 100644 --- a/src/components/models/ModelFeatures.tsx +++ b/src/components/models/ModelFeatures.tsx @@ -116,6 +116,18 @@ const ModelFeatures = ({ model }: { model: WorkersAIModelsSchema }) => {