)`indZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZddlmZe d e ZGd d eeeZGd deZGddeZeeezezZGddeZeezezZeeeefZd*dZd+dZ d+dZ! d,d-d%Z" d,d.d)Z#dS)/z&Elicitation utilities for MCP servers.) annotationsN)Sequence)GenericLiteralTypeVarUnionget_args get_origin) BaseModel) ServerSession) RequestIdElicitSchemaModelT)boundc,eZdZUdZdZded<ded<dS)AcceptedElicitationz)Result when user accepts the elicitation.acceptLiteral['accept']actionrdataN__name__ __module__ __qualname____doc__r__annotations__j/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/elicitation.pyrrs533 (F((((rrc"eZdZUdZdZded<dS)DeclinedElicitationz*Result when user declines the elicitation.declinezLiteral['decline']rNrrrrr r s)44!*F******rr c"eZdZUdZdZded<dS)CancelledElicitationz)Result when user cancels the elicitation.cancelzLiteral['cancel']rNrrrrr#r#s)33 (F((((((rr#c"eZdZUdZdZded<dS)AcceptedUrlElicitationz0Result when user accepts a URL mode elicitation.rrrNrrrrr&r&'s):: (F((((((rr&schematype[BaseModel]returnNonec|jD]W\}}|j}||tjurt |r-t |r=td|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.s r_validate_elicitation_schemar74s"("5";";"="= J*  u~!=!=   , ,    , ,  `Z``/``` rr.typeboolct|}|rX t|tr1t|}t |dko|dt uSn#t $rYnwxYwdS)zMCheck if annotation is a sequence of strings (list[str], Sequence[str], etc).rF)r issubclassrr lenstrr3r.originargss rr2r2Gs  # #F  &(++ 9 ++4yyA~8$q'S.8 9    D  5sAA A'&A'c|tvrdSt|}|tus|tjur(t |}t d|DSdS)zDCheck if a field is a primitive type allowed in elicitation schemas.Tc3bK|]*}|tjup|tvpt|V+dSN)r/r0r4r2).0args r z&_is_primitive_field..bsT  ilC5> ! dS,H%H dL_`cLdLd      rF)r4r rr/ UnionTyper allr?s rr1r1Ws{111t # #F &EO33 ##  pt       5rsessionr messager>type[ElicitSchemaModelT]related_request_idRequestId | None%ElicitationResult[ElicitSchemaModelT]cKt||}||||d{V}|jdkr1|j*||j}t |S|jdkrtS|jdkrtStd|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. )rKrequestedSchemarMNr)rr!r$Unexpected elicitation action: ) r7model_json_schema elicit_formrcontentmodel_validaterr r# ValueError)rJrKr'rM json_schemaresultvalidated_datas relicit_with_validationr[is"!(((**,,K&&#-'F }  V^%?..v~>>"7777 ) # #"$$$ ( " "#%%%J6=JJKKKrurlelicitation_idUrlElicitationResultcK|||||d{V}|jdkrtS|jdkrtS|jdkrt St d|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 )rKr\r]rMNrr!r$rR) elicit_urlrr&r r#rW)rJrKr\r]rMrYs rr`r`s<%% %- &F}  %''' ) # #"$$$ ( " "#%%%J6=JJKKKr)r'r(r)r*)r.r8r)r9rD) rJr rKr>r'rLrMrNr)rO) rJr rKr>r\r>r]r>rMrNr)r^)$r __future__rr/collections.abcrtypingrrrrr r pydanticr mcp.server.sessionr mcp.typesr rrr r#ElicitationResultr&r^r>intfloatr9r4r7r2r1r[r`rrrrjs2,,"""""" $$$$$$IIIIIIIIIIIIIIII,,,,,,W1CCC)W-?%@+++++)+++ )))))9))) ((:;>QQThh)))))Y))) .0CCFZZ!$S%6&     ,,0 %L%L%L%L%LZ,0 -L-L-L-L-L-L-Lr