nikeshn commited on
Commit
e8838b3
·
verified ·
1 Parent(s): b3c541d

Update src/api/agent.py

Browse files
Files changed (1) hide show
  1. 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=boolean,
 
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)