universal-mcp 0.1.8rc2__py3-none-any.whl → 0.1.8rc4__py3-none-any.whl

universal_mcp/tools/tools.py

@@ -4,27 +4,44 @@ import inspect
 from collections.abc import Callable
 from typing import Any, Literal
 
+import httpx
 from loguru import logger
 from pydantic import BaseModel, Field
 
+from universal_mcp.analytics import analytics
 from universal_mcp.applications.application import Application
-from universal_mcp.exceptions import ToolError
+from universal_mcp.exceptions import NotAuthorizedError, ToolError
 from universal_mcp.utils.docstring_parser import parse_docstring
 
 from .func_metadata import FuncMetadata
 
 
+def convert_tool_to_openai_tool(
+    tool: Tool,
+):
+    """Convert a Tool object to an OpenAI function."""
+    return {
+        "type": "function",
+        "function": {
+            "name": tool.name,
+            "description": tool.description,
+            "parameters": tool.parameters,
+        },
+    }
+
+
 def convert_tool_to_mcp_tool(
     tool: Tool,
 ):
     from mcp.server.fastmcp.server import MCPTool
+
     return MCPTool(
         name=tool.name,
         description=tool.description or "",
         inputSchema=tool.parameters,
     )
 
-# MODIFY THIS FUNCTION:
+
 def convert_tool_to_langchain_tool(
     tool: Tool,
 ):
@@ -44,7 +61,9 @@ def convert_tool_to_langchain_tool(
     )
 
     async def call_tool(
-        **arguments: dict[str, any], # arguments received here are validated by StructuredTool
+        **arguments: dict[
+            str, any
+        ],  # arguments received here are validated by StructuredTool
     ):
         # tool.run already handles validation via fn_metadata.call_fn_with_arg_validation
         # It should be able to handle the validated/coerced types from StructuredTool
