From 0759b29ca2afc54de3e932c865b2d7e749cbb7ec Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Thu, 22 Feb 2024 14:40:16 +0800
Subject: [PATCH] feat: retrial node

---
 .../nodes/knowledge-retrieval/mock.ts         | 19 +++++++++++++++++++
 .../nodes/knowledge-retrieval/node.tsx        | 16 +++++++++++++++-
 .../nodes/knowledge-retrieval/types.ts        | 16 ++++++++++++++++
 3 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 web/app/components/workflow/nodes/knowledge-retrieval/mock.ts
 create mode 100644 web/app/components/workflow/nodes/knowledge-retrieval/types.ts

diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/mock.ts b/web/app/components/workflow/nodes/knowledge-retrieval/mock.ts
new file mode 100644
index 0000000000..fbabf5de33
--- /dev/null
+++ b/web/app/components/workflow/nodes/knowledge-retrieval/mock.ts
@@ -0,0 +1,19 @@
+import type { KnowledgeRetrievalNodeType } from './types'
+import { RETRIEVE_TYPE } from '@/types/app'
+
+export const mockData: KnowledgeRetrievalNodeType = {
+  type: 'KnowledgeRetrieval',
+  desc: 'xxx',
+  title: 'KnowledgeRetrieval',
+  query_variable_selector: ['aaa', 'name'],
+  dataset_ids: ['1'],
+  retrieval_mode: RETRIEVE_TYPE.oneWay,
+  multiple_retrieval_config: {
+    top_k: 10,
+    score_threshold: 0.5,
+    reranking_model: {
+      provider: '',
+      model: '',
+    },
+  },
+}
diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/node.tsx b/web/app/components/workflow/nodes/knowledge-retrieval/node.tsx
index cae03dbf53..61b87aaab2 100644
--- a/web/app/components/workflow/nodes/knowledge-retrieval/node.tsx
+++ b/web/app/components/workflow/nodes/knowledge-retrieval/node.tsx
@@ -1,8 +1,22 @@
 import type { FC } from 'react'
+import { Folder } from '@/app/components/base/icons/src/vender/solid/files'
 
 const Node: FC = () => {
   return (
-    <div>knowledge-retrieval</div>
+    <div className='px-3'>
+      <div className='space-y-0.5'>
+        {['product Doc', 'Text completion'].map(name => (
+          <div key={name} className='flex items-center h-[26px] bg-gray-100 rounded-md  px-1 text-xs font-normal text-gray-700'>
+            <div className='mr-1 shrink-0 p-1 bg-[#F5F8FF] rounded-md border-[0.5px] border-[#E0EAFF]'>
+              <Folder className='w-3 h-3 text-[#444CE7]' />
+            </div>
+            <div className='text-xs font-normal text-gray-700'>
+              {name}
+            </div>
+          </div>
+        ))}
+      </div>
+    </div>
   )
 }
 
diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/types.ts b/web/app/components/workflow/nodes/knowledge-retrieval/types.ts
new file mode 100644
index 0000000000..15d941c242
--- /dev/null
+++ b/web/app/components/workflow/nodes/knowledge-retrieval/types.ts
@@ -0,0 +1,16 @@
+import type { CommonNodeType, ValueSelector } from '@/app/components/workflow/types'
+import type { RETRIEVE_TYPE } from '@/types/app'
+
+export type KnowledgeRetrievalNodeType = CommonNodeType & {
+  query_variable_selector: ValueSelector
+  dataset_ids: string[]
+  retrieval_mode: RETRIEVE_TYPE
+  multiple_retrieval_config?: {
+    top_k: number
+    score_threshold: number
+    reranking_model: {
+      provider: string
+      model: string
+    }
+  }
+}