From 59af1c8fecb37877370f92a7b1178564c0c8965d Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 19 Sep 2024 13:35:03 -0700 Subject: [PATCH] fix memory url parsing (#81) --- llama_stack/providers/utils/memory/vector_store.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/llama_stack/providers/utils/memory/vector_store.py b/llama_stack/providers/utils/memory/vector_store.py index d575a985b..1e7a01b12 100644 --- a/llama_stack/providers/utils/memory/vector_store.py +++ b/llama_stack/providers/utils/memory/vector_store.py @@ -102,6 +102,15 @@ async def content_from_doc(doc: MemoryBankDocument) -> str: r = await client.get(doc.content.uri) return r.text + pattern = re.compile("^(https?://|file://|data:)") + if pattern.match(doc.content): + if doc.content.startswith("data:"): + return content_from_data(doc.content) + else: + async with httpx.AsyncClient() as client: + r = await client.get(doc.content) + return r.text + return interleaved_text_media_as_str(doc.content)