/Ui ddlZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZej dZd Zd ed efd Zddd d ededzdedzd efdZddeded eefdZy)N) ToolMessage)BaseToolInjectedToolCallIdtool)CompiledStateGraph) InjectedStateToolNode)Command) Annotatedz\s+__handoff_destination agent_namereturncftjd|jjS)z8Normalize an agent name to be used inside the tool name._) WHITESPACE_REsubstriplower)r s s/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/langgraph_swarm/handoff.py_normalize_agent_namers'   S*"2"2"4 5 ; ; ==)name descriptionrrcdt|dd}t|dtttfdtt t fffd }ti|_|S)a7Create a tool that can handoff control to the requested agent. Args: agent_name: The name of the agent to handoff control to, i.e. the name of the agent node in the multi-agent graph. Agent names should be simple, clear and unique, preferably in snake_case, although you are only limited to the names accepted by LangGraph nodes as well as the tool names accepted by LLM providers (the tool name will look like this: `transfer_to_`). name: Optional name of the tool to use for the handoff. If not provided, the tool name will be `transfer_to_`. description: Optional description for the handoff tool. If not provided, the tool description will be `Ask agent for help`. transfer_to_z Ask agent 'z ' for help)rstate tool_call_idcrtd|}ttj|d|gzdS)NzSuccessfully transferred to )contentrrmessages)r active_agent)gotographupdate)rr PARENT)rr tool_messager rs rhandoff_to_agentz-create_handoff_tool..handoff_to_agent*sM #2:,?% .. %j 1\N BT^_  r) rrr dictrstrr METADATA_KEY_HANDOFF_DESTINATIONmetadata)r rrr's`` rcreate_handoff_toolr,s" |3J?@A#Js OO  # #EU" n%**I i *  # # * * ,E   == $)IT]])Z 67  s 8B)r7)relangchain_core.messagesrlangchain_core.toolsrrrlanggraph.graph.staterlanggraph.prebuiltrr langgraph.typesr typing_extensionsr compilerr*r)rr,listr8rrrCs /CC46#' 6" #: >c>c> ,04((!Dj(>ADj( (V$6Z^_bZcr