phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-05 12:21:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update memory_banks method name to for openapi generator to work correctly without naming collision

Browse source
This commit is contained in:
Xi Yan 2024-09-23 10:29:28 -07:00
parent 2d7ce81302
commit 75357df02d
5 changed files with 229 additions and 103 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -25,7 +25,7 @@ class MemoryBanksClient(MemoryBanks):
async def shutdown(self) -> None:
pass
async def list_memory_banks(self) -> List[MemoryBankSpec]:
async def list_available_memory_banks(self) -> List[MemoryBankSpec]:
async with httpx.AsyncClient() as client:
response = await client.get(
f"{self.base_url}/memory_banks/list",
@ -34,7 +34,7 @@ class MemoryBanksClient(MemoryBanks):
response.raise_for_status()
return [MemoryBankSpec(**x) for x in response.json()]
async def get_memory_bank(
async def get_serving_memory_bank(
self, bank_type: MemoryBankType
) -> Optional[MemoryBankSpec]:
async with httpx.AsyncClient() as client:
@ -55,7 +55,7 @@ class MemoryBanksClient(MemoryBanks):
async def run_main(host: str, port: int, stream: bool):
client = MemoryBanksClient(f"http://{host}:{port}")
response = await client.list_memory_banks()
response = await client.list_available_memory_banks()
cprint(f"list_memory_banks response={response}", "green")