diff --git a/src/backend/base/langflow/components/vectorstores/AstraDB.py b/src/backend/base/langflow/components/vectorstores/AstraDB.py
index 9ab73808e..5ac7f85b4 100644
--- a/src/backend/base/langflow/components/vectorstores/AstraDB.py
+++ b/src/backend/base/langflow/components/vectorstores/AstraDB.py
@@ -10,6 +10,8 @@ from langflow.schema import Record
class AstraDBVectorStoreComponent(CustomComponent):
display_name = "AstraDB Vector Store"
description = "Builds or loads an AstraDB Vector Store"
+ icon = "AstraDB"
+ field_order = ["token", "api_endpoint", "collection_name", "inputs", "embedding"]
def build_config(self):
return {
diff --git a/src/backend/base/langflow/components/vectorstores/AstraDBSearch.py b/src/backend/base/langflow/components/vectorstores/AstraDBSearch.py
index 8432f01f1..bc7e69a9a 100644
--- a/src/backend/base/langflow/components/vectorstores/AstraDBSearch.py
+++ b/src/backend/base/langflow/components/vectorstores/AstraDBSearch.py
@@ -9,6 +9,8 @@ from langflow.schema import Record
class AstraDBSearchComponent(AstraDBVectorStoreComponent, LCVectorStoreComponent):
display_name = "AstraDB Search"
description = "Searches an existing AstraDB Vector Store"
+ icon = "AstraDB"
+ field_order = ["token", "api_endpoint", "collection_name", "input_value", "embedding"]
def build_config(self):
return {
diff --git a/src/frontend/src/icons/AstraDB/AstraDB.jsx b/src/frontend/src/icons/AstraDB/AstraDB.jsx
new file mode 100644
index 000000000..4f20ce65b
--- /dev/null
+++ b/src/frontend/src/icons/AstraDB/AstraDB.jsx
@@ -0,0 +1,16 @@
+const AstraSVG = (props) => (
+
+
+);
+export default AstraSVG;
diff --git a/src/frontend/src/icons/AstraDB/Favicon.svg b/src/frontend/src/icons/AstraDB/Favicon.svg
new file mode 100644
index 000000000..7fe145b1b
--- /dev/null
+++ b/src/frontend/src/icons/AstraDB/Favicon.svg
@@ -0,0 +1,12 @@
+
diff --git a/src/frontend/src/icons/AstraDB/index.tsx b/src/frontend/src/icons/AstraDB/index.tsx
new file mode 100644
index 000000000..6d57406c0
--- /dev/null
+++ b/src/frontend/src/icons/AstraDB/index.tsx
@@ -0,0 +1,9 @@
+import React, { forwardRef } from "react";
+import AstraSVG from "./AstraDB";
+
+export const AstraDBIcon = forwardRef<
+ SVGSVGElement,
+ React.PropsWithChildren<{}>
+>((props, ref) => {
+ return ;
+});
diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts
index 653d7e642..958b0652a 100644
--- a/src/frontend/src/utils/styleUtils.ts
+++ b/src/frontend/src/utils/styleUtils.ts
@@ -138,6 +138,7 @@ import { FaApple, FaGithub } from "react-icons/fa";
import { AWSIcon } from "../icons/AWS";
import { AirbyteIcon } from "../icons/Airbyte";
import { AnthropicIcon } from "../icons/Anthropic";
+import { AstraDBIcon } from "../icons/AstraDB";
import { AzureIcon } from "../icons/Azure";
import { BingIcon } from "../icons/Bing";
import { BotMessageSquareIcon } from "../icons/BotMessageSquare";
@@ -311,6 +312,7 @@ export const nodeIconsLucide: iconsType = {
Amazon: AWSIcon,
Anthropic: AnthropicIcon,
ChatAnthropic: AnthropicIcon,
+ AstraDB: AstraDBIcon,
BingSearchAPIWrapper: BingIcon,
BingSearchRun: BingIcon,
Cohere: CohereIcon,