From 744eb0888cdc0028d89d2b4a74f217ee8f99a0dc Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 16 Dec 2024 12:42:37 -0800 Subject: [PATCH] add tools api with a stub provider impl --- llama_stack/apis/tools/tools.py | 1 + llama_stack/distribution/routers/routing_tables.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/apis/tools/tools.py b/llama_stack/apis/tools/tools.py index 239375b11..58f1c1794 100644 --- a/llama_stack/apis/tools/tools.py +++ b/llama_stack/apis/tools/tools.py @@ -61,6 +61,7 @@ class Tools(Protocol): parameters: List[ToolParameter], returns: ToolReturn, provider_id: Optional[str] = None, + provider_id: Optional[str] = None, provider_resource_id: Optional[str] = None, provider_metadata: Optional[Dict[str, Any]] = None, tool_prompt_format: Optional[ToolPromptFormat] = None, diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index f076475a2..e8c6f620a 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -15,7 +15,7 @@ from llama_stack.apis.shields import * # noqa: F403 from llama_stack.apis.memory_banks import * # noqa: F403 from llama_stack.apis.datasets import * # noqa: F403 from llama_stack.apis.eval_tasks import * # noqa: F403 -from llama_stack.apis.tools import * # noqa: F403 +from llama_stack.apis.tools import * # noqa: F403from llama_stack.apis.tools import * # noqa: F403 from llama_stack.apis.common.content_types import URL from llama_stack.apis.common.type_system import ParamType