From c094be82bbf91027e3db3479edef6fb0dbe74f60 Mon Sep 17 00:00:00 2001
From: hhd <17835559578@163.com>
Date: Fri, 14 Mar 2025 09:52:00 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=8A=9F=E8=83=BD=E6=96=B0=E5=A2=9E?=
 =?UTF-8?q?=E3=80=91Tinyflow=E7=BB=84=E4=BB=B6=E6=94=AF=E6=8C=81=E4=BC=A0?=
 =?UTF-8?q?=E5=85=A5provider=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 demos/vue/src/App.vue         | 10 ++++++++++
 packages/vue/src/Tinyflow.vue | 19 +++++++++++++++++--
 2 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/demos/vue/src/App.vue b/demos/vue/src/App.vue
index 4e996d1..027ea5d 100644
--- a/demos/vue/src/App.vue
+++ b/demos/vue/src/App.vue
@@ -4,6 +4,7 @@
             :className="'custom-class'"
             :style="{ width: '100%', height: '100%' }"
             :data="initialData"
+            :provider="provider"
         />
     </div>
 </template>
@@ -12,6 +13,15 @@
 import { Tinyflow } from '@tinyflow-ai/vue';
 import { ref } from 'vue';
 
+const provider = {
+    llm: ()  => [
+        {
+            value: 'llm',
+            label: 'llm',
+        }
+    ],
+    knowledge: () => []
+}
 const initialData = ref({
     nodes: [],
     edges: []
diff --git a/packages/vue/src/Tinyflow.vue b/packages/vue/src/Tinyflow.vue
index 62ede63..40746a4 100644
--- a/packages/vue/src/Tinyflow.vue
+++ b/packages/vue/src/Tinyflow.vue
@@ -3,7 +3,7 @@
 </template>
 
 <script setup lang="ts">
-import { Tinyflow as TinyflowNative } from '@tinyflow-ai/ui';
+import { Item, Tinyflow as TinyflowNative } from '@tinyflow-ai/ui';
 import '@tinyflow-ai/ui/dist/index.css';
 import { onMounted, onUnmounted, ref } from 'vue';
 
@@ -11,16 +11,31 @@ const props = defineProps<{
     className?: string;
     style?: Record<string, string>;
     data?: Record<string, any>;
+    provider?: {
+        llm?: () => Item[] | Promise<Item[]>;
+        knowledge?: () => Item[] | Promise<Item[]>;
+    };
 }>();
 
 const divRef = ref<HTMLDivElement | null>(null);
 let tinyflow: TinyflowNative | null = null;
+// 定义默认的 provider 方法
+const defaultProvider = {
+    llm: () => [] as Item[],
+    knowledge: () => [] as Item[],
+};
 
 onMounted(() => {
     if (divRef.value) {
+        // 合并默认 provider 和传入的 props.provider
+        const mergedProvider = {
+            ...defaultProvider,
+            ...props.provider,
+        };
         tinyflow = new TinyflowNative({
             element: divRef.value as Element,
-            data: props.data || {}
+            data: props.data || {},
+            provider: mergedProvider,
         });
     }
 });
-- 
Gitee