diff --git a/src/lib/components/workspace/Models/Knowledge.svelte b/src/lib/components/workspace/Models/Knowledge.svelte
index cdc856378a..d4e7de792d 100644
--- a/src/lib/components/workspace/Models/Knowledge.svelte
+++ b/src/lib/components/workspace/Models/Knowledge.svelte
@@ -3,9 +3,7 @@
 	import Selector from './Knowledge/Selector.svelte';
 	import FileItem from '$lib/components/common/FileItem.svelte';
 
-	export let selectedKnowledge = [];
-	export let collections = [];
-
+	export let selectedItems = [];
 	const i18n = getContext('i18n');
 </script>
 
@@ -19,9 +17,9 @@
 	</div>
 
 	<div class="flex flex-col">
-		{#if selectedKnowledge?.length > 0}
+		{#if selectedItems?.length > 0}
 			<div class=" flex flex-wrap items-center gap-2 mt-2">
-				{#each selectedKnowledge as file, fileIdx}
+				{#each selectedItems as file, fileIdx}
 					<FileItem
 						{file}
 						item={file}
@@ -33,7 +31,7 @@
 							: (file?.type ?? 'Collection')}
 						dismissible
 						on:dismiss={(e) => {
-							selectedKnowledge = selectedKnowledge.filter((_, idx) => idx !== fileIdx);
+							selectedItems = selectedItems.filter((_, idx) => idx !== fileIdx);
 						}}
 					/>
 				{/each}
@@ -45,9 +43,9 @@
 				on:select={(e) => {
 					const item = e.detail;
 
-					if (!selectedKnowledge.find((k) => k.id === item.id)) {
-						selectedKnowledge = [
-							...selectedKnowledge,
+					if (!selectedItems.find((k) => k.id === item.id)) {
+						selectedItems = [
+							...selectedItems,
 							{
 								...item
 							}
diff --git a/src/lib/components/workspace/Models/ModelEditor.svelte b/src/lib/components/workspace/Models/ModelEditor.svelte
index c7b31fd164..68b661c64f 100644
--- a/src/lib/components/workspace/Models/ModelEditor.svelte
+++ b/src/lib/components/workspace/Models/ModelEditor.svelte
@@ -19,6 +19,7 @@
 	import { toast } from 'svelte-sonner';
 	import Spinner from '$lib/components/common/Spinner.svelte';
 	import XMark from '$lib/components/icons/XMark.svelte';
+	import { getNoteList } from '$lib/apis/notes';
 
 	const i18n = getContext('i18n');
 
@@ -182,7 +183,10 @@
 	onMount(async () => {
 		await tools.set(await getTools(localStorage.token));
 		await functions.set(await getFunctions(localStorage.token));
-		await knowledgeCollections.set(await getKnowledgeBases(localStorage.token));
+		await knowledgeCollections.set([
+			...(await getNoteList(localStorage.token)),
+			...(await getKnowledgeBases(localStorage.token))
+		]);
 
 		// Scroll to top 'workspace-container' element
 		const workspaceContainer = document.getElementById('workspace-container');
@@ -694,7 +698,7 @@
 					<hr class=" border-gray-100 dark:border-gray-850 my-1.5" />
 
 					<div class="my-2">
-						<Knowledge bind:selectedKnowledge={knowledge} collections={$knowledgeCollections} />
+						<Knowledge bind:selectedItems={knowledge} />
 					</div>
 
 					<div class="my-2">