kanshan/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-22 03:32:58 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

fix(workflow): pass CreatorUserRole into knowledge retrieval dataset queries

Browse Source 
chore: run make lint and make type-check (pass)

Tests: make lint; make type-check.
This commit is contained in:
-LAN- 2026-01-06 19:31:42 +08:00
parent d6bad76fa6
commit b7baaaf904
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -55,6 +55,7 @@ from extensions.ext_database import db
from extensions.ext_redis import redis_client
from libs.json_in_md_parser import parse_and_check_json_markdown
from models.dataset import Dataset, DatasetMetadata, Document, RateLimitLog
from models.enums import CreatorUserRole, UserFrom
from services.feature_service import FeatureService
from .entities import KnowledgeRetrievalNodeData
@ -268,6 +269,7 @@ class KnowledgeRetrievalNode(LLMUsageTrackingMixin, Node[KnowledgeRetrievalNodeD
usage = self._merge_usage(usage, metadata_usage)
all_documents = []
dataset_retrieval = DatasetRetrieval()
creator_user_role = CreatorUserRole.ACCOUNT if self.user_from == UserFrom.ACCOUNT else CreatorUserRole.END_USER
if str(node_data.retrieval_mode) == DatasetRetrieveConfigEntity.RetrieveStrategy.SINGLE and query:
# fetch model config
if node_data.single_retrieval_config is None:
@ -292,7 +294,7 @@ class KnowledgeRetrievalNode(LLMUsageTrackingMixin, Node[KnowledgeRetrievalNodeD
tenant_id=self.tenant_id,
user_id=self.user_id,
app_id=self.app_id,
user_from=self.user_from.value,
creator_user_role=creator_user_role,
query=query,
model_config=model_config,
model_instance=model_instance,
@ -334,7 +336,7 @@ class KnowledgeRetrievalNode(LLMUsageTrackingMixin, Node[KnowledgeRetrievalNodeD
app_id=self.app_id,
tenant_id=self.tenant_id,
user_id=self.user_id,
user_from=self.user_from.value,
creator_user_role=creator_user_role,
available_datasets=available_datasets,
query=query,
top_k=node_data.multiple_retrieval_config.top_k,