phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

fix routing table key list

Browse source
This commit is contained in:
Xi Yan 2024-10-02 18:23:02 -07:00
parent 8d049000e3
commit 703ab9385f
2 changed files with 19 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

19
llama_stack/distribution/routers/routing_tables.py
View file

@ -94,12 +94,21 @@ class ShieldsRoutingTable(CommonRoutingTableImpl, Shields):
async def list_shields(self) -> List[ShieldSpec]:
specs = []
for entry in self.routing_table_config:
specs.append(
ShieldSpec(
shield_type=entry.routing_key,
provider_config=entry,
if isinstance(entry.routing_key, list):
for k in entry.routing_key:
specs.append(
ShieldSpec(
shield_type=k,
provider_config=entry,
)
)
else:
specs.append(
ShieldSpec(
shield_type=entry.routing_key,
provider_config=entry,
)
)
)
return specs
async def get_shield(self, shield_type: str) -> Optional[ShieldSpec]: