Update mcp/opentargets.py

mcp/opentargets.py

@@ -1,14 +1,22 @@
-### `opentargets.py` – tractability & constraint  
-import os, httpx, textwrap, asyncio
+import os, httpx, textwrap
 from functools import lru_cache
-_URL="https://api.platform.opentargets.org/api/v4/graphql"
-_HDR={"Content-Type":"application/json","Accept":"application/json"}
-if os.getenv("OT_KEY"): HDR["Authorization"]="Bearer "+os.getenv("OT_KEY")
-_Q=textwrap.dedent("""query Q($g:String!,$n:Int){associations(geneSymbol:$g,size:$n){rows{score datatypeId disease{id name} target{id symbol}}}}""")
+
+_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')}"
+
+_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(maxsize=512)
-async def fetch_ot(sym:str, n:int=30):
-    p={"query":_Q,"variables":{"g":sym,"n":n}}
-    async with httpx.AsyncClient(timeout=10,headers=_HDR) as c:
-        r=await c.post(_URL,json=p); 
+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"]
-``` :contentReference[oaicite:9]{index=9