From f64c242ee511e23d5c2dc2567be3aeca6ab33c78 Mon Sep 17 00:00:00 2001
From: Jiseop Han <jiseop.han@voyagerx.com>
Date: Tue, 23 Jan 2024 00:01:01 +0900
Subject: [PATCH 1/2] fix coroutine error

---
 custom_routes.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/custom_routes.py b/custom_routes.py
index a6d66c2..99c4c1e 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -203,7 +203,10 @@ async def send_json_override(self, event, data, sid=None):
     prompt_id = data.get('prompt_id')
 
     # now we send everything
-    await asyncio.wait([send(event, data), self.send_json_original(event, data, sid)])
+    await asyncio.wait([
+        asyncio.create_task(send(event, data)),
+        asyncio.create_task(self.send_json_original(event, data, sid))
+    ])
 
     if event == 'execution_start':
         update_run(prompt_id, Status.RUNNING)

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 2/2] 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",