phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update client

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-11 11:53:51 -08:00
parent 4369c2d0b6
commit 949b82fc6d
2 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/apis/memory/client.py
View file

@ -83,7 +83,13 @@ async def run_main(host: str, port: int, stream: bool):
overlap_size_in_tokens=64, overlap_size_in_tokens=64,
) )
await banks_client.register_memory_bank( await banks_client.register_memory_bank(
bank.identifier, bank.memory_bank_type, provider_resource_id=bank.identifier bank.identifier,
VectorMemoryBankParams(
embedding_model="all-MiniLM-L6-v2",
chunk_size_in_tokens=512,
overlap_size_in_tokens=64,
),
provider_resource_id=bank.identifier,
) )
retrieved_bank = await banks_client.get_memory_bank(bank.identifier) retrieved_bank = await banks_client.get_memory_bank(bank.identifier)

4
llama_stack/apis/memory_banks/client.py
View file

@ -58,7 +58,7 @@ class MemoryBanksClient(MemoryBanks):
async def register_memory_bank( async def register_memory_bank(
self, self,
memory_bank_id: str, memory_bank_id: str,
memory_bank_type: MemoryBankType, params: BankParams,
provider_resource_id: Optional[str] = None, provider_resource_id: Optional[str] = None,
provider_id: Optional[str] = None, provider_id: Optional[str] = None,
) -> None: ) -> None:
@ -67,9 +67,9 @@ class MemoryBanksClient(MemoryBanks):
f"{self.base_url}/memory_banks/register", f"{self.base_url}/memory_banks/register",
json={ json={
"memory_bank_id": memory_bank_id, "memory_bank_id": memory_bank_id,
"memory_bank_type": memory_bank_type.value,
"provider_resource_id": provider_resource_id, "provider_resource_id": provider_resource_id,
"provider_id": provider_id, "provider_id": provider_id,
"params": params.dict(),
}, },
headers={"Content-Type": "application/json"}, headers={"Content-Type": "application/json"},
) )