feat: update ops trace (#7102)
This commit is contained in:
parent
7201b56a6d
commit
425174e82f
13 changed files with 118 additions and 129 deletions
|
|
@ -1,4 +1,5 @@
|
|||
from datetime import datetime
|
||||
from enum import Enum
|
||||
from typing import Any, Optional, Union
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, field_validator
|
||||
|
|
@ -105,4 +106,15 @@ trace_info_info_map = {
|
|||
'DatasetRetrievalTraceInfo': DatasetRetrievalTraceInfo,
|
||||
'ToolTraceInfo': ToolTraceInfo,
|
||||
'GenerateNameTraceInfo': GenerateNameTraceInfo,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
class TraceTaskName(str, Enum):
|
||||
CONVERSATION_TRACE = 'conversation'
|
||||
WORKFLOW_TRACE = 'workflow'
|
||||
MESSAGE_TRACE = 'message'
|
||||
MODERATION_TRACE = 'moderation'
|
||||
SUGGESTED_QUESTION_TRACE = 'suggested_question'
|
||||
DATASET_RETRIEVAL_TRACE = 'dataset_retrieval'
|
||||
TOOL_TRACE = 'tool'
|
||||
GENERATE_NAME_TRACE = 'generate_conversation_name'
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue