From 3169fa3c0affceaab660c788064e956ae295c22b Mon Sep 17 00:00:00 2001
From: Sacha Al Himdani <sacha@getprobo.com>
Date: Tue, 9 Dec 2025 10:04:47 +0100
Subject: [PATCH] Fix audit and framework deletion

Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
---
 .../src/components/form/AuditSelectField.tsx  |  9 ++++++--
 apps/console/src/hooks/graph/AuditGraph.ts    | 11 +++++----
 .../src/hooks/graph/NonconformityGraph.ts     | 14 +++++------
 ...onconformityGraphCreateMutation.graphql.ts |  6 ++---
 .../NonconformityGraphNodeQuery.graphql.ts    |  4 ++--
 ...onconformityGraphUpdateMutation.graphql.ts |  4 ++--
 .../organizations/audits/AuditDetailsPage.tsx |  5 +++-
 .../nonconformities/NonconformitiesPage.tsx   |  8 ++++---
 .../NonconformityDetailsPage.tsx              |  3 +--
 .../NonconformitiesPageFragment.graphql.ts    |  4 ++--
 .../dialogs/CreateNonconformityDialog.tsx     |  5 ++--
 pkg/coredata/migrations/20251208T172842Z.sql  |  4 ++++
 pkg/coredata/nonconformity.go                 |  2 +-
 pkg/probo/nonconformity_service.go            | 14 ++++++-----
 pkg/server/api/console/v1/schema.graphql      |  6 ++---
 pkg/server/api/console/v1/schema/schema.go    | 23 +++++++------------
 pkg/server/api/console/v1/types/types.go      |  6 ++---
 pkg/server/api/console/v1/v1_resolver.go      |  8 +++++--
 pkg/server/api/mcp/v1/schema.resolvers.go     |  2 +-
 pkg/server/api/mcp/v1/specification.yaml      | 15 ++++++++----
 pkg/server/api/mcp/v1/types/types.go          | 16 ++++++-------
 21 files changed, 95 insertions(+), 74 deletions(-)
 create mode 100644 pkg/coredata/migrations/20251208T172842Z.sql

