takatost
2 years ago
1 changed files with 2 additions and 0 deletions
+ 2
- 0
api/core/tool/web_reader_tool.py
View File
| @@ -88,6 +88,8 @@ class WebReaderTool(BaseTool): | |||
| texts = character_splitter.split_text(page_contents) | |||
| docs = [Document(page_content=t) for t in texts] | |||
| docs = docs[1:] | |||
| # only use first 5 docs | |||
| if len(docs) > 5: | |||
| docs = docs[:5] | |||
Loading…