Update src/api/agent.py

src/api/agent.py

@@ -2823,11 +2823,30 @@ async def _run_search_mode(
 ) -> Tuple[str, List[dict], List[dict], str]:
     context = await _prepare_queries(question, context, model, is_follow_up)
 
+    # Run PRIMO (or PubMed) and OpenAlex in parallel.
+    # Skip OpenAlex for books-only searches — it only indexes journal articles.
+    use_openalex = context.resource_type != "books"
+
+    import asyncio as _aio
     if context.source == "pubmed":
-        result, source_url = await _search_pubmed_articles(context, limit=6)
+        if use_openalex:
+            (result, source_url), oa_papers = await _aio.gather(
+                _search_pubmed_articles(context, limit=6),
+                _openalex_search(context.topic, limit=5),
+            )
+        else:
+            result, source_url = await _search_pubmed_articles(context, limit=6)
+            oa_papers = []
         source_name = "pubmed"
     else:
-        result, source_url = await _search_primo_catalog(context, limit=6)
+        if use_openalex:
+            (result, source_url), oa_papers = await _aio.gather(
+                _search_primo_catalog(context, limit=6),
+                _openalex_search(context.topic, limit=5),
+            )
+        else:
+            result, source_url = await _search_primo_catalog(context, limit=6)
+            oa_papers = []
         source_name = "primo"
 
     search_results = result.get("results", []) or []
@@ -2853,6 +2872,25 @@ async def _run_search_mode(
 
     answer += _source_badge(source_name, source_url)
 
+    # ── OpenAlex open-access papers section ───────────────────────────────────
+    if oa_papers:
+        answer += (
+            "<br><br><strong>🌍 Open Access Papers (via OpenAlex)</strong><br>"
+            "<span style='font-size:.78rem;color:#6b7280'>"
+            "Freely available peer-reviewed articles on this topic:</span><br>"
+        )
+        for i, paper in enumerate(oa_papers[:5], 1):
+            answer += (
+                f'<br><strong>[{i}]</strong> '
+                f'<a href="{paper["link"]}" target="_blank" '
+                f'style="color:#003366;font-weight:600">{_escape(paper["title"])}</a><br>'
+                f'<span style="font-size:.78rem;color:#6b7280">'
+                f'{_escape(paper["authors"])} · {paper["year"]}'
+                f'{(" · " + _escape(paper["venue"])) if paper.get("venue") else ""}'
+                f'</span>'
+            )
+        answer += _source_badge("openalex_badge", "https://openalex.org")
+
     # ── Bottom: AI tools recommendation panel ─────────────────────────────────
     answer += _ai_tools_footer(context)
 
@@ -2861,6 +2899,10 @@ async def _run_search_mode(
         link = item.get("primo_url") or item.get("link")
         if link:
             sources.append({"title": item.get("title", "Result"), "url": link})
+    # Also include OA papers in sources
+    for paper in oa_papers[:3]:
+        if paper.get("link"):
+            sources.append({"title": paper["title"], "url": paper["link"]})
     return answer, search_results, sources, source_url