phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use interleaved_text_media_as_str() utilityt

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-26 14:40:03 -07:00
parent e61b3d91ef
commit fb78bdc5a9
1 changed files with 1 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

13
llama_toolchain/memory/meta_reference/faiss/faiss.py
View file

@ -30,24 +30,13 @@ async def get_provider_impl(config: FaissImplConfig, _deps: Dict[Api, ProviderSp
return impl return impl
# This should be a broader utility
# This should support local file URLs and data URLs also
async def content_from_doc(doc: MemoryBankDocument) -> str: async def content_from_doc(doc: MemoryBankDocument) -> str:
if isinstance(doc.content, URL): if isinstance(doc.content, URL):
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
r = await client.get(doc.content.uri) r = await client.get(doc.content.uri)
return r.text return r.text
def _process(c): return interleaved_text_media_as_str(doc.content)
if isinstance(c, str):
return c
else:
return "<media>"
if isinstance(doc.content, list):
return " ".join([_process(c) for c in doc.content])
else:
return _process(doc.content)
def make_overlapped_chunks( def make_overlapped_chunks(