jKj dZddlmZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZd d Z d d Z d dd Zy)zInject a required ``context`` parameter into a tool's JSON Schema so agents state their intent. Operates on the already-serialized JSON Schema dict (the ``mcp`` SDK exposes tool ``inputSchema`` as a plain dict).) annotationsN)AnyDictOptionalUnion)%DEFAULT_CONTEXT_PARAMETER_DESCRIPTION)log)MCPAnalyticsContextOptionsc |duS)NFcontexts p/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/mcp/_context_parameters.pyis_context_enabledrs % c<t|tr |jSy)N) isinstancer descriptionrs rget_context_descriptionrs'56""" rcz|}|r7t|jdtrd|dvrtd|d|S|rD|jds"|jds|jdrtd|d|S|sd igd }t j |}t|jdtsi|d<|jd d ur|j d d d|xstd|dd<|r>|jd}t|trd|vr|jd|Sdg|d<|S)uReturn a new JSON Schema dict with a ``context`` string property added. Returns the input unchanged (logging a warning) for schemas that already define ``context`` or use ``oneOf``/``allOf``/``anyOf``. ``required`` controls whether ``context`` is added to the schema's ``required`` list — pass ``False`` where the advertised schema is also used to validate inbound calls (the low-level server), so a call omitting ``context`` is not rejected. propertiesrz WARN: Tool "z>" already has 'context' parameter. Skipping context injection.oneOfallOfanyOfzE" has complex schema (oneOf/allOf/anyOf). Skipping context injection.object)typerrequiredadditionalPropertiesFNstring)rrr) rgetdictr copydeepcopypopr listappend) input_schema tool_namedescription_overriderschema required_lists radd_context_parameter_to_schemar-sUF  vzz,/ 6  - - I;&e f   6::g&&**W*=GAT 9+%j k  """E]]6 "F fjj. 5!|zz()U2 )40+T/T'F<#  :. mT * -$$Y/ M#,F:  Mr)r-Union[bool, MCPAnalyticsContextOptions, None]returnbool)rr.r/ Optional[str])unknownNT) r(Optional[Dict[str, Any]]r)strr*r1rr0r/r3)__doc__ __future__rr#typingrrrr constantsr loggerr typesr rrr-r rrr;sy ># --<-  :*. :*::(: :  :r