o i @sddlZddlmZddlmZddlmZmZzddlm Z Wne y1Z ze de dZ [ wwde j d eejfd d Zd eejd eeee jffd dZdejd e jfddZde je jBe jBd ejfddZdS)N)Sequence)Literal) exceptionsmessages)typesuPlease install the `mcp` package to use the MCP server, you can use the `mcp` optional group — `pip install "pydantic-ai-slim[mcp]"`paramsreturncCsg}g}|jr|tj|jdg}|jD]L}|j}|jdkrN|r-|tj|dg}t|tj r7|j }n tj t |j|jdg}|tj|dq|r[|tj|dg}|t|q|rn|tj|d|ry|tj|d|S)zKConvert from MCP create message request parameters to pydantic-ai messages.contentuser)parts)data media_type) systemPromptappendrSystemPromptPartr role ModelResponse isinstance mcp_types TextContenttext BinaryContentbase64 b64decodermimeTypeUserPromptPart ModelRequestmap_from_sampling_content)r pai_messages request_partsresponse_partsmsgr user_part_contentr%V/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/_mcp.pymap_from_mcp_paramss2   r'r c sBgdtddtjtjBtjBffdd }g}|D]}t|tjr|jdur-| |j|j D]`}t|tj r>| |j t|tj rt|j trV|dtjd|j d q0|j D]6}t|trk|dtjd|d qYt|tjr|jr|dtjd t|j|jd qYtd t|q0q|d t|qd|fS)zConvert from pydantic-ai messages to MCP sampling messages. Returns: A tuple containing the system prompt and a list of sampling messages. r)r assistantr cstj||ddS)Nrr )rrSamplingMessager) sampling_msgsr%r&add_msgAsz&map_from_pai_messages..add_msgNr rtyperimage)r/rrzUnsupported content type: r()rrr ImageContent AudioContentrrr instructionsrr rr rstrris_imagerrrdecoderNotImplementedErrorr/map_from_model_responsejoin)r r- system_prompt pai_messagepartchunkr%r+r&map_from_pai_messages9sD           r?model_responsecCsTg}|jD]}t|tjr||jqtdt|j dt j dd |dS)z2Convert from a model response to MCP text content.zUnexpected part type: z, expected TextPartrr1r.) r rrTextPartrr rUnexpectedModelBehaviorr/__name__rrr:)r@ text_partsr=r%r%r&r9hs   r9r cCs"t|tjr tj|jdStd)z9Convert from sampling content to a pydantic-ai text part.r z;Image and Audio responses in sampling are not yet supported)rrrrrArr8r r%r%r&rts r)rcollections.abcrtypingrr1rrmcprr ImportError _import_errorCreateMessageRequestParamslist ModelMessager'tupler5r*r?rrr9r2r3rArr%r%r%r&s,  &)/