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

InterleavedTextAttachment -> InterleavedTextMedia, introduce memory tool

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-22 17:44:56 -07:00
parent 48c6a32edd
commit 31289e3f47
5 changed files with 56 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_toolchain/memory/api/endpoints.py
View file

@ -6,7 +6,7 @@
from typing import List, Protocol
from llama_models.llama3_1.api.datatypes import InterleavedTextAttachment
from llama_models.llama3.api.datatypes import InterleavedTextMedia
from llama_models.schema_utils import webmethod
from .datatypes import * # noqa: F403
@ -14,7 +14,7 @@ from .datatypes import * # noqa: F403
@json_schema_type
class RetrieveMemoryDocumentsRequest(BaseModel):
query: InterleavedTextAttachment
query: InterleavedTextMedia
bank_ids: str