Merge pull request stanford-oval#132 from zenith110/users/zenith110/131

[RM Enhancement] Fixed SerperRM when knowledge graph is none to properly return data
hw92 · Aug 5, 2024 · 17ff5d5 · 17ff5d5
2 parents cbb38da + 51d0d2f
commit 17ff5d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/knowledge_storm/rm.py b/knowledge_storm/rm.py
@@ -432,8 +432,8 @@ def forward(self, query_or_queries: Union[str, List[str]], exclude_urls: List[st
  collected_results.append(
  {
  'snippets': snippets,
- 'title': result.get('title'),
- 'url': result.get('link'),
+ 'title': organic.get('title'),
+ 'url': organic.get('link'),
  'description': '',
  }
  )