add feed blacklist

Qetesh · Aug 13, 2024 · ed78d9a · ed78d9a
1 parent 9c5c31a
commit ed78d9a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -11,12 +11,13 @@
 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=[