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 endpoints

Browse source
This commit is contained in:
Xi Yan 2024-09-22 23:59:53 -07:00
parent 1ac188e1b3
commit f467a0482c
5 changed files with 42 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/apis/memory/client.py
View file

@ -38,7 +38,7 @@ class MemoryClient(Memory):
async def get_memory_bank(self, bank_id: str) -> Optional[MemoryBank]:
async with httpx.AsyncClient() as client:
r = await client.get(
f"{self.base_url}/memory_banks/get",
f"{self.base_url}/memory/get",
params={
"bank_id": bank_id,
},
@ -59,7 +59,7 @@ class MemoryClient(Memory):
) -> MemoryBank:
async with httpx.AsyncClient() as client:
r = await client.post(
f"{self.base_url}/memory_banks/create",
f"{self.base_url}/memory/create",
json={
"name": name,
"config": config.dict(),
@ -81,7 +81,7 @@ class MemoryClient(Memory):
) -> None:
async with httpx.AsyncClient() as client:
r = await client.post(
f"{self.base_url}/memory_bank/insert",
f"{self.base_url}/memory/insert",
json={
"bank_id": bank_id,
"documents": [d.dict() for d in documents],
@ -99,7 +99,7 @@ class MemoryClient(Memory):
) -> QueryDocumentsResponse:
async with httpx.AsyncClient() as client:
r = await client.post(
f"{self.base_url}/memory_bank/query",
f"{self.base_url}/memory/query",
json={
"bank_id": bank_id,
"query": query,

18
llama_stack/apis/memory/memory.py
View file

@ -96,7 +96,7 @@ class MemoryBank(BaseModel):
class Memory(Protocol):
@webmethod(route="/memory_banks/create")
@webmethod(route="/memory/create")
async def create_memory_bank(
self,
name: str,
@ -104,13 +104,13 @@ class Memory(Protocol):
url: Optional[URL] = None,
) -> MemoryBank: ...
@webmethod(route="/memory_banks/list", method="GET")
@webmethod(route="/memory/list", method="GET")
async def list_memory_banks(self) -> List[MemoryBank]: ...
@webmethod(route="/memory_banks/get", method="GET")
@webmethod(route="/memory/get", method="GET")
async def get_memory_bank(self, bank_id: str) -> Optional[MemoryBank]: ...
@webmethod(route="/memory_banks/drop", method="DELETE")
@webmethod(route="/memory/drop", method="DELETE")
async def drop_memory_bank(
self,
bank_id: str,
@ -118,7 +118,7 @@ class Memory(Protocol):
# this will just block now until documents are inserted, but it should
# probably return a Job instance which can be polled for completion
@webmethod(route="/memory_bank/insert")
@webmethod(route="/memory/insert")
async def insert_documents(
self,
bank_id: str,
@ -126,14 +126,14 @@ class Memory(Protocol):
ttl_seconds: Optional[int] = None,
) -> None: ...
@webmethod(route="/memory_bank/update")
@webmethod(route="/memory/update")
async def update_documents(
self,
bank_id: str,
documents: List[MemoryBankDocument],
) -> None: ...
@webmethod(route="/memory_bank/query")
@webmethod(route="/memory/query")
async def query_documents(
self,
bank_id: str,
@ -141,14 +141,14 @@ class Memory(Protocol):
params: Optional[Dict[str, Any]] = None,
) -> QueryDocumentsResponse: ...
@webmethod(route="/memory_bank/documents/get", method="GET")
@webmethod(route="/memory/documents/get", method="GET")
async def get_documents(
self,
bank_id: str,
document_ids: List[str],
) -> List[MemoryBankDocument]: ...
@webmethod(route="/memory_bank/documents/delete", method="DELETE")
@webmethod(route="/memory/documents/delete", method="DELETE")
async def delete_documents(
self,
bank_id: str,