phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 14:57:20 +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

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

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