diff --git a/web/src/pages/agent/canvas/node/retrieval-node.tsx b/web/src/pages/agent/canvas/node/retrieval-node.tsx
index 04829b5b7..35c07799a 100644
--- a/web/src/pages/agent/canvas/node/retrieval-node.tsx
+++ b/web/src/pages/agent/canvas/node/retrieval-node.tsx
@@ -23,7 +23,7 @@ function InnerRetrievalNode({
   isConnectable = true,
   selected,
 }: NodeProps<BaseNode<RetrievalFormSchemaType>>) {
-  const knowledgeBaseIds: string[] = get(data, 'form.kb_ids', []);
+  const knowledgeBaseIds: string[] = get(data, 'form.dataset_ids', []);
   const memoryIds: string[] = get(data, 'form.memory_ids', []);
   const { list: knowledgeList } = useFetchKnowledgeList(true);
 
diff --git a/web/src/pages/agent/form/retrieval-form/next.tsx b/web/src/pages/agent/form/retrieval-form/next.tsx
index 945c2f277..44a6fe9ad 100644
--- a/web/src/pages/agent/form/retrieval-form/next.tsx
+++ b/web/src/pages/agent/form/retrieval-form/next.tsx
@@ -46,7 +46,7 @@ export const RetrievalPartialSchema = {
   keywords_similarity_weight: z.coerce.number(),
   top_n: z.coerce.number(),
   top_k: z.coerce.number(),
-  database_ids: z.array(z.string()),
+  dataset_ids: z.array(z.string()),
   rerank_id: z.string(),
   empty_response: z.string(),
   cross_languages: z.array(z.string()),