jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: update ops trace (#7102)

Browse source
This commit is contained in:
Joe 2024-08-09 15:22:16 +08:00 • committed by GitHub
commit 425174e82f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 118 additions and 129 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/core/rag/retrieval/dataset_retrieval.py
View file

@ -14,7 +14,8 @@ from core.model_manager import ModelInstance, ModelManager
from core.model_runtime.entities.message_entities import PromptMessageTool
from core.model_runtime.entities.model_entities import ModelFeature, ModelType
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.ops.ops_trace_manager import TraceQueueManager, TraceTask, TraceTaskName
from core.ops.entities.trace_entity import TraceTaskName
from core.ops.ops_trace_manager import TraceQueueManager, TraceTask
from core.ops.utils import measure_time
from core.rag.data_post_processor.data_post_processor import DataPostProcessor
from core.rag.datasource.keyword.jieba.jieba_keyword_table_handler import JiebaKeywordTableHandler