anything-llm/collector/scripts/link_utils.py
2023-11-14 16:41:39 -08:00

23 lines
1.1 KiB
Python

import json
from datetime import datetime
from dotenv import load_dotenv
from .watch.utils import guid
from .utils import tokenize
load_dotenv()
def append_meta(request, text, metadata_only = False):
meta = {
'id': guid(),
'url': request.url,
'title': request.html.find('title', first=True).text if len(request.html.find('title')) != 0 else '',
'docAuthor': 'N/A',
'docSource': 'webpage',
'chunkSource': request.url,
'description': request.html.find('meta[name="description"]', first=True).attrs.get('content') if request.html.find('meta[name="description"]', first=True) != None else '',
'published':request.html.find('meta[property="article:published_time"]', first=True).attrs.get('content') if request.html.find('meta[property="article:published_time"]', first=True) != None else datetime.today().strftime('%Y-%m-%d %H:%M:%S'),
'wordCount': len(text.split(' ')),
'pageContent': text,
'token_count_estimate':len(tokenize(text)),
}
return "Article JSON Metadata:\n"+json.dumps(meta)+"\n\n\nText Content:\n" + text if metadata_only == False else meta