MarktAnalyst / tavily_agent.py
tet-ana's picture
Upload 2 files
e3f42ae verified
from langchain_tavily import TavilySearch
from datetime import datetime
from dotenv import load_dotenv
import os
load_dotenv() # Загружает переменные из .env
# Проверка (необязательно, но полезно):
# print("Tavily API Key:", os.getenv("TAVILY_API_KEY"))
web_search = TavilySearch(max_results=5)
def tavily_search_with_date(query):
try:
results = web_search.invoke(query)
except Exception as e:
return [f"❌ Fehler bei TavilySearch: {e}"]
output = []
for r in results.get("results", []):
title = r.get("title", "Keiner Titel" )
url = r.get("url", "")
content = r.get("content", "Keiner Inhalt")
published = r.get("published_date")
if published:
try:
# Konvertiere den Zeitstempel in ein lesbares Format
date = datetime.fromisoformat(published.replace("Z", "+00:00"))
date_str = date.strftime("%d.%m.%Y %H:%M")
except:
date_str = published
else:
date_str = "kein Datum angegeben"
output.append(f"[Tavily] {title}\n📆 {date_str}\n{url}\n{content}")
return output if output else ["❌ Keine Tavily-Ergebnisse."]