~hifUdZddlmZmZmZmZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZeeeeeedZeeefed <Gd d e Zed k(rQdd lmZddl mZeedddge5Zeeeedj?ddddyy#1swYyxYw)zThis module implements the CrewAI adapter. CrewAI tools support only sync functions for their tools. Example Usage: >>> with MCPAdapt(StdioServerParameters(command="uv", args=["run", "src/echo.py"]), CrewAIAdapter()) as tools: >>> print(tools) )AnyCallable CoroutineTypeN)BaseTool) BaseModel) ToolAdapter)create_model_from_json_schemaresolve_refs_and_remove_defs)stringnumberintegerbooleanobjectarrayjson_type_mappingceZdZdZdeedzgejjfdejjde fdZ deedzge e e ejjffdejjde fdZy) CrewAIAdapteraAdapter for `crewai`. Note that `crewai` support only sync tools so we write adapt for sync tools only. Warning: if the mcp tool name is a python keyword, starts with digits or contains dashes, the tool name will be sanitized to become a valid python function name. funcNmcp_toolreturnctj_tjGfddt}|S)zAdapt a MCP tool to a CrewAI tool. Args: func: The function to adapt. mcp_tool: The MCP tool to adapt. Returns: A CrewAI tool. ceZdZUWjZeed<Wj xsdZeed<WZee ed<de de de ffd Z d Z y ) *CrewAIAdapter.adapt..CrewAIMCPToolname description args_schemaargskwargsrcji} jjdi}|jD]c\}}|W||vrS||}t|jdtrd|dvs8|||<>d|vsCt d|dDsY|||<_|||<e|j djS)N propertiestypenullanyOfc3DK|]}|jddk(yw)r#r$N)get).0opts S/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/mcpadapt/crewai_adapter.py zBCrewAIAdapter.adapt..CrewAIMCPTool._run..Ps%#$'!$6 9#s r) inputSchemar'items isinstancelistanycontenttext) selfrr filtered_kwargsschema_propertieskeyvalue prop_schemarrs r*_runz/CrewAIAdapter.adapt..CrewAIMCPTool._runAs24$,$8$8$<$<\2$N!"(,,.5JC}0A)A&7&< &koof&=tD%V)<<7< 4$ 3"#+6w+?# 8= 405,#5&O,44Q7<<<ctj|jjj Dcic] \}}|dk7r||}}}d|j d|d|j |_ycc}}w)Nz$defsz Tool Name: z Tool Arguments: z Tool Description: )jsonref replace_refsrmodel_json_schemar-rr)r3kvrs r*_generate_descriptionz@CrewAIAdapter.adapt..CrewAIMCPTool._generate_description[s!( 4 4((::<!eg 1G| qD &1 ;Mk]ZnosoooA$B sA9N) __name__ __module__ __qualname__rstr__annotations__rrrrrr9rA) ToolInputrrsr* CrewAIMCPToolr<sT D# %'339rK 9+4Ki 4 =# = = =4 Br:rH)r r,r r)r3rrrHrGs `` @r*adaptzCrewAIAdapter.adapt+sG &>>?:))..:  :xU yc399;S;S1S'TTUU))..U Ur:r__main__)StdioServerParameters)MCPAdaptuvrunz src/echo.py)commandrhello)r2) rNtypingrrrrr<rP crewai.toolsrpydanticr mcpadapt.corer mcpadapt.utils.modelingr r rEfloatintboolrOr/rrFrrBrVrWtoolsprintrYrTr:r*rfs21 !%   &4T ?KUKKU\ z)& d%1GH *  e  eAhlll() **  **s 9$B''B0