gUddlmZddlZddlmZmZddZddZejeejeiZ de d<ddZ dd Z dd Zeje ej eiZd e d <dd ZddZy)) annotationsN)AnyCallablecJt|jd|jS)N.)get_namevalueattrnodes U/var/www/openai/venv/lib/python3.12/site-packages/_griffe/agents/nodes/assignments.py_get_attribute_namer s!tzz" #1TYYK 00c|jSN)idr s r _get_name_namer s 77Nrz dict[type, Callable[[Any], str]]_node_name_mapc0tt||S)zExtract name from an assignment node. Parameters: node: The node to extract names from. Returns: A list of names. )rtyper s r rrs $t* %d ++rcXd|jD}|Dcgc]}|s| c}Scc}w)Nc32K|]}t|ywr)r).0targets r z$_get_assign_names..$s 9L&Xf Ls)targets)r namesnames r _get_assign_namesr#s* 9DLL 9E" +UTdDU ++ +s''c:t|j}|r|gSgSr)rrr rs r _get_annassign_namesr"(s DKK DD6!r!rz&dict[type, Callable[[Any], list[str]]]_node_names_mapc0tt||S)zExtract names from an assignment node. Parameters: node: The node to extract names from. Returns: A list of names. )r#rr s r get_namesr%3s 4: &t ,,rct|Dcgc])}|jds|jddd+c}Scc}w)zExtract names from an assignment node, only for instance attributes. Parameters: node: The node to extract names from. Returns: A list of names. zself.r)r% startswithsplitr!s r get_instance_namesr*?s</8o ZodQXAYDJJsA q !o ZZ Zs??)r z ast.Attributereturnstr)r zast.Namer+r,)r ast.ASTr+r,)r z ast.Assignr+ list[str])r z ast.AnnAssignr+r.)r r-r+r.) __future__rasttypingrrrrName Attributer__annotations__rrr"Assign AnnAssignr#r%r*rr r8s}# 1 HHnMM&40 ,, " JJ!MM';7 - [r