Update mcp/mygene.py

mcp/mygene.py

@@ -1,21 +1,14 @@
-import os, httpx
+# mcp/mygene.py
 from functools import lru_cache
+from mcp.clients import BaseClient
 
-_KEY  = os.getenv("MYGENE_KEY")
-_URL  = "https://mygene.info/v3/query"
+class MyGeneClient(BaseClient):
+    def __init__(self):
+        super().__init__("https://mygene.info/v3", api_key_env="MYGENE_KEY")
 
-@lru_cache(maxsize=512)
-async def fetch_gene_info(sym: str) -> dict:
-    if not sym:
-        return {}
-    params = {
-        "q"     : sym,
-        "fields": "symbol,name,summary,alias,entrezgene,location",
-        "size"  : 1,
-    }
-    if _KEY:
-        params["api_key"] = _KEY
-    async with httpx.AsyncClient(timeout=8) as c:
-        r = await c.get(_URL, params=params)
-        r.raise_for_status()
-        return r.json().get("hits", [{}])[0]
+    @lru_cache(512)
+    async def fetch(self, symbol: str) -> Dict:
+        q = {"q": symbol, "fields": "symbol,name,summary,alias,entrezgene,location", "size": 1}
+        return (await self.request("GET", "query", params=q))["hits"][0]
+
+mygene = MyGeneClient()