From 19c714a7d52bc9d4d5dd890129b7500815e1b974 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Tue, 25 Jul 2023 17:14:12 -0300
Subject: [PATCH] Added Beta badge

---
 src/frontend/src/CustomNodes/GenericNode/index.tsx | 6 ++++++
 src/frontend/src/index.css                         | 2 +-
 src/frontend/src/types/api/index.ts                | 3 ++-
 3 files changed, 9 insertions(+), 2 deletions(-)

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({
             </div>
           </div>
           <div className="round-button-div">
+            {data.node.beta && (
+              <Badge size="sm" variant="destructive">
+                Beta
+              </Badge>
+            )}
             <div>
               <Tooltip
                 title={
diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 8613a3805..bb32e1dc8 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -369,7 +369,7 @@ The cursor: default; property value restores the browser's default cursor style
   }
 
   .generic-node-status-position {
-    @apply relative top-[3px] h-5 w-5;
+    @apply relative top-[1.5px] h-5 w-5;
   }
 
   .generic-node-status-animation {
diff --git a/src/frontend/src/types/api/index.ts b/src/frontend/src/types/api/index.ts
index eaf862f29..6bdb430aa 100644
--- a/src/frontend/src/types/api/index.ts
+++ b/src/frontend/src/types/api/index.ts
@@ -14,8 +14,9 @@ export type APIClassType = {
   display_name: string;
   input_types?: Array<string>;
   output_types?: Array<string>;
+  beta?: boolean;
   documentation: string;
-  [key: string]: Array<string> | string | APITemplateType;
+  [key: string]: Array<string> | string | APITemplateType | boolean;
 };
 
 export type TemplateVariableType = {