wanyaokun
80 lines
1.7 KiB
Python
80 lines
1.7 KiB
Python
from pydantic import BaseModel
|
|
import os
|
|
from enum import Enum
|
|
|
|
class BaseConfig(BaseModel):
|
|
projectInfo:str = os.getenv("PROJECT_TITLE","您好，我是博微工程理解小助手，您可以问我有关[线路工程]工程数据的相关问题！")
|
|
|
|
def ParamterCfg(self):
|
|
questions = os.getenv("CONVERSATION_STARTERS", "dev")
|
|
return{
|
|
"opening_statement": self.projectInfo,
|
|
"suggested_questions": questions.split('\n'),
|
|
"suggested_questions_after_answer": {
|
|
"enabled": False
|
|
},
|
|
"speech_to_text": {
|
|
"enabled": False
|
|
},
|
|
"text_to_speech": {
|
|
"enabled": False,
|
|
"language": "",
|
|
"voice": ""
|
|
},
|
|
"retriever_resource": {
|
|
"enabled": True
|
|
},
|
|
"annotation_reply": {
|
|
"enabled": False
|
|
},
|
|
"more_like_this": {
|
|
"enabled": False
|
|
},
|
|
"user_input_form": [],
|
|
"sensitive_word_avoidance": {
|
|
"enabled": False
|
|
},
|
|
"file_upload": {
|
|
"image": {
|
|
"enabled": False,
|
|
"number_limits": 3,
|
|
"transfer_methods": [
|
|
"remote_url"
|
|
]
|
|
}
|
|
},
|
|
"system_parameters": {
|
|
"image_file_size_limit": "10"
|
|
}
|
|
}
|
|
|
|
def ConversationCfg(self):
|
|
return{
|
|
"id": "",
|
|
'user_id':'',
|
|
"name": "",
|
|
"inputs": {},
|
|
"status": "normal",
|
|
"introduction": self.projectInfo,
|
|
"created_at":''
|
|
}
|
|
|
|
@classmethod
|
|
def MessageCfg(cls):
|
|
return {
|
|
"id": "",
|
|
'user_id':'',
|
|
"conversation_id": "",
|
|
"inputs": {},
|
|
"query": "",
|
|
"answer": ""
|
|
}
|
|
|
|
|
|
class ChatEventType(str, Enum):
|
|
WORKFLOW_START = "workflow_started"
|
|
WORKFLOW_FINISHED = "workflow_finished"
|
|
NODE_START = "node_started"
|
|
NODE_FINISHED = "node_finished"
|
|
MESSAGE = "message"
|
|
MESSAGE_END = "message_end" |