refactor: small fixes

src/lib/querybuilder/panel/QueryBuilderContextMenu.tsx

-import { DropdownItem, DropdownItemContainer, Icon, Input, Popover, PopoverTrigger } from '@/lib/components';
+import { useMemo, useState, useEffect } from 'react';
+import { Popover, PopoverTrigger, DropdownItemContainer, DropdownItem, Input, Icon } from '@/lib/components';
+import { ReactFlowInstance, Node } from 'reactflow';
+import { useActiveQuery, useAppDispatch, useQuerybuilderHash } from '../..';
+import { isEqual } from 'lodash-es';
+import { getDataTypeIcon } from '../../components/DataTypeIcon';
+import { QueryUnionType, NodeAttribute, defaultGraph, QueryRelationDirection, QueryGraphEdgeHandle } from 'ts-common';
 import {
   attributeShownToggle,
   setQuerybuilderGraphology,
@@ -117,7 +123,7 @@ export const QueryBuilderContextMenu = (props: {
         <SubMenu>
           <SubMenuTrigger label="Add/remove attribute" />
           <SubMenuContent className="max-h-60 overflow-y-auto">
-            {props.node?.data.attributes.map(attr => (
+            {props.node?.data.attributes.map((attr: { handleData: QueryGraphEdgeHandle }) => (
               <MenuCheckboxItem
                 key={attr.handleData.attributeName + attr.handleData.nodeId}
                 checked={isAttributeAdded(attr)}