From 99329f400bf9583832dfdab1a285d7a8ece5b1ff Mon Sep 17 00:00:00 2001
From: junjun <junjie.xia@fit2cloud.com>
Date: Thu, 18 Sep 2025 17:09:35 +0800
Subject: [PATCH] feat: Vector retrieval matches datasource

---
 backend/apps/chat/task/llm.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py
index 63354fa8..4d72481d 100644
--- a/backend/apps/chat/task/llm.py
+++ b/backend/apps/chat/task/llm.py
@@ -432,6 +432,7 @@ class LLMService:
         if not ignore_auto_select:
             if settings.EMBEDDING_ENABLED:
                 ds = get_ds_embedding(self.session, self.current_user, _ds_list, self.chat_question.question)
+                yield {'content': '{"id":' + ds.id + '}'}
             else:
                 _ds_list_dict = []
                 for _ds in _ds_list: