Function process_cluster

_cluster_utils.py:404–447

( 
    use_llm_func, embeddings_func, \
    clusters,label,nodes,community_report_prompt,\
        relations,generate_relations,layer,temp_clusters_nodes
)

{edges_describe}
```"""
def process_cluster( 
    use_llm_func, embeddings_func, \
    clusters,label,nodes,community_report_prompt,\
        relations,generate_relations,layer,temp_clusters_nodes
):
    indices = [i for i, cluster in enumerate(clusters) if label in cluster]
                # Add the corresponding nodes to the node_clusters list
    cluster_nodes = [nodes[i] for i in indices]
    
    # Base case: if the cluster only has one node, do not attempt to recluster it
    logging.info(f"[Label{str(int(label))} Size: {len(cluster_nodes)}]")
    if len(cluster_nodes) == 1:
        cluster_nodes[0]['parent']=cluster_nodes[0]['entity_name']
        return {
        'community_data': None,
        'temp_node': cluster_nodes[0],
        'index':indices
        }
    name_set=[node['entity_name'] for node in cluster_nodes]
    cluster_intern_relation={**get_direct_relations(name_set,name_set,relations),
        **get_direct_relations(name_set,name_set,generate_relations)}
    describe=_pack_single_community_describe(cluster_nodes,cluster_intern_relation)
    hint_prompt=community_report_prompt.format(input_text=describe)
    response = use_llm_func(hint_prompt)
    data = convert_response_to_json(response)
    data['level'] = layer
    data['children'] = [n['entity_name'] for n in cluster_nodes]
    data['source_id'] = "|".join(set([n['source_id'] for n in cluster_nodes]))

    temp_node = {
        'entity_name': data['entity_name'],
        'description': data['entity_description'],
        'source_id': data['source_id'],
        'entity_type': "aggregate entity",
        'degree': 1,
        'vector': embeddings_func(data['entity_description']),
    }


    return {
        'community_data': data,
        'temp_node': temp_node,
        'index':indices
    }
def process_relation( 
    use_llm_func,community_report,maybe_edge,relations,generate_relations,\
     cluster_cluster_relation_prompt,layer,tokenizer,max_depth