Update mcp/mygene.py

mcp/mygene.py

@@ -1,11 +1,37 @@
-import os, httpx, asyncio
+"""
+MyGene.info async helper
+Docs: https://docs.mygene.info/en/v2/doc/query_service.html  (public, key-free)  📄
+"""
+
+from __future__ import annotations
+import os, asyncio, httpx
 from functools import lru_cache
-_KEY = os.getenv("MYGENE_KEY"); _URL="https://mygene.info/v3/query"
+from typing import Dict
+
+_URL   = "https://mygene.info/v3/query"          # official endpoint :contentReference[oaicite:1]{index=1}
+_KEY   = os.getenv("MYGENE_KEY")                 # optional; see FAQ :contentReference[oaicite:2]{index=2}
+_FIELDS = "symbol,name,summary,alias,entrezgene,location"
+_HDR   = {"Accept": "application/json"}
+_TIMEOUT   = 8
+_MAX_RETRY = 2
+
 @lru_cache(maxsize=512)
-async def fetch_gene_info(sym: str):
-    p={"q":sym,"fields":"symbol,name,summary,alias,entrezgene,location","size":1}
-    if _KEY: p["api_key"]=_KEY
-    async with httpx.AsyncClient(timeout=8) as c:
-        r=await c.get(_URL,params=p); r.raise_for_status()
-        return r.json().get("hits",[{}])[0]
-``` :contentReference[oaicite:7]{index=7}  
+async def fetch_gene_info(symbol: str) -> Dict:
+    """
+    Return first MyGene.info hit or {} (never raises).
+    """
+    params = {"q": symbol, "fields": _FIELDS, "size": 1}
+    if _KEY: params["api_key"] = _KEY
+
+    delay = 0.0
+    for _ in range(_MAX_RETRY):
+        try:
+            async with httpx.AsyncClient(timeout=_TIMEOUT, headers=_HDR) as cli:
+                r = await cli.get(_URL, params=params)
+                r.raise_for_status()
+                return r.json().get("hits", [{}])[0]
+        except (httpx.HTTPStatusError, httpx.ReadTimeout):
+            await asyncio.sleep(delay)
+            delay = 0.8 if delay == 0 else 0      # retry once
+
+    return {}           # graceful fallback