mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-21 00:21:49 +00:00
feat(mcp): add MCPToolset Python types
This commit is contained in:
parent
ef5b7bc62d
commit
5f82916fbb
1 changed files with 34 additions and 0 deletions
34
litellm/types/mcp_server/mcp_toolset.py
Normal file
34
litellm/types/mcp_server/mcp_toolset.py
Normal file
|
|
@ -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
|
||||
Loading…
Add table
Reference in a new issue