scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix: format auto_router.py to pass black linter check

Browse source
This commit is contained in:
David Schneidhoffer 2026-03-20 21:58:54 +01:00
parent 533ec88420
commit 5877de8ef7
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/router_strategy/auto_router/auto_router.py
View file

@ -1,6 +1,7 @@
"""
Auto-Routing Strategy that works with a Semantic Router Config
"""
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
from litellm._logging import verbose_router_logger
@ -123,10 +124,16 @@ class AutoRouter(CustomLogger):
raw_content = user_message.get("content", "")
# Handle Anthropic-style content blocks: [{"type": "text", "text": "..."}]
if isinstance(raw_content, list):
message_content = " ".join(block.get("text", "") for block in raw_content if isinstance(block, dict) and block.get("text", ""))
message_content = " ".join(
block.get("text", "")
for block in raw_content
if isinstance(block, dict) and block.get("text", "")
)
else:
message_content = str(raw_content)
route_choice: Optional[Union[RouteChoice, List[RouteChoice]]] = self.routelayer(text=message_content)
route_choice: Optional[Union[RouteChoice, List[RouteChoice]]] = self.routelayer(
text=message_content
)
verbose_router_logger.debug(f"route_choice: {route_choice}")
if isinstance(route_choice, RouteChoice):
model = route_choice.name or self.default_model