toolplanner-lm / src /tool_selector.py
hmnshudhmn24's picture
Upload 29 files
84613b6 verified
raw
history blame contribute delete
199 Bytes
import json
class ToolSelector:
def __init__(self,p='data/tools.json'): self.t=json.load(open(p))
def select(self,tokens): return [k for k,v in self.t.items() if any(w in tokens for w in v)]