@@ -52,31 +71,30 @@ def convert_tool_to_langchain_tool(
             call_tool_result = await tool.run(arguments)
             return call_tool_result
         except ToolError as e:
-             # Langchain expects ToolException for controlled errors
-             raise ToolException(f"Error running tool '{tool.name}': {e}") from e
+            # Langchain expects ToolException for controlled errors
+            raise ToolException(f"Error running tool '{tool.name}': {e}") from e
         except Exception as e:
             # Catch unexpected errors
             raise ToolException(f"Unexpected error in tool '{tool.name}': {e}") from e
 
-
     return StructuredTool(
         name=tool.name,
-        description=tool.description or f"Tool named {tool.name}.", # Provide fallback description
+        description=tool.description
+        or f"Tool named {tool.name}.",  # Provide fallback description
         coroutine=call_tool,
-        args_schema=tool.fn_metadata.arg_model, # <<< --- ADD THIS LINE
+        args_schema=tool.fn_metadata.arg_model,  # <<< --- ADD THIS LINE
         # handle_tool_error=True, # Optional: Consider adding error handling config
         # return_direct=False,    # Optional: Default is usually fine
         # response_format="content", # This field might not be valid for StructuredTool, check LangChain docs if needed. Let's remove for now.
     )
 
+
 class Tool(BaseModel):
     """Internal tool registration info."""
 
     fn: Callable[..., Any] = Field(exclude=True)
     name: str = Field(description="Name of the tool")
-    description: str = Field(
-        description="Summary line from the tool's docstring"
-    )
+    description: str = Field(description="Summary line from the tool's docstring")
     args_description: dict[str, str] = Field(
         default_factory=dict, description="Descriptions of arguments from the docstring"
     )
@@ -84,11 +102,12 @@ class Tool(BaseModel):
         default="", description="Description of the return value from the docstring"
     )
     raises_description: dict[str, str] = Field(
-        default_factory=dict, description="Descriptions of exceptions raised from the docstring"
+        default_factory=dict,
+        description="Descriptions of exceptions raised from the docstring",
     )
     tags: list[str] = Field(
         default_factory=list, description="Tags for categorizing the tool"
-        )
+    )
     parameters: dict[str, Any] = Field(description="JSON schema for tool parameters")
     fn_metadata: FuncMetadata = Field(
         description="Metadata about the function including a pydantic model for tool"
@@ -111,7 +130,7 @@ class Tool(BaseModel):
 
         raw_doc = inspect.getdoc(fn)
         parsed_doc = parse_docstring(raw_doc)
-        
+
         is_async = inspect.iscoroutinefunction(fn)
 
         func_arg_metadata = FuncMetadata.func_metadata(
@@ -135,19 +154,26 @@ class Tool(BaseModel):
     async def run(
         self,
         arguments: dict[str, Any],
-        context = None,
+        context=None,
     ) -> Any:
         """Run the tool with arguments."""
         try:
             return await self.fn_metadata.call_fn_with_arg_validation(
-                self.fn,
-                self.is_async,
-                arguments,
-                None
+                self.fn, self.is_async, arguments, None
             )
+        except NotAuthorizedError as e:
+            message = f"Not authorized to call tool {self.name}: {e.message}"
+            return message
+        except httpx.HTTPError as e:
+            message = f"HTTP error calling tool {self.name}: {str(e)}"
+            raise ToolError(message) from e
+        except ValueError as e:
+            message = f"Invalid arguments for tool {self.name}: {e}"
+            raise ToolError(message) from e
         except Exception as e:
             raise ToolError(f"Error executing tool {self.name}: {e}") from e
-        
+
+
 class ToolManager:
     """Manages FastMCP tools."""
 
@@ -159,17 +185,25 @@ class ToolManager:
         """Get tool by name."""
         return self._tools.get(name)
 
-    def list_tools(self, format: Literal["mcp", "langchain"] = "mcp") -> list[Tool]:
+    def list_tools(
+        self, format: Literal["mcp", "langchain", "openai"] = "mcp"
+    ) -> list[Tool]:
         """List all registered tools."""
         if format == "mcp":
             return [convert_tool_to_mcp_tool(tool) for tool in self._tools.values()]
         elif format == "langchain":
-            return [convert_tool_to_langchain_tool(tool) for tool in self._tools.values()]
+            return [
+                convert_tool_to_langchain_tool(tool) for tool in self._tools.values()
+            ]
+        elif format == "openai":
+            return [convert_tool_to_openai_tool(tool) for tool in self._tools.values()]
         else:
             raise ValueError(f"Invalid format: {format}")
 
     # Modified add_tool to accept name override explicitly
-    def add_tool(self, fn: Callable[..., Any] | Tool, name: str | None = None) -> Tool: # Changed any to Any
+    def add_tool(
+        self, fn: Callable[..., Any] | Tool, name: str | None = None
+    ) -> Tool:  # Changed any to Any
         """Add a tool to the server, allowing name override."""
         # Create the Tool object using the provided name if available
         tool = fn if isinstance(fn, Tool) else Tool.from_function(fn, name=name)
@@ -178,10 +212,14 @@ class ToolManager:
             if self.warn_on_duplicate_tools:
                 # Check if it's the *exact* same function object being added again
                 if existing.fn is not tool.fn:
-                     logger.warning(f"Tool name '{tool.name}' conflicts with an existing tool. Skipping addition of new function.")
+                    logger.warning(
+                        f"Tool name '{tool.name}' conflicts with an existing tool. Skipping addition of new function."
+                    )
                 else:
-                     logger.debug(f"Tool '{tool.name}' with the same function already exists.")
-            return existing # Return the existing tool if name conflicts
+                    logger.debug(
+                        f"Tool '{tool.name}' with the same function already exists."
+                    )
+            return existing  # Return the existing tool if name conflicts
 
         logger.debug(f"Adding tool: {tool.name}")
         self._tools[tool.name] = tool
@@ -190,39 +228,51 @@ class ToolManager:
     async def call_tool(
         self,
         name: str,
-        arguments: dict[str, Any], # Changed any to Any
-        context = None,
-    ) -> Any: # Changed any to Any
+        arguments: dict[str, Any],
+        context=None,
+    ) -> Any:
         """Call a tool by name with arguments."""
         tool = self.get_tool(name)
         if not tool:
             raise ToolError(f"Unknown tool: {name}")
-
-        return await tool.run(arguments)
+        try:
+            result = await tool.run(arguments)
+            analytics.track_tool_called(name, "success")
+            return result
+        except Exception as e:
+            analytics.track_tool_called(name, "error", str(e))
+            raise
 
     def get_tools_by_tags(self, tags: list[str]) -> list[Tool]:
         """Get tools by tags."""
-        return [tool for tool in self._tools.values() if any(tag in tool.tags for tag in tags)]
-    
+        return [
+            tool
+            for tool in self._tools.values()
+            if any(tag in tool.tags for tag in tags)
+        ]
+
     def register_tools_from_app(
         self,
         app: Application,
         tools: list[str] | None = None,
-        tags: list[str] | None = None
+        tags: list[str] | None = None,
     ) -> None:
-        
         try:
             available_tool_functions = app.list_tools()
         except TypeError as e:
-            logger.error(f"Error calling list_tools for app '{app.name}'. Does its list_tools method accept arguments? It shouldn't. Error: {e}")
+            logger.error(
+                f"Error calling list_tools for app '{app.name}'. Does its list_tools method accept arguments? It shouldn't. Error: {e}"
+            )
             return
         except Exception as e:
-             logger.error(f"Failed to get tool list from app '{app.name}': {e}")
-             return
+            logger.error(f"Failed to get tool list from app '{app.name}': {e}")
+            return
 
         if not isinstance(available_tool_functions, list):
-             logger.error(f"App '{app.name}' list_tools() did not return a list. Skipping registration.")
-             return
+            logger.error(
+                f"App '{app.name}' list_tools() did not return a list. Skipping registration."
+            )
+            return
 
         # Determine the effective filter lists *before* the loop for efficiency
         # Use an empty list if None is passed, simplifies checks later
@@ -230,13 +280,17 @@ class ToolManager:
 
         # For tags, determine the filter list based on priority: passed 'tags' or default 'important'
         # This list is only used if tools_name_filter is empty.
-        active_tags_filter = tags if tags else ["important"]# Default filter
+        active_tags_filter = tags if tags else ["important"]  # Default filter
 
-        logger.debug(f"Registering tools for '{app.name}'. Name filter: {tools_name_filter or 'None'}. Tag filter (if name filter empty): {active_tags_filter}")
+        logger.debug(
+            f"Registering tools for '{app.name}'. Name filter: {tools_name_filter or 'None'}. Tag filter (if name filter empty): {active_tags_filter}"
+        )
 
         for tool_func in available_tool_functions:
             if not callable(tool_func):
-                logger.warning(f"Item returned by {app.name}.list_tools() is not callable: {tool_func}. Skipping.")
+                logger.warning(
+                    f"Item returned by {app.name}.list_tools() is not callable: {tool_func}. Skipping."
+                )
                 continue
 
             try:
@@ -244,20 +298,22 @@ class ToolManager:
                 # This parses docstring (including tags), gets signature etc.
                 tool_instance = Tool.from_function(tool_func)
             except Exception as e:
-                 logger.error(f"Failed to create Tool object from function '{getattr(tool_func, '__name__', 'unknown')}' in app '{app.name}': {e}")
-                 continue # Skip this tool if metadata creation fails
+                logger.error(
+                    f"Failed to create Tool object from function '{getattr(tool_func, '__name__', 'unknown')}' in app '{app.name}': {e}"
+                )
+                continue  # Skip this tool if metadata creation fails
 
             # --- Modify the Tool instance before filtering/registration ---
             original_name = tool_instance.name
             prefixed_name = f"{app.name}_{original_name}"
-            tool_instance.name = prefixed_name # Update the name
+            tool_instance.name = prefixed_name  # Update the name
 
             # Add the app name itself as a tag for categorization
             if app.name not in tool_instance.tags:
-                 tool_instance.tags.append(app.name)
+                tool_instance.tags.append(app.name)
 
             # --- Filtering Logic ---
-            should_register = False # Default to not registering
+            should_register = False  # Default to not registering
 
             if tools_name_filter:
                 # --- Primary Filter: Check against specific tool names ---
@@ -272,14 +328,15 @@ class ToolManager:
                 # tool_instance.tags includes tags parsed from the docstring + app.name
                 if any(tag in tool_instance.tags for tag in active_tags_filter):
                     should_register = True
-                    logger.debug(f"Tool '{tool_instance.name}' matched tag filter {active_tags_filter}.")
+                    logger.debug(
+                        f"Tool '{tool_instance.name}' matched tag filter {active_tags_filter}."
+                    )
                 # else:
                 #     logger.debug(f"Tool '{tool_instance.name}' did NOT match tag filter {active_tags_filter}. Tool tags: {tool_instance.tags}")
 
-
             # --- Add the tool if it passed the filters ---
             if should_register:
                 # Pass the fully configured Tool *instance* to add_tool
                 self.add_tool(tool_instance)
             # else: If not registered, optionally log it for debugging:
-            #    logger.trace(f"Tool '{tool_instance.name}' skipped due to filters.") # Use trace level
+            #    logger.trace(f"Tool '{tool_instance.name}' skipped due to filters.") # Use trace level