mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
Update litellm/proxy/_experimental/mcp_server/semantic_tool_filter.py
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
parent
a14ee746cd
commit
3e80bb2abc
1 changed files with 5 additions and 12 deletions
|
|
@ -201,18 +201,11 @@ class SemanticMCPToolFilter:
|
||||||
]
|
]
|
||||||
|
|
||||||
# Reorder based on semantic router's ordering
|
# Reorder based on semantic router's ordering
|
||||||
ordered_filtered = []
|
name_to_tool = {tool.name: tool for tool in filtered}
|
||||||
for name in matched_names:
|
ordered_filtered = [
|
||||||
for tool in filtered:
|
name_to_tool[name] for name in matched_names
|
||||||
if tool.name == name:
|
if name in name_to_tool
|
||||||
ordered_filtered.append(tool)
|
]
|
||||||
break
|
|
||||||
|
|
||||||
verbose_logger.info(
|
|
||||||
f"Semantic filter: {len(available_tools)} -> {len(ordered_filtered)} tools. "
|
|
||||||
f"Top matches: {matched_names[:3]}"
|
|
||||||
)
|
|
||||||
|
|
||||||
return ordered_filtered if ordered_filtered else available_tools
|
return ordered_filtered if ordered_filtered else available_tools
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue