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

Feat/fix ops trace (#5672)

Browse source 
Co-authored-by: takatost <takatost@gmail.com>
This commit is contained in:
Joe 2024-06-28 00:24:37 +08:00 • committed by GitHub
commit e8b8f6c6dd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
17 changed files with 372 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -12,7 +12,7 @@ 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 TraceTask, TraceTaskName
from core.ops.ops_trace_manager import TraceQueueManager, TraceTask, TraceTaskName
from core.ops.utils import measure_time
from core.rag.datasource.retrieval_service import RetrievalService
from core.rag.models.document import Document
@ -357,7 +357,7 @@ class DatasetRetrieval:
db.session.commit()
# get tracing instance
trace_manager = self.application_generate_entity.trace_manager if self.application_generate_entity else None
trace_manager: TraceQueueManager = self.application_generate_entity.trace_manager if self.application_generate_entity else None
if trace_manager:
trace_manager.add_trace_task(
TraceTask(