Merge pull request stanford-oval#236 from zhoucheng89/main

Fix SerperRM bug.
feldges · Oct 27, 2024 · a06e0d4 · a06e0d4
2 parents f1af6d8 + c946f3c
commit a06e0d4
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(
                         {