phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 06:53:47 +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,
)
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)