Need Help on tool search

Hello everybody, on my ai agent system i build a tool search on BM25. i try also use Embedding gemma but with not best result. do you have any other idea? i try also Jev but this make system more slow and GPU consume. here my repo