代码提交

router/graph_router.py

@@ -1,6 +1,6 @@
 import logging
 import sys,os
-
+import traceback
 from agent.cdss.capbility import CDSSCapability
 from agent.cdss.models.schemas import CDSSInput, CDSSInt, CDSSText
 from model.response import StandardResponse
@@ -78,12 +78,12 @@ async def neighbor_search(
             data={"可能诊断":output.diagnosis.value,"症状":keywords}
         )
     except Exception as e:
-        print(e)
-        raise e
+        traceback.print_exc()
+        logger.error(f"get_disease_detail failed: {str(e)}")
         return StandardResponse(
             success=False,
-            error_code=500,
-            error_msg=str(e)
+            errorCode=500,
+            errorMsg=str(e)
         )
 
 @router.get("/{disease_name}/detail", response_model=StandardResponse)
@@ -95,12 +95,13 @@ async def get_disease_detail(
         result = service.get_disease_detail(disease_name,'疾病')
         return StandardResponse(success=True, data=result)
     except Exception as e:
+        traceback.print_exc()
         logger.error(f"get_disease_detail failed: {str(e)}")
         return StandardResponse(
             success=False,
-            error_code=500,
-            error_msg=str(e)
+            errorCode=500,
+            errorMsg=str(e)
         )
 
-#capability = CDSSCapability()
+capability = CDSSCapability()
 graph_router = router

service/cdss_service.py

@@ -29,33 +29,33 @@ class CdssService:
         edgeService = KGEdgeService(next(get_db()))
 
         #category='疾病相关实验室检查项目'
-        examinations = edgeService.get_edges_by_nodes(src_id=disease_id, category='疾病相关实验室检查项目')
+        laboratoryTests = edgeService.get_edges_by_nodes(src_id=disease_id, category='疾病相关实验室检查项目')
         #category='疾病相关辅助检查项目'
-        physical_examinations = edgeService.get_edges_by_nodes(src_id=disease_id, category='疾病相关辅助检查项目')
+        examinations = edgeService.get_edges_by_nodes(src_id=disease_id, category='疾病相关辅助检查项目')
         # category='疾病相关鉴别诊断'
         differential_diagnosis = edgeService.get_edges_by_nodes(src_id=disease_id, category='疾病相关鉴别诊断')
 
 
-        e_nodes = []
-        pe_nodes = []
+        test_nodes = []
+        exam_nodes = []
         ddx_nodes = []
 
-        if examinations:
-            e_nodes = [exam['dest_node'] for exam in examinations]
-            self.add_props_to_nodes(e_nodes)
+        if laboratoryTests:
+            test_nodes = [test['dest_node'] for test in laboratoryTests]
+            self.add_props_to_nodes(test_nodes)
         
-        if physical_examinations:
-            pe_nodes = [pe['dest_node'] for pe in physical_examinations]
-            self.add_props_to_nodes(pe_nodes)
+        if examinations:
+            exam_nodes = [exam['dest_node'] for exam in examinations]
+            self.add_props_to_nodes(exam_nodes)
 
         if differential_diagnosis:
-            ddx_nodes = [pe['dest_node'] for pe in differential_diagnosis]
+            ddx_nodes = [ddx['dest_node'] for ddx in differential_diagnosis]
             self.add_props_to_nodes(ddx_nodes,['Differential Diagnosis'])
         
         return {
             'disease':diseases[0],
-            'examinations': e_nodes,
-            'physical_examinations': pe_nodes,
+            'laboratory_tests': test_nodes,
+            'examinations': exam_nodes,
             'differential_diagnosis': ddx_nodes
         }