diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 385d441d3..942d62135 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -16,6 +16,7 @@ import PromptAreaComponent from "../../../../components/promptComponent";
import TextAreaComponent from "../../../../components/textAreaComponent";
import ToggleShadComponent from "../../../../components/toggleShadComponent";
import { Button } from "../../../../components/ui/button";
+import { RefreshButton } from "../../../../components/ui/refreshButton";
import {
INPUT_HANDLER_HOVER,
LANGFLOW_SUPPORTED_TYPES,
@@ -68,6 +69,7 @@ export default function ParameterComponent({
const nodes = useFlowStore((state) => state.nodes);
const edges = useFlowStore((state) => state.edges);
const setNode = useFlowStore((state) => state.setNode);
+ const [isLoading, setIsLoading] = useState(false);
const flow = currentFlow?.data?.nodes ?? null;
@@ -420,14 +422,11 @@ export default function ParameterComponent({
/>
{data.node?.template[name].refresh && (
-
+
)}
)}
@@ -466,14 +465,11 @@ export default function ParameterComponent({
/>
{data.node?.template[name].refresh && (
-
+
)}
) : left === true && type === "code" ? (
diff --git a/src/frontend/src/components/ui/refreshButton.tsx b/src/frontend/src/components/ui/refreshButton.tsx
new file mode 100644
index 000000000..7374686f2
--- /dev/null
+++ b/src/frontend/src/components/ui/refreshButton.tsx
@@ -0,0 +1,38 @@
+import { useState } from "react";
+import IconComponent from "../../components/genericIconComponent";
+import { NodeDataType } from "../../types/flow";
+import { cn } from "../../utils/utils";
+function RefreshButton({
+ name,
+ data,
+ handleUpdateValues,
+}: {
+ name: string;
+ data: NodeDataType;
+ handleUpdateValues: (name: string, data: NodeDataType) => void;
+}) {
+ const [isLoading, setIsLoading] = useState(false);
+
+ const handleClick = () => {
+ setIsLoading(true);
+ console.log("refreshing");
+ handleUpdateValues(name, data);
+ setInterval(() => {
+ setIsLoading(false);
+ }, 500);
+ };
+
+ return (
+
+ );
+}
+
+export { RefreshButton };