Skip to content

Commit 99329f4

Browse files
committed
feat: Vector retrieval matches datasource
1 parent 26502e7 commit 99329f4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backend/apps/chat/task/llm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -432,6 +432,7 @@ def select_datasource(self):
432432
if not ignore_auto_select:
433433
if settings.EMBEDDING_ENABLED:
434434
ds = get_ds_embedding(self.session, self.current_user, _ds_list, self.chat_question.question)
435+
yield {'content': '{"id":' + ds.id + '}'}
435436
else:
436437
_ds_list_dict = []
437438
for _ds in _ds_list:

0 commit comments

Comments
 (0)