phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

models API

Browse source
This commit is contained in:
Xi Yan 2024-09-21 17:27:19 -07:00
parent f0580251a3
commit 20a4302877
3 changed files with 81 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/distribution/routers/routers.py
View file

@ -47,7 +47,6 @@ class MemoryRouter(Memory):
config: MemoryBankConfig,
url: Optional[URL] = None,
) -> MemoryBank:
print("MemoryRouter: create_memory_bank")
bank_type = config.type
bank = await self.routing_table.get_provider_impl(
self.api, bank_type
@ -56,7 +55,6 @@ class MemoryRouter(Memory):
return bank
async def get_memory_bank(self, bank_id: str) -> Optional[MemoryBank]:
print("MemoryRouter: get_memory_bank")
return await self.get_provider_from_bank_id(bank_id).get_memory_bank(bank_id)
async def insert_documents(
@ -65,7 +63,6 @@ class MemoryRouter(Memory):
documents: List[MemoryBankDocument],
ttl_seconds: Optional[int] = None,
) -> None:
print("MemoryRouter: insert_documents")
return await self.get_provider_from_bank_id(bank_id).insert_documents(
bank_id, documents, ttl_seconds
)