代码提交

agent/cdss/libs/cdss_helper2.py

@@ -324,7 +324,7 @@ class CDSSHelper(GraphHelper):
 
         results = self.validDisease(results, start_nodes)
 
-        sorted_count_diags = sorted(results.items(), key=lambda x: x[1]["count"], reverse=True)[:20]
+        sorted_count_diags = sorted(results.items(), key=lambda x: x[1]["count"], reverse=True)[:10]
         diags = {}
         target_symptom_names = []
         for symptom_id in node_ids:

router/text_search.py

@@ -541,7 +541,7 @@ async def node_props_search(request: NodePropsSearchRequest, db: Session = Depen
 
         # 遍历props_ids查询属性信息
         for prop_id in request.props_ids:
-            prop = prop_service.get_props_by_id(prop_id)
+            prop = prop_service.get_prop_by_id(prop_id)
             if not prop:
                 logger.warning(f"属性不存在: {prop_id}")
                 continue

service/kg_prop_service.py

@@ -48,6 +48,27 @@ class KGPropService:
                 'prop_title': props.prop_title,
                 'type': props.type
             }
+        except Exception as e:
+            traceback.print_exc()
+            logger.error(f"根据ref_id查询属性失败: {str(e)}")
+            raise ValueError("查询失败")
+
+
+    def get_prop_by_id(self, id: int)-> dict:
+        try:
+            query = self.db.query(KGProp).filter(KGProp.id == id)
+
+            props = query.first()
+            if not props:
+                return None
+            return {
+                'id': props.id,
+                'category': props.category,
+                'prop_name': props.prop_name,
+                'prop_value': props.prop_value,
+                'prop_title': props.prop_title,
+                'type': props.type
+            }
         except Exception as e:
             traceback.print_exc()
             logger.error(f"根据id查询属性失败: {str(e)}")