Improve Workers AI model catalog search discoverability - #27188
Improve Workers AI model catalog search discoverability#27188mattrothenberg wants to merge 1 commit into
Conversation
| }, | ||
| "created_at": "2024-02-06 18:13:59.060", | ||
| "tags": [], | ||
| "tags": [ |
There was a problem hiding this comment.
If these aren't present on the upstream data source, they won't persist for the next update.
@thatsKevinJain, is that something you can help with?
There was a problem hiding this comment.
Sorry, I should have followed up here. We've made + merged some changes that enable adding these tags upstream and exposing them over the API.
There was a problem hiding this comment.
+1, the API is updated to surface these tags.
There was a problem hiding this comment.
The API is updated to surface the new tags, we should be good to merge this PR
|
@kodster28 the API has been updated to provide these |
|
Hey there, we've marked this pull request as stale because there's no recent activity on it. This label is helps us identify PRs that might need updates (or to be closed out by our team if no longer relevant). |
|
Hey there, we've closed out this pull request because it's been stale for a while and there's been no additional action on it. If these changes are still relevant, open a new pull request (or flag to us in a GitHub issue). |
Problem
A user reported they couldn't find moderation/safety models when searching for "moderation" in the model catalog. The search only checked model names, missing relevant models like
llama-guard-3-8bthat are designed for content moderation and safety classification.Solution
This PR improves search discoverability by:
["moderation", "safety", "content-filtering", "guardrails"]Screen.Recording.2025-12-16.at.8.46.57.PM.mov
Changes
src/components/ModelCatalog.tsx- Updated search filter logic to check name, description, and tagssrc/content/workers-ai-models/llama-guard-3-8b.json- Added moderation/safety tagssrc/content/workers-ai-models/llamaguard-7b-awq.json- Added moderation/safety tagsImpact
Users can now find safety/moderation models by searching:
Important
The model JSON files are generated by
bin/fetch-ai-models.jswhich fetches fromhttps://ai-cloudflare-com.pages.dev/api/models. The tags I added will be overwritten the next time this script runs.Should we:
Documentation checklist