diff --git a/autogpt_platform/backend/backend/api/features/builder/db.py b/autogpt_platform/backend/backend/api/features/builder/db.py index 4a0eb62e8a..b4b612bb09 100644 --- a/autogpt_platform/backend/backend/api/features/builder/db.py +++ b/autogpt_platform/backend/backend/api/features/builder/db.py @@ -509,8 +509,10 @@ async def _get_static_counts(): def _matches_llm_model(schema_cls: type[BlockSchema], query: str) -> bool: for field in schema_cls.model_fields.values(): if field.annotation == LlmModel: + # Normalize query same as model slugs (hyphens to spaces) for matching + normalized_model_query = query.replace("-", " ") # Check if query matches any value in llm_models from registry - if any(query in name for name in _get_llm_models()): + if any(normalized_model_query in name for name in _get_llm_models()): return True return False