diff --git a/litellm/types/mcp_server/mcp_toolset.py b/litellm/types/mcp_server/mcp_toolset.py new file mode 100644 index 00000000000..7f78a22bfe2 --- /dev/null +++ b/litellm/types/mcp_server/mcp_toolset.py @@ -0,0 +1,34 @@ +from datetime import datetime +from typing import List, Optional + +from pydantic import BaseModel +from typing_extensions import TypedDict + + +class MCPToolsetTool(TypedDict): + server_id: str + tool_name: str + + +class MCPToolset(BaseModel): + toolset_id: str + toolset_name: str + description: Optional[str] = None + tools: List[MCPToolsetTool] = [] + created_at: Optional[datetime] = None + created_by: Optional[str] = None + updated_at: Optional[datetime] = None + updated_by: Optional[str] = None + + +class NewMCPToolsetRequest(BaseModel): + toolset_name: str + description: Optional[str] = None + tools: List[MCPToolsetTool] = [] + + +class UpdateMCPToolsetRequest(BaseModel): + toolset_id: str + toolset_name: Optional[str] = None + description: Optional[str] = None + tools: Optional[List[MCPToolsetTool]] = None