Update mcp/mygene.py

mcp/mygene.py

@@ -1,21 +1,22 @@
- """Async wrapper for MyGene.info API (https://mygene.info)
-     Exposes `fetch_gene_info` which returns the top hit for a free‑text query.
-     No API‑key required; 500 QPS community quota.
- """
+#!/usr/bin/env python3
+"""Async wrapper for MyGene.info API (https://mygene.info)
 
-import httpx, asyncio
+Provides a single helper:
+    • `fetch_gene_info(query)` → returns the first matching hit dict or `{}`.
+
+No API‑key needed; community tier allows 500 QPS.
+"""
+from __future__ import annotations
+
+import httpx
 from functools import lru_cache
 
 _BASE = "https://mygene.info/v3"
 
-@lru_cache(maxsize=128)
-async def fetch_gene_info(query: str) -> dict:  # noqa: C901 – simple IO fn
-    """Return first hit dict or {} if nothing found.
 
-    Parameters
-    ----------
-    query : str – gene symbol / keyword / Entrez id / HGNC etc.
-    """
+@lru_cache(maxsize=128)
+async def fetch_gene_info(query: str) -> dict:  # noqa: D401
+    """Return first hit for *query* or an empty dict."""
     params = {
         "q": query,
         "fields": "symbol,name,summary,alias,entrezgene,clinvar,location,go",
@@ -25,4 +26,4 @@ async def fetch_gene_info(query: str) -> dict:  # noqa: C901 – simple IO fn
         resp = await client.get(f"{_BASE}/query", params=params)
         resp.raise_for_status()
         hits = resp.json().get("hits", [])
-        return hits[0] if hits else {}
+        return hits[0] if hits else {}