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

added memory_bank endpoint

Browse source
This commit is contained in:
Raghotham Murthy 2024-07-11 00:14:28 -07:00
parent 86c2993296
commit 067ec4ce50
3 changed files with 380 additions and 93 deletions
Download patch file Download diff file Expand all files Collapse all files

32
source/api_definitions.py
View file

@ -270,29 +270,43 @@ class MemoryBanks(Protocol):
@webmethod(route="/memory_banks/get")
def get_memory_banks(
self,
self
) -> List[MemoryBank]: ...
@webmethod(route="/memory_banks/insert")
@webmethod(route="/memory_banks/drop")
def remove_memory_bank(
self,
bank_uuid: str,
) -> None: ...
@webmethod(route="/memory_bank/insert")
def post_insert_memory_documents(
self,
bank_uuid: str,
documents: List[MemoryBankDocument],
) -> None: ...
@webmethod(route="/memory_banks/delete")
@webmethod(route="/memory_bank/update")
def post_update_memory_documents(
self,
bank_uuid: str,
documents: List[MemoryBankDocument],
) -> None: ...
@webmethod(route="/memory_bank/get")
def post_get_memory_documents(
self,
bank_uuid: str,
document_uuids: List[str],
) -> List[MemoryBankDocument]: ...
@webmethod(route="/memory_bank/delete")
def post_delete_memory_documents(
self,
bank_uuid: str,
document_uuids: List[str],
) -> None: ...
@webmethod(route="/memory_banks/drop")
def remove_memory_bank(
self,
bank_uuid: str,
) -> None: ...
@dataclass
class KPromptGenerations: