diff --git a/space_flow/src/CustomNodes/MemoryNode/index.tsx b/space_flow/src/CustomNodes/MemoryNode/index.tsx new file mode 100644 index 000000000000..0584ea62d62f --- /dev/null +++ b/space_flow/src/CustomNodes/MemoryNode/index.tsx @@ -0,0 +1,31 @@ +import { Transition } from "@headlessui/react"; +import { Handle, Position } from "reactflow"; + +export default function MemoryNode({ data }) { + return ( + +
+ + +
+ Memory +
+ +
+
+ ); +} diff --git a/space_flow/src/pages/FlowPage/components/extraSidebarComponent/index.tsx b/space_flow/src/pages/FlowPage/components/extraSidebarComponent/index.tsx index 7e9a153226b9..ba5d5c7ef175 100644 --- a/space_flow/src/pages/FlowPage/components/extraSidebarComponent/index.tsx +++ b/space_flow/src/pages/FlowPage/components/extraSidebarComponent/index.tsx @@ -1,4 +1,4 @@ -import { Bars2Icon, CommandLineIcon, LightBulbIcon, LinkIcon, RocketLaunchIcon, ShieldCheckIcon, ViewColumnsIcon } from "@heroicons/react/24/outline"; +import { Bars2Icon, CommandLineIcon, CpuChipIcon, LightBulbIcon, LinkIcon, RocketLaunchIcon, ShieldCheckIcon, ViewColumnsIcon } from "@heroicons/react/24/outline"; import { llm_chain } from "../../../../data_assets/llm_chain"; import { prompt } from "../../../../data_assets/prompt"; import DisclosureComponent from "../DisclosureComponent"; @@ -23,6 +23,9 @@ export function ExtraSidebar() { if (nodeType === "validatorNode") { json = JSON.stringify({ content: "" }); } + if (nodeType === "memoryNode") { + json = JSON.stringify({ content: "" }); + } event.dataTransfer.setData("json", json); } @@ -80,6 +83,18 @@ export function ExtraSidebar() { + +
onDragStart(event, "memoryNode")} + > + Memory + +
+
); } diff --git a/space_flow/src/pages/FlowPage/index.tsx b/space_flow/src/pages/FlowPage/index.tsx index f17ca6e65d25..00832f38485b 100644 --- a/space_flow/src/pages/FlowPage/index.tsx +++ b/space_flow/src/pages/FlowPage/index.tsx @@ -13,6 +13,7 @@ import { ExtraSidebar } from "./components/extraSidebarComponent"; import AgentNode from "../../CustomNodes/AgentNode"; import ChainNode from "../../CustomNodes/ChainNode"; import ValidatorNode from "../../CustomNodes/ValidatorNode"; +import MemoryNode from "../../CustomNodes/MemoryNode"; const nodeTypes = { promptNode: PromptNode, @@ -20,6 +21,7 @@ const nodeTypes = { chainNode: ChainNode, agentNode: AgentNode, validatorNode: ValidatorNode, + memoryNode:MemoryNode }; export default function FlowPage() {