fix: summary no docs (#1073)

langgenius · Aug 31, 2023 · 78e26f8 · 78e26f8
1 parent 2191312
commit 78e26f8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/api/core/tool/web_reader_tool.py b/api/core/tool/web_reader_tool.py
@@ -88,11 +88,9 @@ def _run(self, url: str, summary: bool = False, cursor: int = 0) -> str:
             texts = character_splitter.split_text(page_contents)
             docs = [Document(page_content=t) for t in texts]
 
-            if len(docs) == 0:
+            if len(docs) == 0 or docs[0].page_content.endswith('TEXT:'):
                 return "No content found."
 
-            docs = docs[1:]
-
             # only use first 5 docs
             if len(docs) > 5:
                 docs = docs[:5]