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/llm_generator/llm_generator.py
View file

@ -14,7 +14,8 @@ from core.model_manager import ModelManager
from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage
from core.model_runtime.entities.model_entities import ModelType
from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
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.prompt.utils.prompt_template_parser import PromptTemplateParser