Question: How to embedding entities and relations, not only Chunk

[cwr31]

async def run_kg_builder():
    # 1. Build KG and Store in Neo4j Database
    kg_builder = SimpleKGPipeline(
        llm=ex_llm,
        driver=neo4j_driver,
        embedder=embedder,
        from_pdf=False,
    )
    await kg_builder.run_async(text='''Coffee can make you more alert and improve your concentration.''')

I use config above and then found that only Chunk node have property key named embedding, but entities(eg: Coffee) do not have, so those nodes will not show in verctor search.

[stellasia]

Hi @cwr31 ,

You're right that only the Chunk's text get embedded during the process. If you want to embed other properties, you'll have to create a separate process for that.

__Entity__ nodes can still be used in retrieval though, using VectorCypherRetriever for instance. This webinar gives some example queries you can use (notebook).