Upload 2 files

core/sources/ctx.py

@@ -239,6 +239,32 @@ async def fetch_ctx_genetox(cas_or_query: str, http: httpx.AsyncClient) -> Dict[
     if not q:
         return {"ok": False, "error": "Empty query"}
 
+    # Prefer worker proxy if configured (matches production behavior)
+    if settings.worker_base_url:
+        try:
+            worker_url = settings.worker_base_url.rstrip("/") + "/ctx-genetox"
+            payload = {"dtxsid": q} if q.upper().startswith("DTXSID") else {"query": q}
+            r = await http.post(worker_url, json=payload, timeout=25.0)
+            if r.status_code < 400:
+                data = r.json()
+                if data.get("summary"):
+                    dtxsid = data.get("dtxsid")
+                    return {
+                        "ok": True,
+                        "dtxsid": dtxsid,
+                        "summary": data.get("summary"),
+                        "dashboard_url": dashboard_details_url(dtxsid or q),
+                    }
+                if data.get("resolveUrl"):
+                    return {
+                        "ok": False,
+                        "error": data.get("message") or "No DTXSID found for this query.",
+                        "dashboard_search": data.get("resolveUrl"),
+                    }
+            # If worker errors, fall through to direct CTX
+        except Exception:
+            pass
+
     dtxsid = await resolve_dtxsid(q, http)
     if not dtxsid:
         return {

core/sources/ntp.py

@@ -5,6 +5,8 @@ from urllib.parse import urljoin
 
 import httpx
 
+from core.config import settings
+
 REPORTS_URL = "https://ntp.niehs.nih.gov/publications/reports"
 BASE = "https://ntp.niehs.nih.gov"
 INDEX_URL = "https://ntp.niehs.nih.gov/data/tr"
@@ -92,6 +94,32 @@ async def search_technical_reports(query: str, http: httpx.AsyncClient, limit: i
     if not q:
         return {"ok": False, "error": "Empty query", "items": []}
 
+    # Prefer worker proxy if configured (matches production behavior)
+    if settings.worker_base_url:
+        try:
+            worker_url = settings.worker_base_url.rstrip("/") + "/ntp-tr"
+            r = await http.post(worker_url, json={"query": q, "limit": limit}, timeout=25.0)
+            if r.status_code < 400:
+                data = r.json()
+                rows = data.get("results") or []
+                items: List[Dict[str, Any]] = []
+                for row in rows:
+                    if not isinstance(row, dict):
+                        continue
+                    items.append(
+                        {
+                            "num": (row.get("tr") or "").replace("TR-", ""),
+                            "tr": row.get("tr"),
+                            "report_page": row.get("page") or row.get("url"),
+                            "title": row.get("title") or "NTP Technical Report",
+                            "year": row.get("year"),
+                            "pdf": row.get("pdf"),
+                        }
+                    )
+                return {"ok": True, "query": q, "items": items}
+        except Exception:
+            pass
+
     try:
         r = await http.get(REPORTS_URL, timeout=25, follow_redirects=True)
         r.raise_for_status()