|
|
@@ -142,7 +142,7 @@ def get_base_prompt():
|
|
|
'custom_debugging': chat_service.VOLCENGINE_BOT_DEEPSEEK_V3_SEARCH,
|
|
|
}
|
|
|
if scene not in prompt_map:
|
|
|
- return wrap_response(404, msg="scene not found")
|
|
|
+ return wrap_response(404, msg='scene not found')
|
|
|
data = {
|
|
|
'model_name': model_map[scene],
|
|
|
'content': prompt_map[scene]
|
|
|
@@ -252,7 +252,7 @@ def handle_bad_request(e):
|
|
|
return wrap_response(400, msg='Bad Request: {}'.format(e.description))
|
|
|
|
|
|
|
|
|
-if __name__ == "__main__":
|
|
|
+if __name__ == '__main__':
|
|
|
parser = ArgumentParser()
|
|
|
parser.add_argument('--prod', action='store_true')
|
|
|
parser.add_argument('--host', default='127.0.0.1')
|
|
|
@@ -266,20 +266,17 @@ if __name__ == "__main__":
|
|
|
|
|
|
user_db_config = config['storage']['user']
|
|
|
staff_db_config = config['storage']['staff']
|
|
|
- user_manager = MySQLUserManager(
|
|
|
- user_db_config['mysql'], user_db_config['table'], staff_db_config['table']
|
|
|
- )
|
|
|
+ user_manager = MySQLUserManager(user_db_config['mysql'], user_db_config['table'], staff_db_config['table'])
|
|
|
app.user_manager = user_manager
|
|
|
|
|
|
wecom_db_config = config['storage']['user_relation']
|
|
|
user_relation_manager = MySQLUserRelationManager(
|
|
|
- user_db_config['mysql'],
|
|
|
- wecom_db_config['mysql'],
|
|
|
+ user_db_config['mysql'],wecom_db_config['mysql'],
|
|
|
config['storage']['staff']['table'],
|
|
|
user_db_config['table'],
|
|
|
wecom_db_config['table']['staff'],
|
|
|
wecom_db_config['table']['relation'],
|
|
|
- wecom_db_config['table']['user'],
|
|
|
+ wecom_db_config['table']['user']
|
|
|
)
|
|
|
app.user_relation_manager = user_relation_manager
|
|
|
app.history_dialogue_service = HistoryDialogueService(
|
luojunhui