ouyang/AstrBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: websearch

Browse Source
This commit is contained in:
Soulter
2024-12-16 20:26:07 +08:00
parent acb3af8ab8
commit 9f3af95198
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
packages/web_searcher/main.py
View File

@@ -22,7 +22,7 @@ class Main(star.Star):
self.sogo_search = Sogo() self.sogo_search = Sogo()
self.google = Google() self.google = Google()
async def _tidy_text(text: str) -> str: async def _tidy_text(self, text: str) -> str:
'''清理文本，去除空格、换行符等''' '''清理文本，去除空格、换行符等'''
return text.strip().replace("\n", " ").replace("\r", " ").replace(" ", " ") return text.strip().replace("\n", " ").replace("\r", " ").replace(" ", " ")
@@ -36,7 +36,7 @@ class Main(star.Star):
doc = Document(html) doc = Document(html)
ret = doc.summary(html_partial=True) ret = doc.summary(html_partial=True)
soup = BeautifulSoup(ret, 'html.parser') soup = BeautifulSoup(ret, 'html.parser')
ret = self._tidy_text(soup.get_text()) ret = await self._tidy_text(soup.get_text())
return ret return ret
async def _request_from_llm(self, event: AstrMessageEvent, resources: str) -> str: async def _request_from_llm(self, event: AstrMessageEvent, resources: str) -> str:
@@ -126,4 +126,5 @@ class Main(star.Star):
url(string): The url of the website to fetch content from url(string): The url of the website to fetch content from
''' '''
resp = await self._get_from_url(url) resp = await self._get_from_url(url)
resp = await self._request_from_llm(event, resp)
event.set_result(MessageEventResult().message(resp)) event.set_result(MessageEventResult().message(resp))