I have a function that finds the real id

I have a function that finds the real id of a node from a custom id I added in the metadata when loading the documents:

def get_real_node_id_from_custom_id(custom_id, index):
    for node in index.docstore.docs.values():
        if node.metadata.get('custom_id') == custom_id:
            return node.id_
    return None

This works fine when I use the default llamaindex vector store, but not when I use chromadb. The only difference in my code is how I create the storage context:

    if my_vector_store == "default":
        storage_context = StorageContext.from_defaults()
    elif my_vector_store == "chromadb":
        chroma_client = chromadb.PersistentClient(path="./storage")
        chroma_collection = chroma_client.create_collection("quickstart")
        vector_store = ChromaVectorStore(chroma_collection=chroma_collection)
        storage_context = StorageContext.from_defaults(vector_store=vector_store)

When using chromadb, it looks like index.docstore.docs is an empty {}.
How can I search my index for a specific metadata value when the vector store is chromadb?

Find answers from the community

I have a function that finds the real id