Merge pull request #179 from ag2ai/graphragfalkordbsetupfix

Setup.py > underscores to hyphens fix

.github/workflows/contrib-graph-rag-tests.yml

@@ -45,7 +45,7 @@ jobs:
           pip install pytest
       - name: Install FalkorDB SDK when on linux
         run: |
-          pip install -e .[graph_rag_falkor_db]
+          pip install -e .[graph-rag-falkor-db]
       - name: Set AUTOGEN_USE_DOCKER based on OS
         shell: bash
         run: |

notebook/agentchat_graph_rag_falkordb.ipynb

@@ -13,7 +13,7 @@
     "FalkorDB's GraphRAG-SDK is a dependency for this notebook, which can be installed with ag2 via pip:\n",
     "\n",
     "```bash\n",
-    "pip install ag2[graph_rag_falkor_db]\n",
+    "pip install ag2[graph-rag-falkor-db]\n",
     "```\n",
     "\n",
     "or if you have already installed ag2/autogen/pyautogen\n",

notebook/agentchat_swarm_graphrag_trip_planner.ipynb

@@ -25,7 +25,7 @@
     "FalkorDB's GraphRAG-SDK is a dependency for this notebook, which can be installed with ag2 via pip:\n",
     "\n",
     "```bash\n",
-    "pip install ag2[graph_rag_falkor_db]\n",
+    "pip install ag2[graph-rag-falkor-db]\n",
     "```\n",
     "\n",
     "For more information, please refer to the [installation guide](/docs/installation/).\n",

setup.py

@@ -76,6 +76,9 @@
 # pysqlite3-binary used so it doesn't need to compile pysqlite3
 autobuild = ["chromadb", "sentence-transformers", "huggingface-hub", "pysqlite3-binary"]
 
+# NOTE: underscores in pip install, e.g. pip install ag2[graph_rag_falkor_db], will automatically
+# convert to hyphens. So, do not include underscores in the name of extras.
+
 extra_require = {
     "test": [
         "ipykernel",
@@ -93,7 +96,7 @@
     "retrievechat-pgvector": retrieve_chat_pgvector,
     "retrievechat-mongodb": [*retrieve_chat, "pymongo>=4.0.0"],
     "retrievechat-qdrant": [*retrieve_chat, "qdrant_client", "fastembed>=0.3.1"],
-    "graph_rag_falkor_db": graph_rag_falkor_db,
+    "graph-rag-falkor-db": graph_rag_falkor_db,
     "autobuild": autobuild,
     "captainagent": autobuild + ["pandas"],
     "teachable": ["chromadb"],