from typing import Dict, List, Any, Optional
import unreal
import json
def search_assets(
search_term: str, asset_class: Optional[str] = None
) -> Dict[str, Any]:
asset_registry = unreal.AssetRegistryHelpers.get_asset_registry()
all_assets = asset_registry.get_all_assets()
matching_assets = []
search_term_lower = search_term.lower()
for asset in all_assets:
asset_name = str(asset.asset_name).lower()
package_path = str(asset.package_path).lower()
asset_class_name = str(asset.asset_class_path.asset_name).lower()
name_match = search_term_lower in asset_name
path_match = search_term_lower in package_path
class_match = True
if asset_class:
class_match = asset_class.lower() in asset_class_name
if (name_match or path_match) and class_match:
matching_assets.append(
{
"name": str(asset.asset_name),
"path": str(asset.package_path),
"class": str(asset.asset_class_path.asset_name),
"package_name": str(asset.package_name),
}
)
def relevance_score(asset):
name_exact = search_term_lower == asset["name"].lower()
name_starts = asset["name"].lower().startswith(search_term_lower)
return (name_exact * 3) + (name_starts * 2) + 1
matching_assets.sort(key=relevance_score, reverse=True)
return {
"search_term": search_term,
"asset_class_filter": asset_class,
"total_matches": len(matching_assets),
"assets": matching_assets[:50], # Limit to 50 results
}
def main():
result = search_assets("${search_term}", "${asset_class}")
print(json.dumps(result, indent=2))
if __name__ == "__main__":
main()
