Views
No views yet
1from sentence_transformers import CrossEncoder
2
3model = CrossEncoder("tencent/R3-rerank-0.6b")
4query = "I need to compose music"
5skills = [ # The format is "name | description | skill_md"
6 "music-composer | Composes original music | Creates music for various media formats ...",
7 "music-lyricist | Writes lyrics for songs | Creates lyrics for various music genres ...",
8 "music-editor | Edits and mixes music tracks | Provides audio editing and mixing services ...",
9]
10pairs = [(query, skill) for skill in skills]
11scores = model.predict(pairs)
12print(scores)
13# [ 0.34937477 -1.7738094 -1.6604462 ]1@inproceedings{r3skill2026,
2 title = {Skill Is Not Document: A Query-Conditional Benchmark and Two-Stage Retriever for LLM Agent Skill Routing},
3 author = {Wang, Zifei and Wen, Wei and Ji, Qiang and Qiao, Ruizhi and Sun, Xing},
4 year = {2026},
5 url = {https://arxiv.org/abs/2606.03565},
6}