Issue with useFormValue hook in a custom document action

Error: FormBuilder: missing context value
    at useFormBuilder (useFormBuilder.ts:13:11)
    at useFormValue (useFormValue.ts:29:19)
    at SetAndPublishAction (SetAndPublishAction.ts:9:26)
    at HookStateContainer2 (HookStateContainer.ts:24:23)
    at renderWithHooks (react-dom.development.js:16305:18)
    at mountIndeterminateComponent (react-dom.development.js:20074:13)
    at beginWork (react-dom.development.js:21587:16)
    at HTMLUnknownElement.callCallback2 (react-dom.development.js:4164:14)
    at Object.invokeGuardedCallbackDev (react-dom.development.js:4213:16)
    at invokeGuardedCallback (react-dom.development.js:4277:31)

import { useState, useEffect } from 'react'
import { useDataset, useDocumentOperation, useFormValue } from 'sanity'
import { toHTML } from '@portabletext/to-html'


export function SetAndPublishAction(props: any) {
    const { patch, publish } = useDocumentOperation(props.id, props.type)
    const [isPublishing, setIsPublishing] = useState(false)
    const documentData = useFormValue([props.id, "country"])
    const dataset = useDataset()

    useEffect(() => {
        // if the isPublishing state was set to true and the draft has changed
        // to become `null` the document has been published
        if (isPublishing && !props.draft) {
            setIsPublishing(false)
        }
    }, [props.draft])

    return {
        disabled: publish.disabled,
        label: isPublishing ? 'Publishing' : 'Publish & Update',
        onHandle: () => {
            // This will update the button text
            setIsPublishing(true)

            // Set publishedAt to current date and time
            patch.execute([{ set: { name: "hello world" } }])
            console.log("logging test here: ,", dataset)

            // Perform the publish
            publish.execute()

            // Signal that the action is completed
            props.onComplete()
        },
    }
}

export default defineConfig({
  name: 'default',
  title: 'escape-cms',

  projectId: '*****',
  dataset: 'production',

  plugins: [deskTool(
    {
      structure: myStructure,
      defaultDocumentNode: defaultDocumentNodeResolver,
    }
  ), visionTool()],

  document: {
    actions: (prev: any) =>
      prev.map((originalAction: any) =>
        originalAction.action === 'publish' ? SetAndPublishAction : originalAction
      ),
  },

  schema: {
    types: schemaTypes,
  },
})

const documentData = useFormValue([props.type as string, "name"])