Function parse_element

src/codegraphcontext/viz/server.py:524–551

(val, nodes_dict, edges)

def parse_element(val, nodes_dict, edges):
    if val is None:
        return
    
    # If it is a Path
    if hasattr(val, 'nodes') and hasattr(val, 'relationships'):
        for n in val.nodes:
            parse_element(n, nodes_dict, edges)
        for r in val.relationships:
            parse_element(r, nodes_dict, edges)
        return
        
    type_name = type(val).__name__
    
    is_dict_rel = isinstance(val, dict) and (
        any(k in val for k in ('_src', '_dst', '_SRC', '_DST', 'src_node', 'dest_node'))
    )
    is_dict_node = isinstance(val, dict) and (
        '_label' in val or '_LABEL' in val or 'labels' in val
    ) and not is_dict_rel

    if type_name in ('Node', 'KuzuNode') or is_dict_node:
        parse_node(val, nodes_dict)
    elif type_name in ('Relationship', 'KuzuRelationship') or is_dict_rel:
        parse_rel(val, edges)
    elif isinstance(val, (list, tuple)):
        for item in val:
            parse_element(item, nodes_dict, edges)


class AIQueryRequest(BaseModel):