diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx index 9c933dcf7..2a264a59c 100644 --- a/src/frontend/src/CustomNodes/GenericNode/index.tsx +++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx @@ -4,6 +4,7 @@ import { NodeToolbar, useUpdateNodeInternals } from "reactflow"; import ShadTooltip from "../../components/ShadTooltipComponent"; import Tooltip from "../../components/TooltipComponent"; import IconComponent from "../../components/genericIconComponent"; +import { Badge } from "../../components/ui/badge"; import { useSSE } from "../../contexts/SSEContext"; import { alertContext } from "../../contexts/alertContext"; import { TabsContext } from "../../contexts/tabsContext"; @@ -111,6 +112,11 @@ export default function GenericNode({
+ {data.node.beta && ( + + Beta + + )}
; output_types?: Array; + beta?: boolean; documentation: string; - [key: string]: Array | string | APITemplateType; + [key: string]: Array | string | APITemplateType | boolean; }; export type TemplateVariableType = {