Update mcp/opentargets.py

mcp/opentargets.py

@@ -1,22 +1,22 @@
-import os, httpx, textwrap
+# mcp/opentargets.py
+import textwrap
 from functools import lru_cache
+from mcp.clients import BaseClient
 
-_URL  = "https://api.platform.opentargets.org/api/v4/graphql"
-_HDR  = {"Content-Type": "application/json", "Accept": "application/json"}
-if os.getenv("OT_KEY"):                       # optional higher quota
-    _HDR["Authorization"] = f"Bearer {os.getenv('OT_KEY')}"
+class OTClient(BaseClient):
+    QUERY = textwrap.dedent("""
+      query ($g: String!, $n: Int!){
+        associations(geneSymbol: $g, size: $n) {
+          rows { score datatypeId datasourceId disease { id name } target { id symbol } }
+        }
+      }
+    """)
+    def __init__(self):
+        super().__init__("https://api.platform.opentargets.org/api/v4/graphql", api_key_env="OT_KEY")
 
-_QUERY = textwrap.dedent("""
-query Assoc($g:String!, $n:Int!) {
-  associations(geneSymbol:$g, size:$n) {
-    rows { score datasourceId disease { id name } target { id symbol } }
-  }
-}""")
+    @lru_cache(512)
+    async def fetch(self, symbol: str, n: int = 30):
+        payload = {"query": self.QUERY, "variables": {"g": symbol, "n": n}}
+        return (await self.request("POST", "", json=payload))["data"]["associations"]["rows"]
 
-@lru_cache(maxsize=512)
-async def fetch_ot(sym: str, n: int = 30):
-    payload = {"query": _QUERY, "variables": {"g": sym, "n": n}}
-    async with httpx.AsyncClient(timeout=10, headers=_HDR) as c:
-        r = await c.post(_URL, json=payload)
-        r.raise_for_status()
-        return r.json()["data"]["associations"]["rows"]
+ot = OTClient()