Merge pull request #236 from zhoucheng89/main

Fix SerperRM bug.
stanford-oval · Oct 27, 2024 · dcfa2bd · dcfa2bd
2 parents 18d0258 + d7d8f40
commit dcfa2bd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/knowledge_storm/rm.py b/knowledge_storm/rm.py
@@ -537,7 +537,9 @@ def forward(self, query_or_queries: Union[str, List[str]], exclude_urls: List[st
                     snippets = [organic.get("snippet")]
                     if self.ENABLE_EXTRA_SNIPPET_EXTRACTION:
                         snippets.extend(
-                            valid_url_to_snippets.get(url, {}).get("snippets", [])
+                            valid_url_to_snippets.get(url.strip("'"), {}).get(
+                                "snippets", []
+                            )
                         )
                     collected_results.append(
                         {