o i@s dZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZddlmZe d ed ZGd d d eeeZGd ddeZGdddeZeeeBeBZGdddeZeeBeBZeeeefZd/ddZd0ddZ d0ddZ! d1d2d(d)Z" d1d3d-d.Z#dS)4z&Elicitation utilities for MCP servers.) annotationsN)Sequence)GenericLiteralTypeVarUnionget_args get_origin) BaseModel) ServerSession) RequestIdElicitSchemaModelT)boundc@s&eZdZUdZdZded<ded<dS)AcceptedElicitationz)Result when user accepts the elicitation.acceptLiteral['accept']actionr dataN__name__ __module__ __qualname____doc__r__annotations__rrJ/home/ubuntu/.local/lib/python3.10/site-packages/mcp/server/elicitation.pyrs   rc@eZdZUdZdZded<dS)DeclinedElicitationz*Result when user declines the elicitation.declinezLiteral['decline']rNrrrrrr rc@r)CancelledElicitationz)Result when user cancels the elicitation.cancelzLiteral['cancel']rNrrrrrr rr c@r)AcceptedUrlElicitationz0Result when user accepts a URL mode elicitation.rrrNrrrrrr"'rr"schematype[BaseModel]returnNonecCsZ|jD]%\}}|j}|dus|tjurqt|rqt|r qtd|dtddS)zCValidate that a Pydantic model only contains primitive field types.NzElicitation schema field 'z' must be a primitive type zw, a sequence of strings (list[str], etc.), or Optional of these types. Nested models and complex types are not allowed.) model_fieldsitems annotationtypesNoneType_is_primitive_field_is_string_sequence TypeError_ELICITATION_PRIMITIVE_TYPES)r# field_name field_infor)rrr_validate_elicitation_schema4sr2r)typeboolcCsXt|}|r*zt|trt|}t|dko|dtuWSWdSty)YdSwdS)zMCheck if annotation is a sequence of strings (list[str], Sequence[str], etc).rF)r issubclassrrlenstrr.r)originargsrrrr-Gs  r-cCsD|tvrdSt|}|tus|tjur t|}tdd|DSdS)zDCheck if a field is a primitive type allowed in elicitation schemas.Tcss*|]}|tjup|tvpt|VqdSN)r*r+r/r-).0argrrr bs z&_is_primitive_field..F)r/r rr* UnionTyperallr9rrrr,Wsr,sessionr messager8type[ElicitSchemaModelT]related_request_idRequestId | None%ElicitationResult[ElicitSchemaModelT]cst||}|j|||dIdH}|jdkr)|jdur)||j}t|dS|jdkr1tS|jdkr9tSt d|j)a)Elicit information from the client/user with schema validation (form mode). This method can be used to interactively ask for additional information from the client within a tool's execution. The client might display the message to the user and collect a response according to the provided schema. Or in case a client is an agent, it might decide how to handle the elicitation -- either by asking the user or automatically generating a response. For sensitive data like credentials or OAuth flows, use elicit_url() instead. )rCrequestedSchemarENr)rrr!Unexpected elicitation action: ) r2model_json_schema elicit_formrcontentmodel_validaterrr ValueError)rBrCr#rE json_schemaresultvalidated_datarrrelicit_with_validationis      rRurlelicitation_idUrlElicitationResultcsZ|j||||dIdH}|jdkrtS|jdkrtS|jdkr%tStd|j)a[Elicit information from the user via out-of-band URL navigation (URL mode). This method directs the user to an external URL where sensitive interactions can occur without passing data through the MCP client. Use this for: - Collecting sensitive credentials (API keys, passwords) - OAuth authorization flows with third-party services - Payment and subscription flows - Any interaction where data should not pass through the LLM context The response indicates whether the user consented to navigate to the URL. The actual interaction happens out-of-band. When the elicitation completes, the server should send an ElicitCompleteNotification to notify the client. Args: session: The server session message: Human-readable explanation of why the interaction is needed url: The URL the user should navigate to elicitation_id: Unique identifier for tracking this elicitation related_request_id: Optional ID of the request that triggered this elicitation Returns: UrlElicitationResult indicating accept, decline, or cancel )rCrSrTrENrrr!rI) elicit_urlrr"rr rN)rBrCrSrTrErPrrrrVs   rV)r#r$r%r&)r)r3r%r4r<) rBr rCr8r#rDrErFr%rG) rBr rCr8rSr8rTr8rErFr%rU)$r __future__rr*collections.abcrtypingrrrrrr pydanticr mcp.server.sessionr mcp.typesr r rrr ElicitationResultr"rUr8intfloatr4r/r2r-r,rRrVrrrrs.             -