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

client fix

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-11 15:41:46 -08:00
parent 949b82fc6d
commit cf416cb6b8
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -100,12 +100,12 @@ async def run_main(host: str, port: int, stream: bool):
# register memory bank for the first time # register memory bank for the first time
response = await client.register_memory_bank( response = await client.register_memory_bank(
VectorMemoryBank( memory_bank_id="test_bank2",
identifier="test_bank2", params=VectorMemoryBankParams(
embedding_model="all-MiniLM-L6-v2", embedding_model="all-MiniLM-L6-v2",
chunk_size_in_tokens=512, chunk_size_in_tokens=512,
overlap_size_in_tokens=64, overlap_size_in_tokens=64,
) ),
) )
cprint(f"register_memory_bank response={response}", "blue") cprint(f"register_memory_bank response={response}", "blue")