address feedback

This commit is contained in:
Dinesh Yeduguru 2024-12-18 18:42:11 -08:00
parent 72dab3e4bf
commit ea0ca7454a
2 changed files with 17 additions and 9 deletions

View file

@ -6,7 +6,7 @@
from typing import Any, Dict
from llama_stack.apis.tools import InvokeToolResult, Tool, ToolRuntime
from llama_stack.apis.tools import Tool, ToolInvocationResult, ToolRuntime
from llama_stack.providers.datatypes import ToolsProtocolPrivate
from .config import MetaReferenceToolRuntimeConfig
@ -26,5 +26,7 @@ class MetaReferenceToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
async def unregister_tool(self, tool_id: str) -> None:
pass
async def invoke_tool(self, tool_id: str, args: Dict[str, Any]) -> InvokeToolResult:
async def invoke_tool(
self, tool_id: str, args: Dict[str, Any]
) -> ToolInvocationResult:
pass