From 70fb70a71c9df5bdbdae871b329b628b8cd1b78e Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 23 Sep 2024 16:44:25 -0700 Subject: [PATCH] fix URL issue with agents --- .../providers/impls/meta_reference/agents/agent_instance.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/llama_stack/providers/impls/meta_reference/agents/agent_instance.py b/llama_stack/providers/impls/meta_reference/agents/agent_instance.py index 7d949603e..0ac26a857 100644 --- a/llama_stack/providers/impls/meta_reference/agents/agent_instance.py +++ b/llama_stack/providers/impls/meta_reference/agents/agent_instance.py @@ -7,6 +7,7 @@ import asyncio import copy import os +import re import secrets import shutil import string @@ -378,6 +379,11 @@ class ChatAgent(ShieldRunnerMixin): elif attachments and AgentTool.code_interpreter.value in enabled_tools: urls = [a.content for a in attachments if isinstance(a.content, URL)] + # TODO: we need to migrate URL away from str type + pattern = re.compile("^(https?://|file://|data:)") + urls += [ + URL(uri=a.content) for a in attachments if pattern.match(a.content) + ] msg = await attachment_message(self.tempdir, urls) input_messages.append(msg)