From ed78d9a9006b367452dd8711176dc23330edc320 Mon Sep 17 00:00:00 2001
From: Qetesh <4559341+Qetesh@users.noreply.github.com>
Date: Tue, 13 Aug 2024 17:53:20 +0800
Subject: add feed blacklist

---
 main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 962c6cc..17b24ff 100644
--- a/main.py
+++ b/main.py
@@ -11,12 +11,13 @@ miniflux_api_key = os.getenv('miniflux_api_key')
 llm_base_url = os.getenv('llm_base_url')
 llm_api_key = os.getenv('llm_api_key')
 llm_model = os.getenv('llm_model')
+feed_blacklist = os.getenv('feed_blacklist').split(',')
 
 miniflux_client = miniflux.Client(miniflux_base_url, api_key=miniflux_api_key)
 llm_client = OpenAI(base_url=llm_base_url, api_key=llm_api_key)
 
 def process_entry(entry):
-    if not entry['content'].startswith('摘要'):
+    if (not entry['content'].startswith('摘要')) & (entry['feed']['site_url'] not in feed_blacklist):
         completion = llm_client.chat.completions.create(
             model=llm_model,
             messages=[
-- 
cgit v1.2.3