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__rrM/home/app/Keep/.python/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.) Z model_fieldsitems annotationtypesNoneType_is_primitive_field_is_string_sequence TypeError_ELICITATION_PRIMITIVE_TYPES)r# field_nameZ field_infor(rrr_validate_elicitation_schema4sr0r(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) UnionTyperallr7rrrr+Wsr+sessionr messager6type[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. )rAZrequestedSchemarCNr)rrr!Unexpected elicitation action: ) r0Zmodel_json_schemaZ elicit_formrcontentZmodel_validaterrr ValueError)r@rAr#rCZ json_schemaresultZvalidated_datarrrelicit_with_validationis      rJurlelicitation_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 )rArKrLrCNrrr!rF) elicit_urlrr"rr rH)r@rArKrLrCrIrrrrNs   rN)r#r$r%r&)r(r1r%r2r:) r@r rAr6r#rBrCrDr%rE) r@r rAr6rKr6rLr6rCrDr%rM)$r __future__rr)collections.abcrtypingrrrrrr Zpydanticr Zmcp.server.sessionr Z mcp.typesr r rrr ZElicitationResultr"rMr6intfloatr2r.r0r,r+rJrNrrrrs.             -