uj-dZddlmZmZmZddlZddlmZddlm Z m Z ddl m Z e Z dZGdd eZGd d eZGd d eZGddeZdS)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/serversceZdZdZdS) MCPApiErrorz"Base exception for MCP API errors.N__name__ __module__ __qualname____doc__f/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/hub/mcp_api.pyr r s,,Drr ceZdZdZdS)MCPApiRequestErrorz,Exception raised when MCP API request fails.Nr rrrrrs66DrrceZdZdZdS)MCPApiResponseErrorz2Exception raised when MCP API response is invalid.Nr rrrrr!s<<Drrc <eZdZdZ ddeedeeddffd ZededefdZ ddeed ee ee fd ee d eede ee ff dZ ddede ee ffdZ ddedeede ee ffdZxZS)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. Nendpointtokenreturncvt|||jtz|_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__s rrzMCPApi.__init__1s7 (%888 ML8r server_idcFd|vr|dddS|S)z#Extract server name from server ID./)split)r$s r_get_server_name_from_idzMCPApi._get_server_name_from_id?s. )  ??3**1- -rfilter total_countsearchcd| |dks|dkrtd|pid||d} ||d}|j|j||}t |nJ#t jj$r3}t d |td ||d}~wwxYw | |} n%#t$r}td ||d}~wwxYw| d g} 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, page_number page_sizer.Frtoken_required)urlheadersjsonzFailed to get MCP servers: %szFailed to get MCP servers: z(Invalid response from MCP servers list: mcp_server_listcg|]D}|dd|dd|dddES)namer+id description)r:r;r<)get).0items r z+MCPApi.list_mcp_servers..sd'''HHVR((((4$$88M266  '''rr-rr-servers) ValueError_build_bearer_headerssessionputr!rrequests exceptionsRequestExceptionloggererrorr_parse_openapi_responserrr=) r"rr,r-r.bodyr6redatar8mcp_config_lists rlist_mcp_serverszMCPApi.list_mcp_serversFsV  +//[35F5FDEE El$    O00E133G   %wT!CCA !! $ $ $ $"3 O O O LL8! < < <$%F1%F%FGGQ N O G//22DD G G G%>1>>@@EF G G((#4b99''& ''' 88M155&   s0AA11B8.B33B8<C C4C//C4c |jd} ||d}|j||}t |nL#t jj$r5}t d|td||d}~wwxYwt d|j  | |}n%#t$r}td||d}~wwxYw|d g}g}|D]} i} | d d | d <| d d | d <| d d | d <g| d<| dgD]b} | d| dpd dd| dd dc|| |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 /operationalTr3)r6z'Failed to get operational MCP servers: NzResponse status code: z/Invalid response from operational MCP servers: r8r:r+r;r< mcp_serversoperational_urlsr5r&typer5r-rrA)r!rDrEr=rrGrHrIrJrKrdebug status_coderLrrappendr() r"rr5r6rNrOrPr8rQr? mcp_configoperational_urls rlist_operational_mcp_serversz#MCPApi.list_operational_mcp_serverssL"000 F00D122G   g 66A !! $ $ $ $"3 F F F LLF1FF G G G$=!==??DE F F  =am==>>> N//22DD N N N%E!EEGGLM N N((#4b99# / /DJ!%&"!5!5Jv #xxb11Jt (,(C(CJ} %(*J} %#'88,>#C#C  =)00,0077=2DDSII"M#''r2222  " ": . . . .88M155&   s0AAB#0BB>C C6C11C6c|std|jd|} ||d}|j||ddi}t |nR#t jj$r;}t d|d |td|d ||d }~wwxYw | |}n(#t$r}td |d ||d }~wwxYw|d d |dd |dd d}|dd }t|} |dg} g} | rc| ra| D]^} | dpd dd| dd d} | | _| |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&Fr3get_operational_urlT)r6paramszFailed to get MCP server z: Nz!Invalid response from MCP server r:r+r<r;)r:r<r;rUr5rVrWrB)rCr!rDrEr=rrGrHrIrJrKrrLrrrr)r(r[)r"r$rr5r6rNrOrPresult server_namerUrQr]r\s rget_mcp_serverzMCPApi.get_mcp_serversR :899 9"00Y00 E00E133G   W.CT-J!LLA !! $ $ $ $"3 E E E LLEYEE!EE F F F$>CD E E  M//22DD M M M%DIDDDDFFKL M M HHVR((88M266((4$$   HHT2&& 55i@@ 88$6;;  3+ 3#3 3 3,0077=2DDSII"M*..ub99  &&z2222+y s0AA%%B496B//B48C C3C..C3)NN)NNr*r+)N)r rrrrstrr staticmethodr)rrintrRr^rd __classcell__)r#s@rrr&s,0(, 9 9#C= 9   915 9 9 9 9 9 9CC\15<@6813 O O ( O !)$sCx.!9O '/smO "*# O 9=S#X O O O O d37M M ,/M ;?S>M M M M b/3SS"%S&smS7;CH~SSSSSSSSrr)rtypingrrrrGmodelscope.hub.apirmodelscope.hub.errorsrrmodelscope.utils.loggerr rJr Exceptionr rrrrrrrnsD '&&&&&&&&&%%%%%%EEEEEEEE...... )      )                +   SSSSSVSSSSSr