add tools api with a stub provider impl

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.common.content_types import URL
 
 from llama_stack.apis.common.type_system import ParamType
@@ -45,6 +45,8 @@ async def register_object_with_provider(obj: RoutableObject, p: Any) -> Routable
         return await p.register_scoring_function(obj)
     elif api == Api.eval:
         return await p.register_eval_task(obj)
+    elif api == Api.tool_runtime:
+        return await p.register_tool(obj)
     else:
         raise ValueError(f"Unknown API {api} for registering object with provider")
 
@@ -57,6 +59,8 @@ async def unregister_object_from_provider(obj: RoutableObject, p: Any) -> None:
         return await p.unregister_model(obj.identifier)
     elif api == Api.datasetio:
         return await p.unregister_dataset(obj.identifier)
+    elif api == Api.tool_runtime:
+        return await p.unregister_tool(obj.identifier)
     else:
         raise ValueError(f"Unregister not supported for {api}")
 
@@ -461,3 +465,61 @@ class EvalTasksRoutingTable(CommonRoutingTableImpl, EvalTasks):
             provider_resource_id=provider_eval_task_id,
         )
         await self.register_object(eval_task)
+
+
+class ToolsRoutingTable(CommonRoutingTableImpl, Tools):
+    async def list_tools(self) -> List[Tool]:
+        return await self.get_all_with_type("tool")
+
+    async def get_tool(self, tool_id: str) -> Tool:
+        return await self.get_object_by_identifier("tool", tool_id)
+
+    async def register_tool_group(
+        self,
+        name: str,
+        tool_group: ToolGroup,
+        provider_id: Optional[str] = None,
+    ) -> None:
+        tools = []
+        if isinstance(tool_group, MCPToolGroup):
+            # TODO: first needs to be resolved to corresponding tools available in the MCP server
+            raise NotImplementedError("MCP tool provider not implemented yet")
+        elif isinstance(tool_group, UserDefinedToolGroup):
+            for tool in tool_group.tools:
+
+                tools.append(
+                    Tool(
+                        identifier=tool.name,
+                        tool_group=name,
+                        name=tool.name,
+                        description=tool.description,
+                        parameters=tool.parameters,
+                        provider_id=provider_id,
+                        tool_prompt_format=tool.tool_prompt_format,
+                        provider_resource_id=tool.name,
+                    )
+                )
+        else:
+            raise ValueError(f"Unknown tool group: {tool_group}")
+
+        for tool in tools:
+            existing_tool = await self.get_tool(tool.name)
+            # Compare existing and new object if one exists
+            if existing_tool:
+                # Compare all fields except provider_id since that might be None in new obj
+                if tool.provider_id is None:
+                    tool.provider_id = existing_tool.provider_id
+                existing_dict = existing_tool.model_dump()
+                new_dict = tool.model_dump()
+
+                if existing_dict != new_dict:
+                    raise ValueError(
+                        f"Object {tool.name} already exists in registry. Please use a different identifier."
+                    )
+            await self.register_object(tool)
+
+    async def unregister_tool(self, tool_id: str) -> None:
+        tool = await self.get_tool(tool_id)
+        if tool is None:
+            raise ValueError(f"Tool {tool_id} not found")
+        await self.unregister_object(tool)