graph_router.py 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. import sys,os
  2. from agent.cdss.capbility import CDSSCapability
  3. from agent.cdss.models.schemas import CDSSInput, CDSSInt, CDSSText
  4. from model.response import StandardResponse
  5. current_path = os.getcwd()
  6. sys.path.append(current_path)
  7. import time
  8. from fastapi import APIRouter, Depends, Query
  9. from typing import Optional, List
  10. import sys
  11. sys.path.append('..')
  12. from utils.agent import call_chat_api,get_conversation_id
  13. import json
  14. router = APIRouter(prefix="/graph", tags=["Knowledge Graph"])
  15. @router.get("/nodes/recommend", response_model=StandardResponse)
  16. async def recommend(
  17. chief: str
  18. ):
  19. start_time = time.time()
  20. app_id = "256fd853-60b0-4357-b11b-8114b4e90ae0"
  21. conversation_id = get_conversation_id(app_id)
  22. result = call_chat_api(app_id, conversation_id, chief)
  23. json_data = json.loads(result)
  24. keyword = " ".join(json_data["symptoms"])
  25. result = await neighbor_search(keyword=keyword, neighbor_type='Check', limit=10)
  26. end_time = time.time()
  27. print(f"recommend执行完成,耗时:{end_time - start_time:.2f}秒")
  28. return result;
  29. @router.get("/nodes/neighbor_search", response_model=StandardResponse)
  30. async def neighbor_search(
  31. keyword: str = Query(..., min_length=2),
  32. limit: int = Query(10, ge=1, le=100),
  33. node_type: Optional[str] = Query(None),
  34. neighbor_type: Optional[str] = Query(None),
  35. min_degree: Optional[int] = Query(None)
  36. ):
  37. """
  38. 根据关键词和属性过滤条件搜索图谱节点
  39. """
  40. try:
  41. print(f"开始执行neighbor_search,参数:keyword={keyword}, limit={limit}, node_type={node_type}, neighbor_type={neighbor_type}, min_degree={min_degree}")
  42. keywords = keyword.split(" ")
  43. record = CDSSInput(
  44. #pat_age=CDSSInt(type="month", value=24),
  45. #pat_sex=CDSSText(type="sex", value="女"),
  46. chief_complaint=keywords,
  47. )
  48. # 使用从main.py导入的capability实例处理CDSS逻辑
  49. output = capability.process(input=record)
  50. print(output.diagnosis.value)
  51. return StandardResponse(
  52. success=True,
  53. data={"可能诊断":output.diagnosis.value,"可能诊断2":output.diagnosis2.value,"推荐检验":output.checks.value,"症状":keywords}
  54. )
  55. except Exception as e:
  56. print(e)
  57. raise e
  58. return StandardResponse(
  59. success=False,
  60. error_code=500,
  61. error_msg=str(e)
  62. )
  63. capability = CDSSCapability()
  64. #def get_capability():
  65. #from main import capability
  66. #return capability
  67. graph_router = router