From cf416cb6b876da9df88d78dba24a0ab1edaa7655 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 11 Nov 2024 15:41:46 -0800 Subject: [PATCH] client fix --- llama_stack/apis/memory_banks/client.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/apis/memory_banks/client.py b/llama_stack/apis/memory_banks/client.py index 8d137e43c..308ee42f4 100644 --- a/llama_stack/apis/memory_banks/client.py +++ b/llama_stack/apis/memory_banks/client.py @@ -100,12 +100,12 @@ async def run_main(host: str, port: int, stream: bool): # register memory bank for the first time response = await client.register_memory_bank( - VectorMemoryBank( - identifier="test_bank2", + memory_bank_id="test_bank2", + params=VectorMemoryBankParams( embedding_model="all-MiniLM-L6-v2", chunk_size_in_tokens=512, overlap_size_in_tokens=64, - ) + ), ) cprint(f"register_memory_bank response={response}", "blue")