phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki 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

10
.pre-commit-config.yaml
View file

@ -52,8 +52,8 @@ repos:
# - id: pydoclint # - id: pydoclint
# args: [--config=pyproject.toml] # args: [--config=pyproject.toml]
- repo: https://github.com/tcort/markdown-link-check # - repo: https://github.com/tcort/markdown-link-check
rev: v3.11.2 # rev: v3.11.2
hooks: # hooks:
- id: markdown-link-check # - id: markdown-link-check
args: ['--quiet'] # args: ['--quiet']

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]: async def list_shields(self) -> List[ShieldSpec]:
specs = [] specs = []
for entry in self.routing_table_config: for entry in self.routing_table_config:
specs.append( if isinstance(entry.routing_key, list):
ShieldSpec( for k in entry.routing_key:
shield_type=entry.routing_key, specs.append(
provider_config=entry, ShieldSpec(
shield_type=k,
provider_config=entry,
)
)
else:
specs.append(
ShieldSpec(
shield_type=entry.routing_key,
provider_config=entry,
)
) )
)
return specs return specs
async def get_shield(self, shield_type: str) -> Optional[ShieldSpec]: async def get_shield(self, shield_type: str) -> Optional[ShieldSpec]: