Update src/api/agent.py
Browse files- src/api/agent.py +5 -4
src/api/agent.py
CHANGED
|
@@ -1600,7 +1600,8 @@ async def _search_primo_catalog(context: SearchContextPayload, limit: int = 5) -
|
|
| 1600 |
boolean = clean_topic or context.topic
|
| 1601 |
result = await search_primo(
|
| 1602 |
SearchRequest(
|
| 1603 |
-
query=
|
|
|
|
| 1604 |
source="primo",
|
| 1605 |
limit=limit,
|
| 1606 |
peer_reviewed=context.peer_reviewed,
|
|
@@ -2400,7 +2401,7 @@ async def _research_snapshot(context: SearchContextPayload, model: str) -> Tuple
|
|
| 2400 |
|
| 2401 |
# Parallel fetch: OpenAlex (fast, free) + PRIMO
|
| 2402 |
papers, (primo_result, primo_url) = await _asyncio.gather(
|
| 2403 |
-
_openalex_search(topic, limit=8),
|
| 2404 |
_search_primo_catalog(context, limit=6),
|
| 2405 |
)
|
| 2406 |
|
|
@@ -2832,7 +2833,7 @@ async def _run_search_mode(
|
|
| 2832 |
if use_openalex:
|
| 2833 |
(result, source_url), oa_papers = await _aio.gather(
|
| 2834 |
_search_pubmed_articles(context, limit=6),
|
| 2835 |
-
_openalex_search(context.topic, limit=5),
|
| 2836 |
)
|
| 2837 |
else:
|
| 2838 |
result, source_url = await _search_pubmed_articles(context, limit=6)
|
|
@@ -2842,7 +2843,7 @@ async def _run_search_mode(
|
|
| 2842 |
if use_openalex:
|
| 2843 |
(result, source_url), oa_papers = await _aio.gather(
|
| 2844 |
_search_primo_catalog(context, limit=6),
|
| 2845 |
-
_openalex_search(context.topic, limit=5),
|
| 2846 |
)
|
| 2847 |
else:
|
| 2848 |
result, source_url = await _search_primo_catalog(context, limit=6)
|
|
|
|
| 1600 |
boolean = clean_topic or context.topic
|
| 1601 |
result = await search_primo(
|
| 1602 |
SearchRequest(
|
| 1603 |
+
query=context.topic, # plain topic as q= fallback
|
| 1604 |
+
boolean_query=boolean, # FIX: boolean goes here so _boolean_to_primo_params() fires
|
| 1605 |
source="primo",
|
| 1606 |
limit=limit,
|
| 1607 |
peer_reviewed=context.peer_reviewed,
|
|
|
|
| 2401 |
|
| 2402 |
# Parallel fetch: OpenAlex (fast, free) + PRIMO
|
| 2403 |
papers, (primo_result, primo_url) = await _asyncio.gather(
|
| 2404 |
+
_openalex_search(context.ai_tool_query or context.topic, limit=8), # FIX: use richer natural query
|
| 2405 |
_search_primo_catalog(context, limit=6),
|
| 2406 |
)
|
| 2407 |
|
|
|
|
| 2833 |
if use_openalex:
|
| 2834 |
(result, source_url), oa_papers = await _aio.gather(
|
| 2835 |
_search_pubmed_articles(context, limit=6),
|
| 2836 |
+
_openalex_search(context.ai_tool_query or context.topic, limit=5), # FIX: use richer natural query
|
| 2837 |
)
|
| 2838 |
else:
|
| 2839 |
result, source_url = await _search_pubmed_articles(context, limit=6)
|
|
|
|
| 2843 |
if use_openalex:
|
| 2844 |
(result, source_url), oa_papers = await _aio.gather(
|
| 2845 |
_search_primo_catalog(context, limit=6),
|
| 2846 |
+
_openalex_search(context.ai_tool_query or context.topic, limit=5), # FIX: use richer natural query
|
| 2847 |
)
|
| 2848 |
else:
|
| 2849 |
result, source_url = await _search_primo_catalog(context, limit=6)
|