Merge commit 'a8390532f7' as 'ai-context/workbench-ui'

ai-context/workbench-ui/src/components/token-cost/Controls.tsx

@@ -0,0 +1,39 @@
+import React, { useState } from "react";
+
+import { Plus } from "lucide-react";
+
+import { Button, Box } from "@chakra-ui/react";
+
+import EditDialog from "./EditDialog";
+
+const Controls = ({ onUpdate }) => {
+  const [createOpen, setCreateOpen] = useState(false);
+
+  const onComplete = () => {
+    onUpdate();
+    setCreateOpen(false);
+  };
+
+  return (
+    <Box>
+      <Button
+        mt={5}
+        ml={5}
+        mb={5}
+        variant="solid"
+        colorPalette="primary"
+        onClick={() => setCreateOpen(true)}
+      >
+        <Plus /> Create Model Cost
+      </Button>
+      <EditDialog
+        open={createOpen}
+        onOpenChange={setCreateOpen}
+        create={true}
+        onComplete={() => onComplete()}
+      />
+    </Box>
+  );
+};
+
+export default Controls;