{hdZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZed e ZGd d e eeZGd de ZGdde ZeeezezZeeeefZddZddZ d ddZy)z&Elicitation utilities for MCP servers.) annotationsN)GenericLiteralTypeVarUnionget_args get_origin) BaseModel) FieldInfo) ServerSession) RequestIdElicitSchemaModelT)boundc*eZdZUdZdZded<ded<y)AcceptedElicitationz)Result when user accepts the elicitation.acceptzLiteral['accept']actionrdataN__name__ __module__ __qualname____doc__r__annotations__W/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/mcp/server/elicitation.pyrrs3 (F ( rrc eZdZUdZdZded<y)DeclinedElicitationz*Result when user declines the elicitation.declinezLiteral['decline']rNrrrrrrs4!*F *rrc eZdZUdZdZded<y)CancelledElicitationz)Result when user cancels the elicitation.cancelzLiteral['cancel']rNrrrrr"r"s3 (F (rr"c|jjD]&\}}t|rtd|dtdy)zCValidate that a Pydantic model only contains primitive field types.zElicitation schema field 'z' must be a primitive type z_ or Optional of these types. Complex types like lists, dicts, or nested models are not allowed.N) model_fieldsitems_is_primitive_field TypeError_ELICITATION_PRIMITIVE_TYPES)schema field_name field_infos r_validate_elicitation_schemar-+sV"("5"5";";"= J":.,ZL8S/01UV rc|j}|tjury|tvryt |}|t us|tj urt|}td|DSy)zDCheck if a field is a primitive type allowed in elicitation schemas.Tc3TK|] }|tjuxs|tv"ywN)typesNoneTyper)).0args r z&_is_primitive_field..Gs(`TW3%..(OC3O,OO`s&(F) annotationr1r2r)r r UnionTyperall)r,r6originargss rr'r'6si&&JU^^#11 #F &EOO3 #`[_``` rcKt||j}|j|||d{}|jdk(r3|jr'|j |j}t |S|jdk(r tS|jdk(r tStd|j7w)aElicit information from the client/user with schema validation. 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. )messagerequestedSchemarelated_request_idNr)rr r#zUnexpected elicitation action: ) r-model_json_schemaelicitrcontentmodel_validaterrr" ValueError)sessionr<r*r> json_schemaresultvalidated_datas relicit_with_validationrHLs!(**,K>>#-"F }} V^^..v~~>"77 ) #"$$ ( "#%%:6==/JKK!s3CCBC)r*ztype[BaseModel]returnNone)r,r rIboolr0) rDr r<strr*ztype[ElicitSchemaModelT]r>zRequestId | NonerIz%ElicitationResult[ElicitSchemaModelT]) r __future__rr1typingrrrrrr pydanticr pydantic.fieldsr mcp.server.sessionr mcp.typesr rrrr"ElicitationResultrLintfloatrKr)r-r'rHrrrrVs," II%,1C)W-?%@+)+ )9) ((:;>QQThh!$S%64,0 #L #L #L %#L) #L + #Lr