From 8ed825aa2b46d08de17543f91b9b3247e654c3f6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A1n=20Jakub=20Nani=C5=A1ta?=
 <jan.jakub.nanista@gmail.com>
Date: Wed, 17 Jul 2024 11:34:07 -0700
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=B9=20CLI:=20Add=20missing=20schema=20?=
 =?UTF-8?q?[3/N]=20(#707)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .changeset/selfish-tables-shout.md      |  5 +++++
 packages/ua-devtools/src/oapp/schema.ts | 14 +++++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100644 .changeset/selfish-tables-shout.md

diff --git a/.changeset/selfish-tables-shout.md b/.changeset/selfish-tables-shout.md
new file mode 100644
index 000000000..c92e51d57
--- /dev/null
+++ b/.changeset/selfish-tables-shout.md
@@ -0,0 +1,5 @@
+---
+"@layerzerolabs/ua-devtools": patch
+---
+
+Add OAppOmniGraphSchema to ua-devtools
diff --git a/packages/ua-devtools/src/oapp/schema.ts b/packages/ua-devtools/src/oapp/schema.ts
index 26d30e9d8..1fd722377 100644
--- a/packages/ua-devtools/src/oapp/schema.ts
+++ b/packages/ua-devtools/src/oapp/schema.ts
@@ -1,5 +1,12 @@
 import { z } from 'zod'
-import { AddressSchema, UIntBigIntSchema, UIntNumberSchema } from '@layerzerolabs/devtools'
+import {
+    AddressSchema,
+    createOmniEdgeSchema,
+    createOmniGraphSchema,
+    createOmniNodeSchema,
+    UIntBigIntSchema,
+    UIntNumberSchema,
+} from '@layerzerolabs/devtools'
 import { Uln302ExecutorConfigSchema, Uln302UlnUserConfigSchema, TimeoutSchema } from '@layerzerolabs/protocol-devtools'
 import {
     ExecutorComposeOption,
@@ -88,3 +95,8 @@ export const OAppEdgeConfigSchema = z
     // We'll pass all unknown properties through without validating them
     .passthrough()
     .partial() satisfies z.ZodSchema<OAppEdgeConfig, z.ZodTypeDef, unknown>
+
+export const OAppOmniGraphSchema = createOmniGraphSchema(
+    createOmniNodeSchema(OAppNodeConfigSchema),
+    createOmniEdgeSchema(OAppEdgeConfigSchema)
+)