Spaces:
Sleeping
Sleeping
File size: 2,063 Bytes
20366da 15ef6a6 20366da 19f3ac6 20366da 49650ff 20366da 53d4ce7 20366da 50cdfc4 20366da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import re
from typing import Any, Optional
from smolagents.tools import Tool
import requests
import markdownify
import smolagents
from tools.feed_processor import DailyHFRSSParser, RSSFeedProcessor
class HFDaylyPapperTool(Tool):
name = "get_actual_ai_news"
description = "Return actual news about AI today."
output_type = "string"
inputs = {'number': {'type': 'integer', 'description': 'How much paper you want research.'}}
def __init__(self, max_results=10, **kwargs):
super().__init__()
self.rss_processor = RSSFeedProcessor()
self.rss_processor.register_feed(
"HuggingFace Daily pappers",
"https://jamesg.blog/hf-papers.xml",
DailyHFRSSParser(),
)
def forward(self, number: int) -> str:
try:
from markdownify import markdownify
from smolagents.utils import truncate_content
except ImportError as e:
raise ImportError(
"You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
) from e
try:
response = self.rss_processor.get_latest_articles(["HuggingFace Daily pappers"], number)[0]
result = ""
for article in response:
result+=self.pretty_str_paper(article)
markdown_content = markdownify(result).strip()
markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
return truncate_content(markdown_content, 10000)
except Exception as e:
return f"An unexpected error occurred: {str(e)}"
def pretty_str_paper(self, article):
return f"*{article.title}*\n" \
f"_Источник_: {article.source}\n" \
f"_Авторы_: {article.authors}\n" \
f"_Опубликовано_: {article.published}\n" \
f"_Описание_: {article.summary}\n\n" |