瀏覽代碼

fix: summary no docs (#1073)

takatost 1 年之前
父節點
當前提交
78e26f8b75
共有 1 個文件被更改,包括 1 次插入3 次删除
  1. 1 3
      api/core/tool/web_reader_tool.py

+ 1 - 3
api/core/tool/web_reader_tool.py

@@ -88,11 +88,9 @@ class WebReaderTool(BaseTool):
             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]