From 6437de4def41160f3e97eb8dcec91038dae9d04e Mon Sep 17 00:00:00 2001
From: BennyKok <itechbenny@gmail.com>
Date: Wed, 24 Jan 2024 02:11:56 +0000
Subject: [PATCH] fix: Add createSelectSchema for workflowVersionTable

---
 web/src/routes/registerGetWorkflow.ts | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/web/src/routes/registerGetWorkflow.ts b/web/src/routes/registerGetWorkflow.ts
index be17571..ca682da 100644
--- a/web/src/routes/registerGetWorkflow.ts
+++ b/web/src/routes/registerGetWorkflow.ts
@@ -1,8 +1,9 @@
-import { workflowVersionSchema } from "@/db/schema";
+import { workflowVersionSchema, workflowVersionTable } from "@/db/schema";
 import type { App } from "@/routes/app";
 import { authError } from "@/routes/authError";
 import { getWorkflowVersion } from "@/server/crudWorkflow";
 import { z, createRoute } from "@hono/zod-openapi";
+import { createSelectSchema } from "drizzle-zod";
 
 const route = createRoute({
   method: "get",
@@ -19,7 +20,20 @@ const route = createRoute({
     200: {
       content: {
         "application/json": {
-          schema: workflowVersionSchema,
+          schema: createSelectSchema(workflowVersionTable, {
+            workflow_api: (schema) =>
+              schema.workflow_api.openapi({
+                type: "object",
+              }),
+            workflow: (schema) =>
+              schema.workflow.openapi({
+                type: "object",
+              }),
+            snapshot: (schema) =>
+              schema.snapshot.openapi({
+                type: "object",
+              }),
+          }),
         },
       },
       description: "Retrieve the output",