o W+ i.@sdZddlmZmZmZddlZddlmZddlm Z ddl m Z e Z dZ Gdd d eZGd d d eZGd d d eZGdddeZdS)z MCP (Model Context Protocol) API interface for ModelScope Hub. This module provides a simple interface to interact with ModelScope MCP plaza (https://www.modelscope.cn/mcp). )AnyDictOptionalN)HubApi)raise_for_http_status) get_loggerz/openapi/v1/mcp/serversc@eZdZdZdS) MCPApiErrorz"Base exception for MCP API errors.N__name__ __module__ __qualname____doc__rrb/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/modelscope/hub/mcp_api.pyr r c@r)MCPApiRequestErrorz,Exception raised when MCP API request fails.Nr rrrrrrrc@r)MCPApiResponseErrorz2Exception raised when MCP API response is invalid.Nr rrrrr!rrc seZdZdZddeeddffdd Zedej de ee ffdd Z ed edefd d Z   ddeedee ee fdeedeede ee ff ddZ ddede ee ffddZ dd edeede ee ffddZZS)MCPApiaU MCP (Model Context Protocol) API interface class. This class provides interfaces to interact with ModelScope MCP servers, such as to list, deploy and manage MCP servers. Note: MCPApi inherits login() from HubApi for authentication. Different methods have different token requirements - see individual method docs. Nendpointreturncstj|d|jt|_dS)z Initialize MCP API. Args: endpoint: The modelscope server address. Defaults to None (uses default endpoint). )rN)super__init__r MCP_API_PATH mcp_base_url)selfr __class__rrr1szMCPApi.__init__rc Cshz|}Wn'tjjy-}ztd|td|jtd||d}~ww|diS)a  Handle HTTP response with unified error handling and JSON parsing. Args: r: requests Response object Returns: Parsed response data dict Raises: MCPApiResponseError: If JSON parsing fails zJSON parsing failed: zResponse content: zInvalid JSON response: Ndata) jsonrequests exceptionsJSONDecodeErrorloggererrortextrget)rresperrr_handle_response<s  zMCPApi._handle_response server_idcCsd|vr |dddS|S)z#Extract server name from server ID./)split)r+rrr_get_server_name_from_idSszMCPApi._get_server_name_from_idtokenfilter total_countsearchc Cs|dus |dks |dkrtd|pid||d}z||}|jj|j||j||d}t|Wntj j yO}zt d|t d||d}~ww||} | d g} d d | D} | d d | dS)a List available MCP servers, if (optional) token is presented, this would return private MCP servers as well. Args: token: Optional access token for authentication filter: Optional filters to apply to the search - 'category': str, server category, e.g. 'communication' - 'tag': str, server tag, e.g. 'social-media' - 'is_hosted': bool, server is hosted When all three are passed in, the intersection is taken. total_count: Number of servers to return, max 100, default 20 search: Optional search query string,e.g. Chinese service name, English service name, author/owner username You can combine `filter` and `search` to retrieve desired MCP servers. Returns: Dict containing: - total_count: Total number of servers - servers: List of server dictionaries with name, id, description Raises: MCPApiRequestError: If API request fails (network, server errors) MCPApiResponseError: If response format is invalid or JSON parsing fails Authentication: Optional, only required if you wish to retrieve private MCP servers. You may leverage the token parameter for one-time authentication, or use api.login() Returns: { 'total_count': 20, 'servers': [ {'name': 'ServerA', 'id': '@demo/ServerA', 'description': 'This is a demo server for xxx.'}, {'name': 'ServerB', 'id': '@demo/ServerB', 'description': 'This is another demo server.'}, ... ] } Nr-dz%total_count must be between 1 and 100)r3Z page_numberZ page_sizer5)urlheadersr cookieszFailed to get MCP servers: %szFailed to get MCP servers: mcp_server_listcSs0g|]}|dd|dd|dddqS)namer1id description)r;r<r=)r').0itemrrr s    z+MCPApi.list_mcp_servers..r4rr4servers) ValueError get_cookiessessionputrbuilder_headersr8rr!r"RequestExceptionr$r%rr*r') rr2r3r4r5bodyr9rr)rr:mcp_config_listrrrlist_mcp_serversZs:+       zMCPApi.list_mcp_serversc CsN|jd}||j}z|j|dd}|jj|||d}t|Wntjj y@}zt d|t d||d}~wwt d|j||}|dg}g} |D]D} i} | d d | d <| d d | d <| d d | d <g| d <| dgD]} | d | ddd| dd dq| | qY|dd| dS)aQ Get list of operational MCP servers that have been triggered hosting service by the user. Returns: Dict containing: - total_counts: Total number of operational servers - servers: List of server info with name, id, description Raises: MCPApiRequestError: If authentication fails or API request fails MCPApiResponseError: If response format is invalid or JSON parsing fails Returns: { 'total_count': 10, 'servers': [ { 'name': 'ServerA', "id": "@Group1/ServerA", 'description': 'This is a demo server for xxx.' 'mcp_servers': [ { 'type': 'sse', 'url': 'https://mcp.api-inference.modelscope.net/{uuid}/sse' }, { 'type': 'streamable_http', 'url': 'https://mcp.api-inference.modelscope.net/{uuid}/streamable_http' }, ... ] }, ... ] } z /operationalT)Z access_tokenZcookies_required)r8r9z'Failed to get operational MCP servers: NzResponse status code: r:r;r1r<r=Z mcp_serversoperational_urlsr7r,typer7r4rrA)rrGr8rDrEr'rr!r"rHr$r%rdebug status_coder*appendr.) rr2r7r8r9rr)rr:rJr? mcp_configoperational_urlrrrlist_operational_mcp_serverssH &       z#MCPApi.list_operational_mcp_serversc Cs>|std|jd|}||j}z||}|jj||ddi|d}t|Wn$tj j yO}zt d|d|t d|d||d}~ww||}|d d |d d |d d d } |d d }t|} |dg} g} | r| r| D]} | ddd| dd d}| |q| | d<| S)aU Get detailed information for a specific MCP Server, a valid token shall be provided if the MCP server is private. Args: server_id: MCP server ID (e.g., "@amap/amap-maps") token: Optional access token for authentication Returns: Dict containing: - name: Server name - description: Server description - id: Server ID - service_config: Connection configuration with type and url Raises: ValueError: If server_id is empty or None MCPApiRequestError: If API request fails or server not found MCPApiResponseError: If response format is invalid or JSON parsing fails Returns: { 'name': 'ServerA', 'description': 'This is a demo server for xxx.', 'id': '@demo/serverA', 'servers': [ { 'type': 'sse', 'url': 'https://mcp.api-inference.modelscope.net/{uuid}/sse' }, { 'type': 'streamable_http', 'url': 'https://mcp.api-inference.modelscope.net/{uuid}/streamable_http' } ... ] } zserver_id cannot be emptyr,Zget_operational_urlT)r8paramsr9zFailed to get MCP server z: Nr;r1r=r<)r;r=r<rLr7rMrNrB)rCrrGr8rDrEr'rr!r"rHr$r%rr*rr/r.rR)rr+r2r7r8r9rr)rresultZ server_namerLrJrTrSrrrget_mcp_serversR)            zMCPApi.get_mcp_server)N)NNr0r1)r r r rrstrr staticmethodr!Responserrr*r/intrKrUrX __classcell__rrrrr&sH    O  N r)rtypingrrrr!Zmodelscope.hub.apirZmodelscope.hub.errorsrZmodelscope.utils.loggerrr$r Exceptionr rrrrrrrs