Add memory management APIs for OpenViking: list, read, write, and delete memories
This commit is contained in:
@ -7,6 +7,7 @@ from pydantic import BaseModel, Field
|
||||
|
||||
|
||||
OperationStatus = Literal["success", "partial_success", "failed"]
|
||||
MemoryWriteMode = Literal["create", "replace", "append"]
|
||||
|
||||
|
||||
class MessageIngestRequest(BaseModel):
|
||||
@ -54,6 +55,15 @@ class ProfileRequest(BaseModel):
|
||||
level: int = Field(default=2, ge=0)
|
||||
|
||||
|
||||
class MemoryWriteRequest(BaseModel):
|
||||
user_id: str = Field(min_length=1)
|
||||
user_key: str = Field(min_length=1)
|
||||
uri: str = Field(min_length=1)
|
||||
content: str
|
||||
mode: MemoryWriteMode = "create"
|
||||
wait: bool = True
|
||||
|
||||
|
||||
class ResourceUploadRequest(BaseModel):
|
||||
user_id: str = Field(min_length=1)
|
||||
user_key: str = Field(min_length=1)
|
||||
@ -116,6 +126,12 @@ class ProfileResponse(BaseModel):
|
||||
backends: dict[str, BackendStatus]
|
||||
|
||||
|
||||
class MemoryOperationResponse(BaseModel):
|
||||
status: OperationStatus
|
||||
memory: Any = None
|
||||
backends: dict[str, BackendStatus]
|
||||
|
||||
|
||||
class ResourceMutationResponse(BaseModel):
|
||||
status: OperationStatus
|
||||
resource: Any = None
|
||||
|
||||
Reference in New Issue
Block a user