phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

bug fix with routing tables

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-22 17:28:45 -07:00
parent 622f143d6b
commit bc394882a0
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -33,15 +33,13 @@ class CommonRoutingTableImpl(RoutingTable):
for p in self.providers.values(): for p in self.providers.values():
await p.shutdown() await p.shutdown()
async def get_provider_impl(self, routing_key: str) -> Optional[Any]: def get_provider_impl(self, routing_key: str) -> Optional[Any]:
return self.providers.get(routing_key) return self.providers.get(routing_key)
async def get_routing_keys(self) -> List[str]: def get_routing_keys(self) -> List[str]:
return self.routing_keys return self.routing_keys
async def get_provider_config( def get_provider_config(self, routing_key: str) -> Optional[GenericProviderConfig]:
self, routing_key: str
) -> Optional[GenericProviderConfig]:
for entry in self.routing_table_config.entries: for entry in self.routing_table_config.entries:
if entry.routing_key == routing_key: if entry.routing_key == routing_key:
return entry return entry