fix: update asyncio event loop retrieval in evaluators and correct gene ID access in NCBISearch

Author: CHERRY-ui8

graphgen/models/evaluator/length_evaluator.py

@@ -13,7 +13,7 @@ def __init__(self, tokenizer_name: str = "cl100k_base", max_concurrent: int = 10
 
     async def evaluate_single(self, pair: QAPair) -> float:
         # In async context, we should use the running loop
-        loop = asyncio.get_event_loop()
+        loop = asyncio.get_running_loop()
         return await loop.run_in_executor(None, self._calculate_length, pair.answer)
 
     def _calculate_length(self, text: str) -> float:

graphgen/models/evaluator/mtld_evaluator.py

@@ -20,7 +20,7 @@ def __init__(self, max_concurrent: int = 100):
     async def evaluate_single(self, pair: QAPair) -> float:
         # In async context, we should use the running loop
         import asyncio
-        loop = asyncio.get_event_loop()
+        loop = asyncio.get_running_loop()
         return await loop.run_in_executor(None, self._calculate_mtld_score, pair.answer)
 
     def _calculate_mtld_score(self, text: str, threshold=0.72) -> float:

graphgen/models/searcher/db/ncbi_searcher.py

@@ -380,7 +380,7 @@ def get_best_hit(self, keyword: str) -> Optional[dict]:
                     search_results = Entrez.read(search_handle)
 
                 if len(gene_id := search_results.get("IdList", [])) > 0:
-                    result = self.get_by_gene_id(gene_id)
+                    result = self.get_by_gene_id(gene_id[0])
                     return result
         except (RequestException, IncompleteRead):
             raise