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

add model context protocol provider

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-19 15:45:32 -08:00
parent dc21e14f64
commit 33af4f919f
8 changed files with 160 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/providers/inline/tool_runtime/brave_search/brave_search.py
View file

@ -4,11 +4,11 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import Any, Dict
from typing import Any, Dict, List
import requests
from llama_stack.apis.tools import Tool, ToolInvocationResult, ToolRuntime
from llama_stack.apis.tools import Tool, ToolGroup, ToolInvocationResult, ToolRuntime
from llama_stack.distribution.request_headers import NeedsRequestProviderData
from llama_stack.providers.datatypes import ToolsProtocolPrivate
@ -42,6 +42,9 @@ class BraveSearchToolRuntimeImpl(
)
return provider_data.api_key
async def discover_tools(self, tool_group: ToolGroup) -> List[Tool]:
raise NotImplementedError("Brave search tool group not supported")
async def invoke_tool(
self, tool_id: str, args: Dict[str, Any]
) -> ToolInvocationResult: