From 9f3af951987b87ef7a2c5cd16f6bc9a30ea99498 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Mon, 16 Dec 2024 20:26:07 +0800 Subject: [PATCH] fix: websearch --- packages/web_searcher/main.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/packages/web_searcher/main.py b/packages/web_searcher/main.py index 4d4a192b5..1ffffb678 100644 --- a/packages/web_searcher/main.py +++ b/packages/web_searcher/main.py @@ -22,7 +22,7 @@ class Main(star.Star): self.sogo_search = Sogo() 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(" ", " ") @@ -36,7 +36,7 @@ class Main(star.Star): doc = Document(html) ret = doc.summary(html_partial=True) soup = BeautifulSoup(ret, 'html.parser') - ret = self._tidy_text(soup.get_text()) + ret = await self._tidy_text(soup.get_text()) return ret 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 ''' resp = await self._get_from_url(url) + resp = await self._request_from_llm(event, resp) event.set_result(MessageEventResult().message(resp)) \ No newline at end of file