From db78b91ec28e4b54dd81473495dabcd0b800e984 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Wed, 13 Mar 2024 15:47:26 +0800
Subject: [PATCH] feat: http output

---
 web/app/components/workflow/constants.ts          | 15 +++++++++++++++
 .../nodes/_base/components/variable/utils.ts      |  6 ++++++
 2 files changed, 21 insertions(+)

diff --git a/web/app/components/workflow/constants.ts b/web/app/components/workflow/constants.ts
index 691c987d71..c0d06c6b58 100644
--- a/web/app/components/workflow/constants.ts
+++ b/web/app/components/workflow/constants.ts
@@ -297,3 +297,18 @@ export const COMPLETION_QUESTION_CLASSIFIER_OUTPUT_STRUCT = [
   },
   ...QUESTION_CLASSIFIER_OUTPUT_STRUCT_COMMON,
 ]
+
+export const HTTP_REQUEST_OUTPUT_STRUCT: Var[] = [
+  {
+    variable: 'body',
+    type: VarType.string,
+  },
+  {
+    variable: 'status_code',
+    type: VarType.string,
+  },
+  {
+    variable: 'headers',
+    type: VarType.string,
+  },
+]
diff --git a/web/app/components/workflow/nodes/_base/components/variable/utils.ts b/web/app/components/workflow/nodes/_base/components/variable/utils.ts
index 4963570ada..150e6d6d4f 100644
--- a/web/app/components/workflow/nodes/_base/components/variable/utils.ts
+++ b/web/app/components/workflow/nodes/_base/components/variable/utils.ts
@@ -5,6 +5,7 @@ import type { NodeOutPutVar } from '@/app/components/workflow/types'
 import {
   CHAT_QUESTION_CLASSIFIER_OUTPUT_STRUCT,
   COMPLETION_QUESTION_CLASSIFIER_OUTPUT_STRUCT,
+  HTTP_REQUEST_OUTPUT_STRUCT,
   KNOWLEDGE_RETRIEVAL_OUTPUT_STRUCT,
   LLM_OUTPUT_STRUCT,
   SUPPORT_OUTPUT_VARS_NODE,
@@ -71,6 +72,11 @@ const formatItem = (item: any, isChatMode: boolean): NodeOutPutVar => {
       res.vars = isChatMode ? CHAT_QUESTION_CLASSIFIER_OUTPUT_STRUCT : COMPLETION_QUESTION_CLASSIFIER_OUTPUT_STRUCT
       break
     }
+
+    case BlockEnum.HttpRequest: {
+      res.vars = HTTP_REQUEST_OUTPUT_STRUCT
+      break
+    }
   }
 
   return res