o )j-@sdZddlmZmZmZddlZddlmZddlm Z m Z ddl m Z e Z dZGdd 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) RequestErrorraise_for_http_status) get_loggerz/openapi/v1/mcp/serversc@eZdZdZdS) MCPApiErrorz"Base exception for MCP API errors.N__name__ __module__ __qualname____doc__rrW/var/www/html/Deteccion_Ine/venv/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 seZdZdZ  ddeedeeddffdd Zededefd d Z   ddeed ee ee fdee deede ee ff ddZ ddede ee ffddZ ddedeede 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. Nendpointtokenreturncs tj||d|jt|_dS)z Initialize MCP API. Args: endpoint: The modelscope server address. Defaults to None (uses default endpoint). token: Optional access token for Bearer authentication. )rrN)super__init__r MCP_API_PATH mcp_base_url)selfrr __class__rrr1s zMCPApi.__init__ server_idcCsd|vr |dddS|S)z#Extract server name from server ID./)split)r rrr_get_server_name_from_id?szMCPApi._get_server_name_from_idfilter total_countsearchc Cs|dus |dks |dkrtd|pid||d}z|j|dd}|jj|j||d}t|WntjjyL}zt d |t d ||d}~wwz| |} Wnt yh}ztd ||d}~ww| d g} d d| D} | dd| 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)r'Z page_numberZ page_sizer)FrZtoken_required)urlheadersjsonzFailed to get MCP servers: %szFailed to get MCP servers: z(Invalid response from MCP servers list: mcp_server_listcSs0g|]}|dd|dd|dddqS)namer&id description)r0r1r2)get).0itemrrr s    z+MCPApi.list_mcp_servers..r(rr(servers) ValueError_build_bearer_headerssessionputrrrequests exceptionsRequestExceptionloggererrorr_parse_openapi_responserrr3) rrr'r(r)bodyr-redatar/mcp_config_listrrrlist_mcp_serversFsJ+    zMCPApi.list_mcp_serversc Csr|jd}z|j|dd}|jj||d}t|Wntjjy9}zt d|t d||d}~wwt d|j z| |}Wnty^}ztd||d}~ww|d g}g}|D]F} i} | d d | d <| d d | d <| d d | d <g| d<| dgD]} | d| dpd dd| dd dq|| qi|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 /operationalTr+)r-z'Failed to get operational MCP servers: NzResponse status code: z/Invalid response from operational MCP servers: r/r0r&r1r2Z mcp_serversoperational_urlsr,r!typer,r(rr7)rr:r;r3rr=r>r?r@rArdebug status_coderBrrappendr#) rrr,r-rDrErFr/rGr5 mcp_configoperational_urlrrrlist_operational_mcp_serverssX &     z#MCPApi.list_operational_mcp_serversc Csl|std|jd|}z|j|dd}|jj||ddid}t|Wn$tjjyJ}zt d|d |t d|d ||d }~wwz| |}Wnt yi}z td |d ||d }~ww|d d |dd |dd d}|dd }t|} |dg} g} | r| r| D]} | dpd dd| 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!Fr+Zget_operational_urlT)r-paramszFailed to get MCP server z: Nz!Invalid response from MCP server r0r&r2r1)r0r2r1rIr,rJrKr8)r9rr:r;r3rr=r>r?r@rArrBrrrr$r#rO)rr rr,r-rDrErFresultZ server_namerIrGrQrPrrrget_mcp_servers\)          zMCPApi.get_mcp_server)NN)NNr%r&)N)r r rrrstrr staticmethodr$rrintrHrRrU __classcell__rrrrr&sT   R  Q r)rtypingrrrr=Zmodelscope.hub.apirZmodelscope.hub.errorsrrZmodelscope.utils.loggerrr@r Exceptionr rrrrrrrs