diff --git a/apps/console/src/components/form/AuditSelectField.tsx b/apps/console/src/components/form/AuditSelectField.tsx
index c7540effd..bcfe22b5a 100644
--- a/apps/console/src/components/form/AuditSelectField.tsx
+++ b/apps/console/src/components/form/AuditSelectField.tsx
@@ -67,6 +67,8 @@ function AuditSelectWithQuery<T extends FieldValues = FieldValues>(
   const data = useLazyLoadQuery<AuditSelectFieldQuery>(auditsQuery, { organizationId }, { fetchPolicy: "network-only" });
   const audits = data?.organization?.audits?.edges?.map((edge) => edge.node).filter((node) => node !== null) ?? [];
 
+  const NONE_VALUE = "__NONE__";
+
   return (
     <Controller
       control={control}
@@ -77,12 +79,15 @@ function AuditSelectWithQuery<T extends FieldValues = FieldValues>(
           id={name}
           variant="editor"
           placeholder={__("Select an audit")}
-          onValueChange={field.onChange}
+          onValueChange={(value) => field.onChange(value === NONE_VALUE ? "" : value)}
           key={audits?.length.toString() ?? "0"}
           {...field}
           className="w-full"
-          value={field.value ?? ""}
+          value={field.value || NONE_VALUE}
         >
+          <Option value={NONE_VALUE}>
+            <span className="text-txt-tertiary">{__("None")}</span>
+          </Option>
           {audits?.map((audit) => (
             <Option key={audit.id} value={audit.id}>
               <div className="flex items-center justify-between w-full">
diff --git a/apps/console/src/hooks/graph/AuditGraph.ts b/apps/console/src/hooks/graph/AuditGraph.ts
index cd10bf811..e69e4dd69 100644
--- a/apps/console/src/hooks/graph/AuditGraph.ts
+++ b/apps/console/src/hooks/graph/AuditGraph.ts
@@ -112,7 +112,8 @@ export const deleteAuditMutation = graphql`
 
 export const useDeleteAudit = (
   audit: { id: string; framework: { name: string } },
-  connectionId: string
+  connectionId: string,
+  onSuccess?: () => void
 ) => {
   const { __ } = useTranslate();
   const [mutate] = useMutationWithToasts(deleteAuditMutation, {
@@ -123,15 +124,17 @@ export const useDeleteAudit = (
 
   return () => {
     confirm(
-      () =>
-        mutate({
+      async () => {
+        await mutate({
           variables: {
             input: {
               auditId: audit.id!,
             },
             connections: [connectionId],
           },
-        }),
+        });
+        onSuccess?.();
+      },
       {
         message: sprintf(
           __(
diff --git a/apps/console/src/hooks/graph/NonconformityGraph.ts b/apps/console/src/hooks/graph/NonconformityGraph.ts
index ebef2e999..ad41abf19 100644
--- a/apps/console/src/hooks/graph/NonconformityGraph.ts
+++ b/apps/console/src/hooks/graph/NonconformityGraph.ts
@@ -168,7 +168,7 @@ export const useCreateNonconformity = (connectionId: string) => {
     organizationId: string;
     referenceId: string;
     description?: string;
-    auditId: string;
+    auditId?: string;
     dateIdentified?: string;
     rootCause: string;
     correctiveAction?: string;
@@ -183,9 +183,6 @@ export const useCreateNonconformity = (connectionId: string) => {
     if (!input.referenceId) {
       return alert(__("Failed to create nonconformity: reference ID is required"));
     }
-    if (!input.auditId) {
-      return alert(__("Failed to create nonconformity: audit is required"));
-    }
     if (!input.ownerId) {
       return alert(__("Failed to create nonconformity: owner is required"));
     }
@@ -199,7 +196,7 @@ export const useCreateNonconformity = (connectionId: string) => {
           organizationId: input.organizationId,
           referenceId: input.referenceId,
           description: input.description,
-          auditId: input.auditId,
+          auditId: input.auditId || undefined,
           dateIdentified: input.dateIdentified,
           rootCause: input.rootCause,
           correctiveAction: input.correctiveAction,
@@ -226,7 +223,7 @@ export const useUpdateNonconformity = () => {
     rootCause?: string;
     correctiveAction?: string;
     ownerId?: string;
-    auditId?: string;
+    auditId?: string | null;
     dueDate?: string | null;
     status?: string;
     effectivenessCheck?: string;
@@ -237,7 +234,10 @@ export const useUpdateNonconformity = () => {
 
     return promisifyMutation(mutate)({
       variables: {
-        input,
+        input: {
+          ...input,
+          auditId: input.auditId || null,
+        },
       },
     });
   };
diff --git a/apps/console/src/hooks/graph/__generated__/NonconformityGraphCreateMutation.graphql.ts b/apps/console/src/hooks/graph/__generated__/NonconformityGraphCreateMutation.graphql.ts
index a1954cd56..0af520b18 100644
--- a/apps/console/src/hooks/graph/__generated__/NonconformityGraphCreateMutation.graphql.ts
+++ b/apps/console/src/hooks/graph/__generated__/NonconformityGraphCreateMutation.graphql.ts
@@ -1,5 +1,5 @@
 /**
- * @generated SignedSource<<a7d4ac5eee7f07fadad4e2c1f25e87e5>>
+ * @generated SignedSource<<db7fca9f154989ebab9e5c9b0f5bfd85>>
  * @lightSyntaxTransform
  * @nogrep
  */
@@ -11,7 +11,7 @@
 import { ConcreteRequest } from 'relay-runtime';
 export type NonconformityStatus = "CLOSED" | "IN_PROGRESS" | "OPEN";
 export type CreateNonconformityInput = {
-  auditId: string;
+  auditId?: string | null | undefined;
   correctiveAction?: string | null | undefined;
   dateIdentified?: any | null | undefined;
   description?: string | null | undefined;
@@ -36,7 +36,7 @@ export type NonconformityGraphCreateMutation$data = {
             readonly name: string;
           };
           readonly id: string;
-        };
+        } | null | undefined;
         readonly createdAt: any;
         readonly dateIdentified: any | null | undefined;
         readonly description: string | null | undefined;
diff --git a/apps/console/src/hooks/graph/__generated__/NonconformityGraphNodeQuery.graphql.ts b/apps/console/src/hooks/graph/__generated__/NonconformityGraphNodeQuery.graphql.ts
index 8df0a1593..9203a62b6 100644
--- a/apps/console/src/hooks/graph/__generated__/NonconformityGraphNodeQuery.graphql.ts
+++ b/apps/console/src/hooks/graph/__generated__/NonconformityGraphNodeQuery.graphql.ts
@@ -1,5 +1,5 @@
 /**
- * @generated SignedSource<<125cf1d460b37ac442d0f30c14e5af11>>
+ * @generated SignedSource<<526115183274c5c81a5a70c16012591e>>
  * @lightSyntaxTransform
  * @nogrep
  */
@@ -21,7 +21,7 @@ export type NonconformityGraphNodeQuery$data = {
         readonly name: string;
       };
       readonly id: string;
-    };
+    } | null | undefined;
     readonly correctiveAction?: string | null | undefined;
     readonly createdAt?: any;
     readonly dateIdentified?: any | null | undefined;
diff --git a/apps/console/src/hooks/graph/__generated__/NonconformityGraphUpdateMutation.graphql.ts b/apps/console/src/hooks/graph/__generated__/NonconformityGraphUpdateMutation.graphql.ts
index d8439fec3..f53793f80 100644
--- a/apps/console/src/hooks/graph/__generated__/NonconformityGraphUpdateMutation.graphql.ts
+++ b/apps/console/src/hooks/graph/__generated__/NonconformityGraphUpdateMutation.graphql.ts
@@ -1,5 +1,5 @@
 /**
- * @generated SignedSource<<9c7a235f3dc4f691c8f9d036536cf272>>
+ * @generated SignedSource<<2f4ca0549a2e01d68d9fb752298beb4c>>
  * @lightSyntaxTransform
  * @nogrep
  */
@@ -35,7 +35,7 @@ export type NonconformityGraphUpdateMutation$data = {
           readonly name: string;
         };
         readonly id: string;
-      };
+      } | null | undefined;
       readonly correctiveAction: string | null | undefined;
       readonly dateIdentified: any | null | undefined;
       readonly description: string | null | undefined;
diff --git a/apps/console/src/pages/organizations/audits/AuditDetailsPage.tsx b/apps/console/src/pages/organizations/audits/AuditDetailsPage.tsx
index 7c655365d..c8d521c13 100644
--- a/apps/console/src/pages/organizations/audits/AuditDetailsPage.tsx
+++ b/apps/console/src/pages/organizations/audits/AuditDetailsPage.tsx
@@ -36,6 +36,7 @@ import { getAuditStateLabel, getAuditStateVariant, auditStates, fileSize, sprint
 import type { AuditGraphNodeQuery } from "/hooks/graph/__generated__/AuditGraphNodeQuery.graphql";
 import { use } from "react";
 import { PermissionsContext } from "/providers/PermissionsContext";
+import { useNavigate } from "react-router";
 
 const updateAuditSchema = z.object({
   name: z.string().nullable().optional(),
@@ -54,10 +55,12 @@ export default function AuditDetailsPage(props: Props) {
   const { __ } = useTranslate();
   const organizationId = useOrganizationId();
   const { isAuthorized } = use(PermissionsContext);
+  const navigate = useNavigate();
 
   const deleteAudit = useDeleteAudit(
     { id: auditEntry.id!, framework: { name: auditEntry.framework!.name} },
-    ConnectionHandler.getConnectionID(organizationId, "AuditsPage_audits")
+    ConnectionHandler.getConnectionID(organizationId, "AuditsPage_audits"),
+    () => navigate(`/organizations/${organizationId}/audits`)
   );
 
   const { control, formState, handleSubmit, register, reset } = useFormWithSchema(updateAuditSchema, {
diff --git a/apps/console/src/pages/organizations/nonconformities/NonconformitiesPage.tsx b/apps/console/src/pages/organizations/nonconformities/NonconformitiesPage.tsx
index 28a24793e..a1b687e68 100644
--- a/apps/console/src/pages/organizations/nonconformities/NonconformitiesPage.tsx
+++ b/apps/console/src/pages/organizations/nonconformities/NonconformitiesPage.tsx
@@ -273,9 +273,11 @@ function NonconformityRow({
         </Badge>
       </Td>
       <Td>
-        {nonconformity.audit.name
-          ? `${nonconformity.audit.framework.name} - ${nonconformity.audit.name}`
-          : nonconformity.audit.framework.name
+        {nonconformity.audit
+          ? nonconformity.audit.name
+            ? `${nonconformity.audit.framework?.name} - ${nonconformity.audit.name}`
+            : nonconformity.audit.framework?.name
+          : <span className="text-txt-tertiary">{__("No audit")}</span>
         }
       </Td>
       <Td>{nonconformity.owner.fullName}</Td>
diff --git a/apps/console/src/pages/organizations/nonconformities/NonconformityDetailsPage.tsx b/apps/console/src/pages/organizations/nonconformities/NonconformityDetailsPage.tsx
index 99996778d..3b4f0a0f7 100644
--- a/apps/console/src/pages/organizations/nonconformities/NonconformityDetailsPage.tsx
+++ b/apps/console/src/pages/organizations/nonconformities/NonconformityDetailsPage.tsx
@@ -47,7 +47,7 @@ const updateNonconformitySchema = z.object({
   effectivenessCheck: z.string().optional(),
   status: z.enum(["OPEN", "IN_PROGRESS", "CLOSED"]),
   ownerId: z.string().min(1, "Owner is required"),
-  auditId: z.string().min(1, "Audit is required"),
+  auditId: z.string().optional(),
 });
 
 type Props = {
@@ -198,7 +198,6 @@ export default function NonconformityDetailsPage(props: Props) {
               name="auditId"
               label={__("Audit")}
               error={formState.errors.auditId?.message}
-              required
               disabled={isSnapshotMode}
             />
 
diff --git a/apps/console/src/pages/organizations/nonconformities/__generated__/NonconformitiesPageFragment.graphql.ts b/apps/console/src/pages/organizations/nonconformities/__generated__/NonconformitiesPageFragment.graphql.ts
index ab757e18f..3f468819d 100644
--- a/apps/console/src/pages/organizations/nonconformities/__generated__/NonconformitiesPageFragment.graphql.ts
+++ b/apps/console/src/pages/organizations/nonconformities/__generated__/NonconformitiesPageFragment.graphql.ts
@@ -1,5 +1,5 @@
 /**
- * @generated SignedSource<<b53ef2729d5b435baeb0d3b068be8cef>>
+ * @generated SignedSource<<310dfd9974ac15af86b8e8ecf21d2b5c>>
  * @lightSyntaxTransform
  * @nogrep
  */
@@ -24,7 +24,7 @@ export type NonconformitiesPageFragment$data = {
           };
           readonly id: string;
           readonly name: string | null | undefined;
-        };
+        } | null | undefined;
         readonly correctiveAction: string | null | undefined;
         readonly createdAt: any;
         readonly dateIdentified: any | null | undefined;
diff --git a/apps/console/src/pages/organizations/nonconformities/dialogs/CreateNonconformityDialog.tsx b/apps/console/src/pages/organizations/nonconformities/dialogs/CreateNonconformityDialog.tsx
index a41e3f828..dfb5d9437 100644
--- a/apps/console/src/pages/organizations/nonconformities/dialogs/CreateNonconformityDialog.tsx
+++ b/apps/console/src/pages/organizations/nonconformities/dialogs/CreateNonconformityDialog.tsx
@@ -27,7 +27,7 @@ import { formatDatetime, getStatusOptions } from "@probo/helpers";
 const schema = z.object({
   referenceId: z.string().min(1, "Reference ID is required"),
   description: z.string().optional(),
-  auditId: z.string().min(1, "Audit is required"),
+  auditId: z.string().optional(),
   dateIdentified: z.string().optional(),
   rootCause: z.string().min(1, "Root cause is required"),
   correctiveAction: z.string().optional(),
@@ -78,7 +78,7 @@ export function CreateNonconformityDialog({
         organizationId,
         referenceId: formData.referenceId,
         description: formData.description || undefined,
-        auditId: formData.auditId,
+        auditId: formData.auditId || undefined,
         dateIdentified: formatDatetime(formData.dateIdentified),
         rootCause: formData.rootCause,
         correctiveAction: formData.correctiveAction || undefined,
@@ -128,7 +128,6 @@ export function CreateNonconformityDialog({
             name="auditId"
             label={__("Audit")}
             error={formState.errors.auditId?.message}
-            required
           />
 
           <div className="space-y-2">
diff --git a/pkg/coredata/migrations/20251208T172842Z.sql b/pkg/coredata/migrations/20251208T172842Z.sql
new file mode 100644
index 000000000..df75af499
--- /dev/null
+++ b/pkg/coredata/migrations/20251208T172842Z.sql
@@ -0,0 +1,4 @@
+ALTER TABLE nonconformities DROP CONSTRAINT IF EXISTS nonconformities_audit_id_fkey;
+ALTER TABLE nonconformities ALTER COLUMN audit_id DROP NOT NULL;
+ALTER TABLE nonconformities ADD CONSTRAINT nonconformities_audit_id_fkey
+    FOREIGN KEY (audit_id) REFERENCES audits(id) ON UPDATE CASCADE ON DELETE SET NULL;
diff --git a/pkg/coredata/nonconformity.go b/pkg/coredata/nonconformity.go
index deee54f6c..f6e02acdf 100644
--- a/pkg/coredata/nonconformity.go
+++ b/pkg/coredata/nonconformity.go
@@ -34,7 +34,7 @@ type (
 		SourceID           *gid.GID            `db:"source_id"`
 		ReferenceID        string              `db:"reference_id"`
 		Description        *string             `db:"description"`
-		AuditID            gid.GID             `db:"audit_id"`
+		AuditID            *gid.GID            `db:"audit_id"`
 		DateIdentified     *time.Time          `db:"date_identified"`
 		RootCause          string              `db:"root_cause"`
 		CorrectiveAction   *string             `db:"corrective_action"`
diff --git a/pkg/probo/nonconformity_service.go b/pkg/probo/nonconformity_service.go
index 14333b2f0..4729dce20 100644
--- a/pkg/probo/nonconformity_service.go
+++ b/pkg/probo/nonconformity_service.go
@@ -35,7 +35,7 @@ type (
 		OrganizationID     gid.GID
 		ReferenceID        string
 		Description        *string
-		AuditID            gid.GID
+		AuditID            *gid.GID
 		DateIdentified     *time.Time
 		RootCause          string
 		CorrectiveAction   *string
@@ -53,7 +53,7 @@ type (
 		RootCause          *string
 		CorrectiveAction   **string
 		OwnerID            *gid.GID
-		AuditID            *gid.GID
+		AuditID            **gid.GID
 		DueDate            **time.Time
 		Status             *coredata.NonconformityStatus
 		EffectivenessCheck **string
@@ -66,7 +66,7 @@ func (cnr *CreateNonconformityRequest) Validate() error {
 	v.Check(cnr.OrganizationID, "organization_id", validator.Required(), validator.GID(coredata.OrganizationEntityType))
 	v.Check(cnr.ReferenceID, "reference_id", validator.Required(), validator.SafeText(NameMaxLength))
 	v.Check(cnr.Description, "description", validator.SafeText(ContentMaxLength))
-	v.Check(cnr.AuditID, "audit_id", validator.Required(), validator.GID(coredata.AuditEntityType))
+	v.Check(cnr.AuditID, "audit_id", validator.GID(coredata.AuditEntityType))
 	v.Check(cnr.RootCause, "root_cause", validator.Required(), validator.SafeText(ContentMaxLength))
 	v.Check(cnr.CorrectiveAction, "corrective_action", validator.SafeText(ContentMaxLength))
 	v.Check(cnr.OwnerID, "owner_id", validator.Required(), validator.GID(coredata.PeopleEntityType))
@@ -149,9 +149,11 @@ func (s *NonconformityService) Create(
 				return fmt.Errorf("cannot load organization: %w", err)
 			}
 
-			audit := &coredata.Audit{}
-			if err := audit.LoadByID(ctx, conn, s.svc.scope, req.AuditID); err != nil {
-				return fmt.Errorf("cannot load audit: %w", err)
+			if req.AuditID != nil {
+				audit := &coredata.Audit{}
+				if err := audit.LoadByID(ctx, conn, s.svc.scope, *req.AuditID); err != nil {
+					return fmt.Errorf("cannot load audit: %w", err)
+				}
 			}
 
 			people := &coredata.People{}
diff --git a/pkg/server/api/console/v1/schema.graphql b/pkg/server/api/console/v1/schema.graphql
index b9498f1da..52780e844 100644
--- a/pkg/server/api/console/v1/schema.graphql
+++ b/pkg/server/api/console/v1/schema.graphql
@@ -2155,7 +2155,7 @@ type Nonconformity implements Node {
   organization: Organization! @goField(forceResolver: true)
   referenceId: String!
   description: String
-  audit: Audit! @goField(forceResolver: true)
+  audit: Audit @goField(forceResolver: true)
   dateIdentified: Datetime
   rootCause: String!
   correctiveAction: String
@@ -3661,7 +3661,7 @@ input CreateNonconformityInput {
   organizationId: ID!
   referenceId: String!
   description: String
-  auditId: ID!
+  auditId: ID
   dateIdentified: Datetime
   rootCause: String!
   correctiveAction: String
@@ -3679,7 +3679,7 @@ input UpdateNonconformityInput {
   rootCause: String
   correctiveAction: String @goField(omittable: true)
   ownerId: ID
-  auditId: ID
+  auditId: ID @goField(omittable: true)
   dueDate: Datetime @goField(omittable: true)
   status: NonconformityStatus
   effectivenessCheck: String @goField(omittable: true)
diff --git a/pkg/server/api/console/v1/schema/schema.go b/pkg/server/api/console/v1/schema/schema.go
index edd8532fa..877fa1b91 100644
--- a/pkg/server/api/console/v1/schema/schema.go
+++ b/pkg/server/api/console/v1/schema/schema.go
@@ -11708,7 +11708,7 @@ type Nonconformity implements Node {
   organization: Organization! @goField(forceResolver: true)
   referenceId: String!
   description: String
-  audit: Audit! @goField(forceResolver: true)
+  audit: Audit @goField(forceResolver: true)
   dateIdentified: Datetime
   rootCause: String!
   correctiveAction: String
@@ -13214,7 +13214,7 @@ input CreateNonconformityInput {
   organizationId: ID!
   referenceId: String!
   description: String
-  auditId: ID!
+  auditId: ID
   dateIdentified: Datetime
   rootCause: String!
   correctiveAction: String
@@ -13232,7 +13232,7 @@ input UpdateNonconformityInput {
   rootCause: String
   correctiveAction: String @goField(omittable: true)
   ownerId: ID
-  auditId: ID
+  auditId: ID @goField(omittable: true)
   dueDate: Datetime @goField(omittable: true)
   status: NonconformityStatus
   effectivenessCheck: String @goField(omittable: true)
@@ -37092,9 +37092,9 @@ func (ec *executionContext) _Nonconformity_audit(ctx context.Context, field grap
 			return ec.resolvers.Nonconformity().Audit(ctx, obj)
 		},
 		nil,
-		ec.marshalNAudit2ᚖgoᚗproboᚗincᚋproboᚋpkgᚋserverᚋapiᚋconsoleᚋv1ᚋtypesᚐAudit,
-		true,
+		ec.marshalOAudit2ᚖgoᚗproboᚗincᚋproboᚋpkgᚋserverᚋapiᚋconsoleᚋv1ᚋtypesᚐAudit,
 		true,
+		false,
 	)
 }
 
@@ -57528,7 +57528,7 @@ func (ec *executionContext) unmarshalInputCreateNonconformityInput(ctx context.C
 			it.Description = data
 		case "auditId":
 			ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("auditId"))
-			data, err := ec.unmarshalNID2goᚗproboᚗincᚋproboᚋpkgᚋgidᚐGID(ctx, v)
+			data, err := ec.unmarshalOID2ᚖgoᚗproboᚗincᚋproboᚋpkgᚋgidᚐGID(ctx, v)
 			if err != nil {
 				return it, err
 			}
@@ -62420,7 +62420,7 @@ func (ec *executionContext) unmarshalInputUpdateNonconformityInput(ctx context.C
 			if err != nil {
 				return it, err
 			}
-			it.AuditID = data
+			it.AuditID = graphql.OmittableOf(data)
 		case "dueDate":
 			ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("dueDate"))
 			data, err := ec.unmarshalODatetime2ᚖtimeᚐTime(ctx, v)
@@ -73824,16 +73824,13 @@ func (ec *executionContext) _Nonconformity(ctx context.Context, sel ast.Selectio
 		case "audit":
 			field := field
 
-			innerFunc := func(ctx context.Context, fs *graphql.FieldSet) (res graphql.Marshaler) {
+			innerFunc := func(ctx context.Context, _ *graphql.FieldSet) (res graphql.Marshaler) {
 				defer func() {
 					if r := recover(); r != nil {
 						ec.Error(ctx, ec.Recover(ctx, r))
 					}
 				}()
 				res = ec._Nonconformity_audit(ctx, field, obj)
-				if res == graphql.Null {
-					atomic.AddUint32(&fs.Invalids, 1)
-				}
 				return res
 			}
 
@@ -83347,10 +83344,6 @@ func (ec *executionContext) marshalNAssignTaskPayload2ᚖgoᚗproboᚗincᚋprob
 	return ec._AssignTaskPayload(ctx, sel, v)
 }
 
-func (ec *executionContext) marshalNAudit2goᚗproboᚗincᚋproboᚋpkgᚋserverᚋapiᚋconsoleᚋv1ᚋtypesᚐAudit(ctx context.Context, sel ast.SelectionSet, v types.Audit) graphql.Marshaler {
-	return ec._Audit(ctx, sel, &v)
-}
-
 func (ec *executionContext) marshalNAudit2ᚖgoᚗproboᚗincᚋproboᚋpkgᚋserverᚋapiᚋconsoleᚋv1ᚋtypesᚐAudit(ctx context.Context, sel ast.SelectionSet, v *types.Audit) graphql.Marshaler {
 	if v == nil {
 		if !graphql.HasFieldError(ctx, graphql.GetFieldContext(ctx)) {
diff --git a/pkg/server/api/console/v1/types/types.go b/pkg/server/api/console/v1/types/types.go
index 220a92441..59e71c378 100644
--- a/pkg/server/api/console/v1/types/types.go
+++ b/pkg/server/api/console/v1/types/types.go
@@ -377,7 +377,7 @@ type CreateNonconformityInput struct {
 	OrganizationID     gid.GID                      `json:"organizationId"`
 	ReferenceID        string                       `json:"referenceId"`
 	Description        *string                      `json:"description,omitempty"`
-	AuditID            gid.GID                      `json:"auditId"`
+	AuditID            *gid.GID                     `json:"auditId,omitempty"`
 	DateIdentified     *time.Time                   `json:"dateIdentified,omitempty"`
 	RootCause          string                       `json:"rootCause"`
 	CorrectiveAction   *string                      `json:"correctiveAction,omitempty"`
@@ -1368,7 +1368,7 @@ type Nonconformity struct {
 	Organization       *Organization                `json:"organization"`
 	ReferenceID        string                       `json:"referenceId"`
 	Description        *string                      `json:"description,omitempty"`
-	Audit              *Audit                       `json:"audit"`
+	Audit              *Audit                       `json:"audit,omitempty"`
 	DateIdentified     *time.Time                   `json:"dateIdentified,omitempty"`
 	RootCause          string                       `json:"rootCause"`
 	CorrectiveAction   *string                      `json:"correctiveAction,omitempty"`
@@ -1997,7 +1997,7 @@ type UpdateNonconformityInput struct {
 	RootCause          *string                       `json:"rootCause,omitempty"`
 	CorrectiveAction   graphql.Omittable[*string]    `json:"correctiveAction,omitempty"`
 	OwnerID            *gid.GID                      `json:"ownerId,omitempty"`
-	AuditID            *gid.GID                      `json:"auditId,omitempty"`
+	AuditID            graphql.Omittable[*gid.GID]   `json:"auditId,omitempty"`
 	DueDate            graphql.Omittable[*time.Time] `json:"dueDate,omitempty"`
 	Status             *coredata.NonconformityStatus `json:"status,omitempty"`
 	EffectivenessCheck graphql.Omittable[*string]    `json:"effectivenessCheck,omitempty"`
diff --git a/pkg/server/api/console/v1/v1_resolver.go b/pkg/server/api/console/v1/v1_resolver.go
index 3fec7e917..019e9c6fc 100644
--- a/pkg/server/api/console/v1/v1_resolver.go
+++ b/pkg/server/api/console/v1/v1_resolver.go
@@ -3984,7 +3984,7 @@ func (r *mutationResolver) UpdateNonconformity(ctx context.Context, input types.
 		RootCause:          input.RootCause,
 		CorrectiveAction:   UnwrapOmittable(input.CorrectiveAction),
 		OwnerID:            input.OwnerID,
-		AuditID:            input.AuditID,
+		AuditID:            UnwrapOmittable(input.AuditID),
 		DueDate:            UnwrapOmittable(input.DueDate),
 		Status:             input.Status,
 		EffectivenessCheck: UnwrapOmittable(input.EffectivenessCheck),
@@ -4596,7 +4596,11 @@ func (r *nonconformityResolver) Audit(ctx context.Context, obj *types.Nonconform
 		panic(fmt.Errorf("cannot get nonconformity: %w", err))
 	}
 
-	audit, err := prb.Audits.Get(ctx, nonconformity.AuditID)
+	if nonconformity.AuditID == nil {
+		return nil, nil
+	}
+
+	audit, err := prb.Audits.Get(ctx, *nonconformity.AuditID)
 	if err != nil {
 		var errNotFound *coredata.ErrAuditNotFound
 		if errors.As(err, &errNotFound) {
diff --git a/pkg/server/api/mcp/v1/schema.resolvers.go b/pkg/server/api/mcp/v1/schema.resolvers.go
index f8beb85a1..7d219f5cd 100644
--- a/pkg/server/api/mcp/v1/schema.resolvers.go
+++ b/pkg/server/api/mcp/v1/schema.resolvers.go
@@ -784,7 +784,7 @@ func (r *Resolver) UpdateNonconformityTool(ctx context.Context, req *mcp.CallToo
 			RootCause:          input.RootCause,
 			CorrectiveAction:   UnwrapOmittable(input.CorrectiveAction),
 			OwnerID:            input.OwnerID,
-			AuditID:            input.AuditID,
+			AuditID:            UnwrapOmittable(input.AuditID),
 			DueDate:            UnwrapOmittable(input.DueDate),
 			Status:             input.Status,
 			EffectivenessCheck: UnwrapOmittable(input.EffectivenessCheck),
diff --git a/pkg/server/api/mcp/v1/specification.yaml b/pkg/server/api/mcp/v1/specification.yaml
index 6fe856650..c635e41b8 100644
--- a/pkg/server/api/mcp/v1/specification.yaml
+++ b/pkg/server/api/mcp/v1/specification.yaml
@@ -1496,7 +1496,6 @@ components:
         - id
         - organization_id
         - reference_id
-        - audit_id
         - root_cause
         - owner_id
         - status
@@ -1525,7 +1524,11 @@ components:
             - "null"
           description: Description
         audit_id:
-          $ref: "#/components/schemas/GID"
+          anyOf:
+            - $ref: "#/components/schemas/GID"
+              description: Audit ID
+            - type: "null"
+              description: No audit
           description: Audit ID
         date_identified:
           type:
@@ -1626,7 +1629,6 @@ components:
       required:
         - organization_id
         - reference_id
-        - audit_id
         - root_cause
         - owner_id
         - status
@@ -1641,7 +1643,11 @@ components:
           type: string
           description: Description
         audit_id:
-          $ref: "#/components/schemas/GID"
+          anyOf:
+            - $ref: "#/components/schemas/GID"
+              description: Audit ID
+            - type: "null"
+              description: No audit
           description: Audit ID
         date_identified:
           type: string
@@ -1718,6 +1724,7 @@ components:
               $ref: "#/components/schemas/GID"
             - type: "null"
           description: Audit ID
+          go.probo.inc/mcpgen/omittable: true
         due_date:
           type: ["string", "null"]
           format: date-time
diff --git a/pkg/server/api/mcp/v1/types/types.go b/pkg/server/api/mcp/v1/types/types.go
index 5aa6a54c6..f1a912f2f 100644
--- a/pkg/server/api/mcp/v1/types/types.go
+++ b/pkg/server/api/mcp/v1/types/types.go
@@ -28,8 +28,8 @@ var (
 	AddFrameworkToolOutputSchema                    = mcp.MustUnmarshalSchema(`{"type":"object","required":["framework"],"properties":{"framework":{"type":"object","required":["id","organization_id","name","created_at","updated_at"],"properties":{"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"description":{"description":"Framework description"},"id":{"type":"string","format":"string"},"name":{"type":"string","description":"Framework name"},"organization_id":{"type":"string","format":"string"},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	AddMeasureToolInputSchema                       = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id","name","category"],"properties":{"category":{"type":"string","description":"Measure category"},"description":{"type":"string","description":"Measure description"},"name":{"type":"string","description":"Measure name"},"organization_id":{"type":"string","format":"string"}}}`)
 	AddMeasureToolOutputSchema                      = mcp.MustUnmarshalSchema(`{"type":"object","required":["measure"],"properties":{"measure":{"type":"object","required":["id","category","name","state","created_at","updated_at"],"properties":{"category":{"type":"string","description":"Measure category"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"description":{"description":"Measure description"},"id":{"type":"string","format":"string"},"name":{"type":"string","description":"Measure name"},"state":{"type":"string","enum":["NOT_STARTED","IN_PROGRESS","NOT_APPLICABLE","IMPLEMENTED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
-	AddNonconformityToolInputSchema                 = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id","reference_id","audit_id","root_cause","owner_id","status"],"properties":{"audit_id":{"type":"string","format":"string"},"corrective_action":{"type":"string","description":"Corrective action"},"date_identified":{"type":"string","description":"Date identified","format":"date-time"},"description":{"type":"string","description":"Description"},"due_date":{"type":"string","description":"Due date","format":"date-time"},"effectiveness_check":{"type":"string","description":"Effectiveness check"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"status":{"description":"Status","anyOf":[{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},{"type":"null","description":"No status"}]}}}`)
-	AddNonconformityToolOutputSchema                = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformity"],"properties":{"nonconformity":{"type":"object","required":["id","organization_id","reference_id","audit_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"type":"string","format":"string"},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
+	AddNonconformityToolInputSchema                 = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id","reference_id","root_cause","owner_id","status"],"properties":{"audit_id":{"description":"Audit ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No audit"}]},"corrective_action":{"type":"string","description":"Corrective action"},"date_identified":{"type":"string","description":"Date identified","format":"date-time"},"description":{"type":"string","description":"Description"},"due_date":{"type":"string","description":"Due date","format":"date-time"},"effectiveness_check":{"type":"string","description":"Effectiveness check"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"status":{"description":"Status","anyOf":[{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},{"type":"null","description":"No status"}]}}}`)
+	AddNonconformityToolOutputSchema                = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformity"],"properties":{"nonconformity":{"type":"object","required":["id","organization_id","reference_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"description":"Audit ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No audit"}]},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	AddObligationToolInputSchema                    = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id","owner_id","status"],"properties":{"actions_to_be_implemented":{"type":"string","description":"Actions to be implemented"},"area":{"type":"string","description":"Area"},"due_date":{"type":"string","description":"Due date","format":"date-time"},"last_review_date":{"type":"string","description":"Last review date","format":"date-time"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"regulator":{"type":"string","description":"Regulator"},"requirement":{"type":"string","description":"Requirement"},"source":{"type":"string","description":"Source"},"status":{"description":"Status","anyOf":[{"type":"string","enum":["NON_COMPLIANT","PARTIALLY_COMPLIANT","COMPLIANT"]},{"type":"null","description":"No status"}]}}}`)
 	AddObligationToolOutputSchema                   = mcp.MustUnmarshalSchema(`{"type":"object","required":["obligation"],"properties":{"obligation":{"type":"object","required":["id","organization_id","owner_id","status","created_at","updated_at"],"properties":{"actions_to_be_implemented":{"description":"Actions to be implemented"},"area":{"description":"Area"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"due_date":{"description":"Due date","format":"date-time"},"id":{"type":"string","format":"string"},"last_review_date":{"description":"Last review date","format":"date-time"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"regulator":{"description":"Regulator"},"requirement":{"description":"Requirement"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"source":{"description":"Source"},"source_id":{"description":"Source ID"},"status":{"type":"string","enum":["NON_COMPLIANT","PARTIALLY_COMPLIANT","COMPLIANT"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	AddPeopleToolInputSchema                        = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id","full_name","primary_email_address","kind"],"properties":{"additional_email_addresses":{"type":"array","description":"Additional email addresses","items":{"type":"string"}},"contract_end_date":{"type":"string","description":"Contract end date","format":"date-time"},"contract_start_date":{"type":"string","description":"Contract start date","format":"date-time"},"full_name":{"type":"string","description":"Full name"},"kind":{"type":"string","enum":["EMPLOYEE","CONTRACTOR","SERVICE_ACCOUNT"]},"organization_id":{"type":"string","format":"string"},"position":{"type":"string","description":"Position"},"primary_email_address":{"type":"string","description":"Primary email address"}}}`)
@@ -71,7 +71,7 @@ var (
 	GetMeasureToolInputSchema                       = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"id":{"type":"string","format":"string"}}}`)
 	GetMeasureToolOutputSchema                      = mcp.MustUnmarshalSchema(`{"type":"object","required":["measure"],"properties":{"measure":{"type":"object","required":["id","category","name","state","created_at","updated_at"],"properties":{"category":{"type":"string","description":"Measure category"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"description":{"description":"Measure description"},"id":{"type":"string","format":"string"},"name":{"type":"string","description":"Measure name"},"state":{"type":"string","enum":["NOT_STARTED","IN_PROGRESS","NOT_APPLICABLE","IMPLEMENTED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	GetNonconformityToolInputSchema                 = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"id":{"type":"string","format":"string"}}}`)
-	GetNonconformityToolOutputSchema                = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformity"],"properties":{"nonconformity":{"type":"object","required":["id","organization_id","reference_id","audit_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"type":"string","format":"string"},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
+	GetNonconformityToolOutputSchema                = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformity"],"properties":{"nonconformity":{"type":"object","required":["id","organization_id","reference_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"description":"Audit ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No audit"}]},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	GetObligationToolInputSchema                    = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"id":{"type":"string","format":"string"}}}`)
 	GetObligationToolOutputSchema                   = mcp.MustUnmarshalSchema(`{"type":"object","required":["obligation"],"properties":{"obligation":{"type":"object","required":["id","organization_id","owner_id","status","created_at","updated_at"],"properties":{"actions_to_be_implemented":{"description":"Actions to be implemented"},"area":{"description":"Area"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"due_date":{"description":"Due date","format":"date-time"},"id":{"type":"string","format":"string"},"last_review_date":{"description":"Last review date","format":"date-time"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"regulator":{"description":"Regulator"},"requirement":{"description":"Requirement"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"source":{"description":"Source"},"source_id":{"description":"Source ID"},"status":{"type":"string","enum":["NON_COMPLIANT","PARTIALLY_COMPLIANT","COMPLIANT"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	GetPeopleToolInputSchema                        = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"id":{"type":"string","format":"string"}}}`)
@@ -111,7 +111,7 @@ var (
 	ListMeasuresToolInputSchema                     = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id"],"properties":{"cursor":{"type":"string","format":"string"},"filter":{"type":"object","properties":{"query":{"type":"string","description":"Search query"},"state":{"type":"string","enum":["NOT_STARTED","IN_PROGRESS","NOT_APPLICABLE","IMPLEMENTED"]}}},"order_by":{"type":"object","required":["field","direction"],"properties":{"direction":{"type":"string","enum":["ASC","DESC"]},"field":{"type":"string","enum":["CREATED_AT","NAME"]}}},"organization_id":{"type":"string","format":"string"},"size":{"type":"integer","description":"Page size"}}}`)
 	ListMeasuresToolOutputSchema                    = mcp.MustUnmarshalSchema(`{"type":"object","required":["measures"],"properties":{"measures":{"type":"array","items":{"type":"object","required":["id","category","name","state","created_at","updated_at"],"properties":{"category":{"type":"string","description":"Measure category"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"description":{"description":"Measure description"},"id":{"type":"string","format":"string"},"name":{"type":"string","description":"Measure name"},"state":{"type":"string","enum":["NOT_STARTED","IN_PROGRESS","NOT_APPLICABLE","IMPLEMENTED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}},"next_cursor":{"type":"string","format":"string"}}}`)
 	ListNonconformitiesToolInputSchema              = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id"],"properties":{"cursor":{"type":"string","format":"string"},"filter":{"type":"object","properties":{"snapshot_id":{"type":"string","format":"string"}}},"order_by":{"type":"object","required":["field","direction"],"properties":{"direction":{"type":"string","enum":["ASC","DESC"]},"field":{"type":"string","enum":["CREATED_AT","REFERENCE_ID","DATE_IDENTIFIED","DUE_DATE","STATUS"]}}},"organization_id":{"type":"string","format":"string"},"size":{"type":"integer","description":"Page size"}}}`)
-	ListNonconformitiesToolOutputSchema             = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformities"],"properties":{"next_cursor":{"type":"string","format":"string"},"nonconformities":{"type":"array","items":{"type":"object","required":["id","organization_id","reference_id","audit_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"type":"string","format":"string"},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}}`)
+	ListNonconformitiesToolOutputSchema             = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformities"],"properties":{"next_cursor":{"type":"string","format":"string"},"nonconformities":{"type":"array","items":{"type":"object","required":["id","organization_id","reference_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"description":"Audit ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No audit"}]},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}}`)
 	ListObligationsToolInputSchema                  = mcp.MustUnmarshalSchema(`{"type":"object","required":["organization_id"],"properties":{"cursor":{"type":"string","format":"string"},"filter":{"type":"object","properties":{"snapshot_id":{"type":"string","format":"string"}}},"order_by":{"type":"object","required":["field","direction"],"properties":{"direction":{"type":"string","enum":["ASC","DESC"]},"field":{"type":"string","enum":["CREATED_AT","LAST_REVIEW_DATE","DUE_DATE","STATUS"]}}},"organization_id":{"type":"string","format":"string"},"size":{"type":"integer","description":"Page size"}}}`)
 	ListObligationsToolOutputSchema                 = mcp.MustUnmarshalSchema(`{"type":"object","required":["obligations"],"properties":{"next_cursor":{"type":"string","format":"string"},"obligations":{"type":"array","items":{"type":"object","required":["id","organization_id","owner_id","status","created_at","updated_at"],"properties":{"actions_to_be_implemented":{"description":"Actions to be implemented"},"area":{"description":"Area"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"due_date":{"description":"Due date","format":"date-time"},"id":{"type":"string","format":"string"},"last_review_date":{"description":"Last review date","format":"date-time"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"regulator":{"description":"Regulator"},"requirement":{"description":"Requirement"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"source":{"description":"Source"},"source_id":{"description":"Source ID"},"status":{"type":"string","enum":["NON_COMPLIANT","PARTIALLY_COMPLIANT","COMPLIANT"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}}`)
 	ListOrganizationsToolInputSchema                = mcp.MustUnmarshalSchema(`{"type":"object","properties":{"organization_id":{"type":"string","format":"string"}}}`)
@@ -161,7 +161,7 @@ var (
 	UpdateMeasureToolInputSchema                    = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"category":{"type":"string","description":"Measure category"},"description":{"description":"Measure description"},"id":{"type":"string","format":"string"},"name":{"type":"string","description":"Measure name"},"state":{"type":"string","enum":["NOT_STARTED","IN_PROGRESS","NOT_APPLICABLE","IMPLEMENTED"]}}}`)
 	UpdateMeasureToolOutputSchema                   = mcp.MustUnmarshalSchema(`{"type":"object","required":["measure"],"properties":{"measure":{"type":"object","required":["id","category","name","state","created_at","updated_at"],"properties":{"category":{"type":"string","description":"Measure category"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"description":{"description":"Measure description"},"id":{"type":"string","format":"string"},"name":{"type":"string","description":"Measure name"},"state":{"type":"string","enum":["NOT_STARTED","IN_PROGRESS","NOT_APPLICABLE","IMPLEMENTED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	UpdateNonconformityToolInputSchema              = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"audit_id":{"description":"Audit ID","anyOf":[{"type":"string","format":"string"},{"type":"null"}]},"corrective_action":{"description":"Corrective action"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"owner_id":{"description":"Owner ID","anyOf":[{"type":"string","format":"string"},{"type":"null"}]},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]}}}`)
-	UpdateNonconformityToolOutputSchema             = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformity"],"properties":{"nonconformity":{"type":"object","required":["id","organization_id","reference_id","audit_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"type":"string","format":"string"},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
+	UpdateNonconformityToolOutputSchema             = mcp.MustUnmarshalSchema(`{"type":"object","required":["nonconformity"],"properties":{"nonconformity":{"type":"object","required":["id","organization_id","reference_id","root_cause","owner_id","status","created_at","updated_at"],"properties":{"audit_id":{"description":"Audit ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No audit"}]},"corrective_action":{"description":"Corrective action"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"date_identified":{"description":"Date identified","format":"date-time"},"description":{"description":"Description"},"due_date":{"description":"Due date","format":"date-time"},"effectiveness_check":{"description":"Effectiveness check"},"id":{"type":"string","format":"string"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"reference_id":{"type":"string","description":"Reference ID"},"root_cause":{"type":"string","description":"Root cause"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"status":{"type":"string","enum":["OPEN","IN_PROGRESS","CLOSED"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	UpdateObligationToolInputSchema                 = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"actions_to_be_implemented":{"description":"Actions to be implemented"},"area":{"description":"Area"},"due_date":{"description":"Due date","format":"date-time"},"id":{"type":"string","format":"string"},"last_review_date":{"description":"Last review date","format":"date-time"},"owner_id":{"description":"Owner ID","anyOf":[{"type":"string","format":"string"},{"type":"null"}]},"regulator":{"description":"Regulator"},"requirement":{"description":"Requirement"},"source":{"description":"Source"},"status":{"description":"Status","anyOf":[{"type":"string","enum":["NON_COMPLIANT","PARTIALLY_COMPLIANT","COMPLIANT"]},{"type":"null","description":"No status"}]}}}`)
 	UpdateObligationToolOutputSchema                = mcp.MustUnmarshalSchema(`{"type":"object","required":["obligation"],"properties":{"obligation":{"type":"object","required":["id","organization_id","owner_id","status","created_at","updated_at"],"properties":{"actions_to_be_implemented":{"description":"Actions to be implemented"},"area":{"description":"Area"},"created_at":{"type":"string","description":"Creation timestamp","format":"date-time"},"due_date":{"description":"Due date","format":"date-time"},"id":{"type":"string","format":"string"},"last_review_date":{"description":"Last review date","format":"date-time"},"organization_id":{"type":"string","format":"string"},"owner_id":{"type":"string","format":"string"},"regulator":{"description":"Regulator"},"requirement":{"description":"Requirement"},"snapshot_id":{"description":"Snapshot ID","anyOf":[{"type":"string","format":"string"},{"type":"null","description":"No snapshot"}]},"source":{"description":"Source"},"source_id":{"description":"Source ID"},"status":{"type":"string","enum":["NON_COMPLIANT","PARTIALLY_COMPLIANT","COMPLIANT"]},"updated_at":{"type":"string","description":"Update timestamp","format":"date-time"}}}}}`)
 	UpdatePeopleToolInputSchema                     = mcp.MustUnmarshalSchema(`{"type":"object","required":["id"],"properties":{"additional_email_addresses":{"description":"Additional email addresses","anyOf":[{"type":"array","items":{"type":"string"}},{"type":"null","description":"No additional email addresses"}]},"contract_end_date":{"description":"Contract end date","format":"date-time"},"contract_start_date":{"description":"Contract start date","format":"date-time"},"full_name":{"type":"string","description":"Full name"},"id":{"type":"string","format":"string"},"kind":{"type":"string","enum":["EMPLOYEE","CONTRACTOR","SERVICE_ACCOUNT"]},"position":{"description":"Position"},"primary_email_address":{"type":"string","description":"Primary email address"}}}`)
@@ -344,7 +344,7 @@ type AddMeasureOutput struct {
 // AddNonconformityInput represents the schema
 type AddNonconformityInput struct {
 	// Audit ID
-	AuditID gid.GID `json:"audit_id"`
+	AuditID *gid.GID `json:"audit_id,omitempty"`
 	// Corrective action
 	CorrectiveAction *string `json:"corrective_action,omitempty"`
 	// Date identified
@@ -1450,7 +1450,7 @@ type MeasureOrderBy struct {
 // Nonconformity represents the schema
 type Nonconformity struct {
 	// Audit ID
-	AuditID gid.GID `json:"audit_id"`
+	AuditID *gid.GID `json:"audit_id,omitempty"`
 	// Corrective action
 	CorrectiveAction *string `json:"corrective_action,omitempty"`
 	// Creation timestamp
@@ -1965,7 +1965,7 @@ type UpdateMeasureOutput struct {
 // UpdateNonconformityInput represents the schema
 type UpdateNonconformityInput struct {
 	// Audit ID
-	AuditID *gid.GID `json:"audit_id,omitempty"`
+	AuditID mcp.Omittable[*gid.GID] `json:"audit_id,omitempty"`
 	// Corrective action
 	CorrectiveAction mcp.Omittable[*string] `json:"corrective_action,omitempty"`
 	// Date identified