feat: add datastax components bundle (#8184)

* feat: add datastax components bundle

* Update __init__.py

* Remove old astra assistants folder

* Remove old tools

* Update __init__.py

* Update test_assistants_components.py

---------

Co-authored-by: Edwin Jose <edwin.jose@datastax.com>

src/backend/base/langflow/components/datastax/__init__.py

@@ -1,4 +1,6 @@
 from .astra_assistant_manager import AstraAssistantManager
+from .astradb_cql import AstraDBCQLToolComponent
+from .astradb_tool import AstraDBToolComponent
 from .create_assistant import AssistantsCreateAssistant
 from .create_thread import AssistantsCreateThread
 from .dotenv import Dotenv
@@ -14,6 +16,8 @@ __all__ = [
     "AssistantsListAssistants",
     "AssistantsRun",
     "AstraAssistantManager",
+    "AstraDBCQLToolComponent",
+    "AstraDBToolComponent",
     "Dotenv",
     "GetEnvVar",
 ]

src/backend/base/langflow/components/tools/__init__.py

@@ -35,13 +35,9 @@ from .yahoo_finance import YfinanceToolComponent
 
 with warnings.catch_warnings():
     warnings.simplefilter("ignore", LangChainDeprecationWarning)
-    from .astradb import AstraDBToolComponent
-    from .astradb_cql import AstraDBCQLToolComponent
 
 __all__ = [
     "ArXivComponent",
-    "AstraDBCQLToolComponent",
-    "AstraDBToolComponent",
     "BingSearchAPIComponent",
     "CalculatorComponent",
     "CalculatorToolComponent",

src/backend/tests/integration/components/assistants/test_assistants_components.py

@@ -1,5 +1,5 @@
 import pytest
-from langflow.components.astra_assistants import (
+from langflow.components.datastax import (
     AssistantsCreateAssistant,
     AssistantsCreateThread,
     AssistantsGetAssistantName,

src/frontend/src/utils/styleUtils.ts

@@ -243,7 +243,7 @@ export const SIDEBAR_BUNDLES = [
   { display_name: "AssemblyAI", name: "assemblyai", icon: "AssemblyAI" },
   {
     display_name: "DataStax",
-    name: "astra_assistants",
+    name: "datastax",
     icon: "AstraDB",
   },
   { display_name: "Olivya", name: "olivya", icon: "Olivya" },