phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 12:07:34 +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

10
llama_toolchain/agentic_system/meta_reference/agentic_system.py
View file

@ -16,9 +16,6 @@ from llama_toolchain.inference.api.datatypes import BuiltinTool
from llama_toolchain.memory.api import Memory
from llama_toolchain.safety.api import Safety
from llama_toolchain.agentic_system.api import * # noqa: F403
from .agent_instance import ChatAgent
from .config import MetaReferenceImplConfig
from llama_toolchain.tools.builtin import (
BraveSearchTool,
CodeInterpreterTool,
@ -27,7 +24,8 @@ from llama_toolchain.tools.builtin import (
)
from llama_toolchain.tools.safety import with_safety
from .agent_instance import AgentInstance
from .agent_instance import AgentInstance, ChatAgent
from .config import MetaReferenceImplConfig
logger = logging.getLogger()
@ -76,7 +74,6 @@ class MetaReferenceAgenticSystemImpl(AgenticSystem):
system_id = str(uuid.uuid4())
builtin_tools = []
custom_tool_definitions = []
cfg = request.agent_config
for dfn in cfg.available_tools:
if isinstance(dfn.tool_name, BuiltinTool):
@ -104,8 +101,6 @@ class MetaReferenceAgenticSystemImpl(AgenticSystem):
tool, self.safety_api, dfn.input_shields, dfn.output_shields
)
)
else:
custom_tool_definitions.append(dfn)
AGENT_INSTANCES_BY_ID[system_id] = ChatAgent(
agent_config=cfg,
@ -113,7 +108,6 @@ class MetaReferenceAgenticSystemImpl(AgenticSystem):
safety_api=self.safety_api,
memory_api=self.memory_api,
builtin_tools=builtin_tools,
custom_tool_definitions=custom_tool_definitions,
)
return AgenticSystemCreateResponse(