iUdZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZej$dk\r ddl mZddl mZn ddlmZddlmZdd lmZd d lmZmZmZmZmZd d lm Z d d l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*d dl+m,Z,m-Z-d dl.m/Z/d dl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:d dl;mZ>d dl?m@Z@d dlAmAZAd dlmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZej0eZedZe Gdde eZ d'dededeeee fzdedzdeee geeee ffgee ff dZde deee ffd Zd!edeee ffd"Z d(ded#ed$eee dzdeQfd%Zgd&Zy))zClaude SDK for Python.N) AwaitableCallable) dataclass) AnnotatedAnyGenericTypeVarUnionget_args get_origin) )get_type_hints) is_typeddict)ToolAnnotations)ClaudeSDKErrorCLIConnectionErrorCLIJSONDecodeErrorCLINotFoundError ProcessError)import_session_to_store) ForkSessionResultdelete_sessiondelete_session_via_store fork_sessionfork_session_via_storerename_sessionrename_session_via_store tag_sessiontag_session_via_store)InMemorySessionStoreproject_key_for_directory)fold_session_summary) get_session_infoget_session_info_from_storeget_session_messagesget_session_messages_from_storeget_subagent_messages get_subagent_messages_from_store list_sessionslist_sessions_from_storelist_subagentslist_subagents_from_store) Transport) __version__)ClaudeSDKClient)query)VAgentDefinitionAssistantMessage BaseHookInput CanUseToolClaudeAgentOptions ContentBlockContextUsageCategoryContextUsageResponseDeferredToolUse HookCallback HookContextHookEventMessage HookInputHookJSONOutput HookMatcherMcpSdkServerConfigMcpServerConfigMcpServerConnectionStatus McpServerInfoMcpServerStatusMcpServerStatusConfigMcpStatusResponseMcpToolAnnotations McpToolInfoMessageMirrorErrorMessageNotificationHookInputNotificationHookSpecificOutputPermissionModePermissionRequestHookInput#PermissionRequestHookSpecificOutputPermissionResultPermissionResultAllowPermissionResultDenyPermissionUpdatePostToolUseFailureHookInput$PostToolUseFailureHookSpecificOutputPostToolUseHookInputPreCompactHookInputPreToolUseHookInputRateLimitEvent RateLimitInfoRateLimitStatus RateLimitType ResultMessageSandboxIgnoreViolationsSandboxNetworkConfigSandboxSettingsSdkBetaSdkPluginConfigSDKSessionInfoServerToolNameServerToolResultBlockServerToolUseBlock SessionKeySessionListSubkeysKeySessionMessage SessionStoreSessionStoreEntrySessionStoreFlushModeSessionStoreListEntrySessionSummaryEntry SettingSource StopHookInput StreamEventSubagentStartHookInputSubagentStartHookSpecificOutputSubagentStopHookInput SystemMessage TaskBudgetTaskNotificationMessageTaskNotificationStatusTaskProgressMessageTaskStartedMessage TaskUsage TextBlock ThinkingBlockThinkingConfigThinkingConfigAdaptiveThinkingConfigDisabledThinkingConfigEnabledToolPermissionContextToolResultBlock ToolUseBlock UserMessageUserPromptSubmitHookInputTceZdZUdZeed<eed<eeeee fzed<e ege eee ffed<dZ e dzed<y) SdkMcpToolzDefinition for an SDK MCP tool.name description input_schemahandlerN annotations)__name__ __module__ __qualname____doc__str__annotations__typerdictrrrrr`/volume1/homes/robertsu/coba/app/.venv/lib/python3.12/site-packages/claude_agent_sdk/__init__.pyrrsU) Iq'DcN** qc9T#s(^44 55*.K4'.rrrrrrreturncdttgttttffdt tffd }|S)a+ Decorator for defining MCP tools with type safety. Creates a tool that can be used with SDK MCP servers. The tool runs in-process within your Python application, providing better performance than external MCP servers. Args: name: Unique identifier for the tool. This is what Claude will use to reference the tool in function calls. description: Human-readable description of what the tool does. This helps Claude understand when to use the tool. input_schema: Schema defining the tool's input parameters. Can be either: - A dictionary mapping parameter names to types (e.g., {"text": str}) - A TypedDict class for more complex schemas - A JSON Schema dictionary for full validation Use ``Annotated[type, "description"]`` to add a description to a parameter in either dict-style or TypedDict schemas. Returns: A decorator function that wraps the tool implementation and returns an SdkMcpTool instance ready for use with create_sdk_mcp_server(). Example: Basic tool with simple schema: >>> @tool("greet", "Greet a user", {"name": str}) ... async def greet(args): ... return {"content": [{"type": "text", "text": f"Hello, {args['name']}!"}]} Tool with multiple parameters: >>> @tool("add", "Add two numbers", {"a": float, "b": float}) ... async def add_numbers(args): ... result = args["a"] + args["b"] ... return {"content": [{"type": "text", "text": f"Result: {result}"}]} Tool with error handling: >>> @tool("divide", "Divide two numbers", {"a": float, "b": float}) ... async def divide(args): ... if args["b"] == 0: ... return {"content": [{"type": "text", "text": "Error: Division by zero"}], "is_error": True} ... return {"content": [{"type": "text", "text": f"Result: {args['a'] / args['b']}"}]} Notes: - The tool function must be async (defined with async def) - The function receives a single dict argument with the input parameters - The function should return a dict with a "content" key containing the response - Errors can be indicated by including "is_error": True in the response rrc$t|S)N)rrrrr)r)rrrrrs r decoratorztool..decorators!#%#   r)rrrrrr)rrrrrs```` rtoolrsCn  3%4S>!::;  C    rpy_typec^t|}t|dddvrtt|dS|tur=t|}t|d}|ddD]}t |t s||d<|S|S|t urddiS|turdd iS|turdd iS|turdd iSt|d d}|tust |tjre|j}|Dcgc]}|tjus|}}t|dk(rt|dSd |Dcgc] }t|c}iS|t ur$t|dd}|rdt|ddSddiS|t"urddiS|t urddiS|t"urddiSt%|r t'|SddiScc}wcc}w)z7Convert a Python type annotation to a JSON Schema dict._nameN) NotRequiredRequiredReadOnlyrrrrstringintegernumberboolean __origin__anyOf__args__array)ritemsobject)r getattr_python_type_to_json_schemar r isinstancerintfloatboolr builtin_types UnionTyperNoneTypelenlistrr_typeddict_to_json_schema)roriginargsschemametaanon_none item_argss rrrs  Fvw%)PP*8G+F%%F*td_classct|d}i}|jD]\}}t|||<t|dt |j }d|d}|rt ||d<|S)z0Convert a TypedDict class to a JSON Schema dict.T)include_extras__required_keys__rr propertiesrequired)_get_type_hintsrrrsetkeyssorted)rhintsr field_name field_type required_keysrs rrr sz HT :E!#J"'++- J!>> @tool("add", "Add numbers", {"a": float, "b": float}) ... async def add(args): ... return {"content": [{"type": "text", "text": f"Sum: {args['a'] + args['b']}"}]} >>> >>> @tool("multiply", "Multiply numbers", {"a": float, "b": float}) ... async def multiply(args): ... return {"content": [{"type": "text", "text": f"Product: {args['a'] * args['b']}"}]} >>> >>> calculator = create_sdk_mcp_server( ... name="calculator", ... version="2.0.0", ... tools=[add, multiply] ... ) >>> >>> # Use with Claude >>> options = ClaudeAgentOptions( ... mcp_servers={"calc": calculator}, ... allowed_tools=["add", "multiply"] ... ) Server with application state access: >>> class DataStore: ... def __init__(self): ... self.items = [] ... >>> store = DataStore() >>> >>> @tool("add_item", "Add item to store", {"item": str}) ... async def add_item(args): ... store.items.append(args["item"]) ... return {"content": [{"type": "text", "text": f"Added: {args['item']}"}]} >>> >>> server = create_sdk_mcp_server("store", tools=[add_item]) Notes: - The server runs in the same process as your Python application - Tools have direct access to your application's variables and state - No subprocess or IPC overhead for tool calls - Server lifecycle is managed automatically by the SDK See Also: - tool(): Decorator for creating tool functions - ClaudeAgentOptions: Configuration for using servers with query() r)Server) AudioContentCallToolResultEmbeddedResource ImageContent ResourceLink TextContentTool)rtool_defrct|jtrd|jvr7d|jvr)t|jdtr |jSi}|jj D]\}}t |||<d|t |jdSt|jrt|jSdidS)Nrrr)rrrr) rrrrrrrrrr)rr param_name param_types r _build_schemaz,create_sdk_mcp_server.._build_schemas(//6h333$(=(=="8#8#8#@#F#000 .6.C.C.I.I.K*J -H-TJz*/L%", $Z__%6 7 H11201F1FGG$B7 7rzSdkMcpTool[Any]NcX|jyt|jdd}|yd|iS)NmaxResultSizeCharszanthropic/maxResultSizeChars)rr)rmax_sizes r _build_metaz*create_sdk_mcp_server.._build_metas< ##+x335I4PH2H= =r)rr inputSchemar_metacKSw)z#Return the list of available tools.r)cached_tool_listsr list_toolsz)create_sdk_mcp_server..list_toolss$ #sr argumentsc K|vrtd|d|}|j|d{}g}d|vrx|dD]o}|jd}|dk(r|jd|d7|dk(r!|jd|d |d  ]|d k(rg}|jd }|jd} |jd} |r|j|| r|jt | | r|j| |jd|rdj |nd|dk(rN|jdxsi} d| vr|jd| dCt jdZt jd|r ||jddS7w)z,Execute a tool by name with given arguments.zTool 'z ' not foundNcontentrtext)rrimagedatamimeType)rrr resource_linkrurir z Resource linkresourcez>Binary embedded resource cannot be converted to text, skippingz4Unsupported content type %r in tool result, skippingis_errorF)risError) ValueErrorrgetappendrjoinloggerwarning)rrrresultritem item_typeparts link_namerdescrrrrrrrtool_maps r call_toolz(create_sdk_mcp_server..call_tools8# 6${!;<<~H#++I66F F""9-D $ 0I F*{T&\'RS"g-(%,%)&\)-j)9#o5 "$(HHV$4 "hhuo#xx 6$!LL3!LLS2!LL.'%+9>TYYu%5O #j0#'88J#7#=2!X-#NN +hv>N O#NN `R%Q.Z"J)F q7s-GGFGsdk)rrinstance) mcp.serverr mcp.typesrrrrrrrrrrrrmodel_validaterrrrrrB)rrrrrserverrrrrrrrrrrrrrs @@@@@@@@rcreate_sdk_mcp_serverr 2sqP"D' *F "3 >S#X8M >,"  "   $MM#+#7#7#0#:#+#7#7(2  "       $$t* $  $    A #A $sCx.A SA A  A H 5tf EEyCF  s E AE )zr2r0r/r1rOrCrBrFrGrDrErHrIrJrr4rwr|r{ryrzr}r_r;r[r\r]r^rsrKr7rxr~rrrrrrrrfrhrgr8r9r:r6rrRrSrTrUr<r=r?r>r5rZrXrVrWrrrrvrYrMrtrPrNrurQr@rAr3rqrdr+r%r'r-r)rerkrirlrmrnrorprjr"r$rLr#rr,r&r(r.r*rr rrrrr!rrrcrbrar`r rrrrrrrr)N)z1.0.0N)rloggingsystypesrcollections.abcrr dataclassesrtypingrrrr r r r version_inforrrtyping_extensionsr r_errorsrrrrr_internal.session_importr_internal.session_mutationsrrrrrrrr r!_internal.session_storer"r#_internal.session_summaryr$_internal.sessionsr%r&r'r(r)r*r+r,r-r._internal.transportr/_versionr0clientr1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r^r_r`rarbrcrdrerfrgrhrirjrkrlrmrnrorprqrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrrrr getLoggerrrrrrrrrrrrr __all__rrrr!sD /!PPPw8#D.%>   U;   +!#WWWWWWWWWWWWWWWWWWWWWWv   8 $ CL // /+/ B BBc3h'B!4' B xyc3h889:JsOKL BJ33c3h3lc3h&NRUF UFUF.2:c?.Cd.JUFUFpK r