knowledge_dify.py 732 B

12345678910111213141516171819202122232425262728293031323334
  1. from base.config import config_handler
  2. from logger.loggering import server_logger
  3. from utils.common import handler_err
  4. from utils.request_tool import KnowledgeDifyAction
  5. class KnowledgeDify:
  6. """
  7. 私有知识库查询处理(dify)
  8. """
  9. def __init__(self):
  10. self.knowledge_dify_action = KnowledgeDifyAction()
  11. def get_request_knowledge_dify(self , trace_id: str, question: str):
  12. """
  13. 获取知识库内容
  14. :param question: 问题
  15. :return:
  16. """
  17. record_list = self.knowledge_dify_action.get_request_knowledge_retrieve_list(
  18. params={"query": question} , trace_id=trace_id
  19. )
  20. return record_list