tool_utils.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. import time
  2. from math import log
  3. import os
  4. from dotenv import load_dotenv
  5. from foundation.core_enums import ErrorCodeEnum
  6. from functools import wraps
  7. from foundation.logger.loggering import server_logger
  8. from foundation.utils.common import handler_err
  9. from foundation.base.config import config_handler
  10. # 获取当前文件的目录
  11. current_dir = os.path.dirname(__file__)
  12. # 构建到 .env 的相对路径
  13. conf_file_path = os.path.join(current_dir , '../', '.env')
  14. #server_logger.info(f"当前目录: {conf_file_path}")
  15. # 加载环境变量
  16. load_dotenv(dotenv_path=conf_file_path)
  17. def verify_param(param: dict):
  18. """
  19. 验证请求参数
  20. """
  21. input_data = param.get("input")
  22. session_id = param.get("config").get("session_id")
  23. if input_data is None:
  24. raise ValueError(ErrorCodeEnum.INPUT_INFO_EMPTY.__str__)
  25. if session_id is None:
  26. raise ValueError(ErrorCodeEnum.SESSION_ID_EMPTY.__str__)
  27. def get_system_prompt() -> str:
  28. """
  29. 获取系统提示语
  30. """
  31. system_prompt = config_handler.get("system", "SYSTEM_PROMPT")
  32. server_logger.info(f"获取系统提示语: {system_prompt}")
  33. return str(system_prompt)