o h@sdZddlmZddlZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZed e d ZGd d d e eeZGd dde ZGddde ZeeeBeBZeeeefZd&ddZd'ddZ d(d)d$d%ZdS)*z&Elicitation utilities for MCP servers.) annotationsN)GenericLiteralTypeVarUnionget_args get_origin) BaseModel) FieldInfo) ServerSession) RequestIdElicitSchemaModelT)boundc@s&eZdZUdZdZded<ded<dS)AcceptedElicitationz)Result when user accepts the elicitation.acceptzLiteral['accept']actionr dataN__name__ __module__ __qualname____doc__r__annotations__rrW/var/www/html/openai_agents/venv/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']rNrrrrrrrrschematype[BaseModel]returnNonecCs6|jD]\}}t|std|dtdqdS)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)r! field_name field_inforrr_validate_elicitation_schema+sr,r+r boolcCsX|j}|tjur dS|tvrdSt|}|tus|tjur*t|}tdd|DSdS)zDCheck if a field is a primitive type allowed in elicitation schemas.Tcss"|] }|tjup |tvVqdSN)typesNoneTyper)).0argrrr Gs z&_is_primitive_field..F) annotationr/r0r)rr UnionTyperall)r+r4originargsrrrr'6s r'sessionr messagestrtype[ElicitSchemaModelT]related_request_idRequestId | None%ElicitationResult[ElicitSchemaModelT]cst||}|j|||dIdH}|jdkr)|jdur)||j}t|dS|jdkr1tS|jdkr9tSt d|j)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. )r:requestedSchemar=Nr)rrr zUnexpected elicitation action: ) r,model_json_schemaelicitrcontentmodel_validaterrr ValueError)r9r:r!r= json_schemaresultvalidated_datarrrelicit_with_validationLs      rI)r!r"r#r$)r+r r#r-r.) r9r r:r;r!r<r=r>r#r?) r __future__rr/typingrrrrrrpydanticr pydantic.fieldsr mcp.server.sessionr mcp.typesr r rrrElicitationResultr;intfloatr-r)r,r'rIrrrrs$