fix(models): fix lint errors

Author: ChenZiHong-Gavin

graphgen/operators/search_wikipedia.py

@@ -38,7 +38,7 @@ async def _process_single_entity(entity_name: str,
     except ValueError:
         response = None
 
-    logger.info(f"Entity {entity_name} search result: {str(search_results)} response: {response}")
+    logger.info("Entity %s search result: %s response: %s", entity_name, str(search_results), response)
 
     return entity_name, response
 

models/search/wiki_search.py

@@ -1,10 +1,10 @@
-import wikipedia
-from wikipedia import set_lang
-
 from typing import List, Union
-from utils import detect_main_language
 from dataclasses import dataclass
 
+import wikipedia
+from wikipedia import set_lang
+from utils import detect_main_language, logger
+
 
 @dataclass
 class WikiSearch:
@@ -22,6 +22,7 @@ async def summary(self, query: str) -> Union[str, None]:
         try:
             result = wikipedia.summary(query, auto_suggest=False, redirect=False)
         except wikipedia.exceptions.DisambiguationError as e:
+            logger.error("DisambiguationError: %s", e)
             result = None
         return result
 
@@ -30,5 +31,6 @@ async def page(self, query: str) -> Union[str, None]:
         try:
             result = wikipedia.page(query, auto_suggest=False, redirect=False).content
         except wikipedia.exceptions.DisambiguationError as e:
+            logger.error("DisambiguationError: %s", e)
             result = None
         return result

models/strategy/travserse_strategy.py

@@ -12,9 +12,9 @@ class TraverseStrategy(BaseStrategy):
     # 每个方向拓展的最大边数
     max_extra_edges: int = 5
     # 最长token数
-    max_tokens: int = 512
+    max_tokens: int = 256
     # 每个方向拓展的最大深度
-    max_depth: int = 5
+    max_depth: int = 2
     # 同一层中选边的策略（如果是双向拓展，同一层指的是两边连接的边的集合）
     edge_sampling: str = "max_loss" # "max_loss" or "min_loss" or "random"
     # 孤立节点的处理策略