Browse Source

临床表现症状标记2

yuchengwei 3 weeks ago
parent
commit
ee1e1fefe6
1 changed files with 16 additions and 14 deletions
  1. 16 14
      router/text_search.py

+ 16 - 14
router/text_search.py

@@ -558,13 +558,14 @@ async def node_props_search(request: NodePropsSearchRequest, db: Session = Depen
                         try:
                             # 获取症状节点
                             symptom_node = node_service.get_node_by_name_category(symptom, '症状')
-                            # 获取症状相关同义词
-                            edges = edge_service.get_edges_by_nodes(src_id=symptom_node['id'], category='症状同义词')
-                            if edges:
-                                # 添加同义词
-                                for edge in edges:
-                                    if edge['dest_node'] and edge['dest_node'].get('name'):
-                                        symptom_list.append(edge['dest_node']['name'])
+                            # 获取症状相关同义词(包括1.0和2.0版本)
+                            for category in ['症状同义词', '症状同义词2.0']:
+                                edges = edge_service.get_edges_by_nodes(src_id=symptom_node['id'], category=category)
+                                if edges:
+                                    # 添加同义词
+                                    for edge in edges:
+                                        if edge['dest_node'] and edge['dest_node'].get('name'):
+                                            symptom_list.append(edge['dest_node']['name'])
                         except ValueError:
                             # 如果找不到节点,只添加原始症状
                             continue
@@ -601,13 +602,14 @@ async def node_props_search(request: NodePropsSearchRequest, db: Session = Depen
                     symptom_list.append(symptom)
                     # 获取症状节点
                     symptom_node = node_service.get_node_by_name_category(symptom, '症状')
-                    # 获取症状相关同义词
-                    edges = edge_service.get_edges_by_nodes(src_id=symptom_node['id'], category='症状同义词')
-                    if edges:
-                        # 添加同义词
-                        for edge in edges:
-                            if edge['dest_node'] and edge['dest_node'].get('name'):
-                                symptom_list.append(edge['dest_node']['name'])
+                    # 获取症状相关同义词(包括1.0和2.0版本)
+                    for category in ['症状同义词', '症状同义词2.0']:
+                        edges = edge_service.get_edges_by_nodes(src_id=symptom_node['id'], category=category)
+                        if edges:
+                            # 添加同义词
+                            for edge in edges:
+                                if edge['dest_node'] and edge['dest_node'].get('name'):
+                                    symptom_list.append(edge['dest_node']['name'])
                 except ValueError:
                     # 如果找不到节点,只添加原始症状
                     continue