Issues with NetworkX when using graph subindex

[arkim822]

Hello. Apologies for posting something like this but I am unable to figure out what I'm doing wrong when trying to use the subindex. I keep getting this error message:

ImportError: NetworkX is not available - install "graph" extra to enable

And I have tried to solve it by running these commands (Ubuntu/Linux environment)

sudo apt-get install graphviz libgraphviz-dev
pip install pygraphviz pydot lxml
pip install networkx
pip install networkx[extra]

This is the code I'm running

import txtai
embeddings = txtai.Embeddings(
    content=True,
    defaults=False,
    functions=[
        {"name": "graph", "function": "indexes.act.graph.attribute"}
    ],
     expressions=[
         {"name": "topic", "expression": "graph(indexid, 'topic')"},
     ],
    indexes={
        "dataset_name": {
            "keyword": True,
            "columns": {
                "text": "dataset_name"
            },
            "graph": {
                "topics": {}
            }
        },
        "zeno_link":{
            "columns": {
                "text": "zeno_link"
            }
        },
        "all_text":{
            "columns": {
                "text": "all_text"
            }
        }
    }
)
embeddings.index(alltext_dataset)

embeddings.search("select id, dataset_name, zeno_link, all_text, score from txtai where similar('Linux terminal')", 1)

txtai==6.3.0
networkx==3.2.1

[davidmezzetti]

What if you run:

pip install txtai[graph]

[arkim822]

Thank you @davidmezzetti !! That did it :D

Really appreciate the help. Would it be a good idea for me to add this to the subindex notebook?

[davidmezzetti]

What is the exact notebook again?

[arkim822]

One of them is:
https://github.com/neuml/txtai/blob/master/examples/46_Whats_new_in_txtai_6_0.ipynb

I can keep scanning for more and open a PR, as well.

[davidmezzetti]

Don't worry about it. I'll take a look and update as appropriate. Thank you for the heads up.

[davidmezzetti]

Colab removed some libraries that were installed previously causing graph notebooks to fail unless they install the graph extra. The graph extra should be installed when using graphs regardless.