fix #153

Author: versun

config/settings.py

@@ -200,9 +200,13 @@
 
 STATIC_URL = "static/"
 STATIC_ROOT = BASE_DIR / "static"
+STATICFILES_BACKEND = "whitenoise.storage.CompressedManifestStaticFilesStorage"
+if "test" in sys.argv:
+    STATICFILES_BACKEND = "django.contrib.staticfiles.storage.StaticFilesStorage"
+
 STORAGES = {
     "staticfiles": {
-        "BACKEND": "whitenoise.storage.CompressedManifestStaticFilesStorage",
+        "BACKEND": STATICFILES_BACKEND,
     },
 }
 

core/models/agent.py

@@ -702,7 +702,7 @@ def translate(self, text: str, target_language: str, **kwargs) -> dict:
         time.sleep(self.interval)
         return {"text": self.translated_text, "tokens": 10, "characters": len(text)}
 
-    def summarize(self, text: str, target_language: str) -> dict:
+    def summarize(self, text: str, target_language: str, **kwargs) -> dict:
         logger.info(">>> Test Summarize [%s]: %s", target_language, text)
         time.sleep(self.interval)
         return {"text": self.translated_text, "tokens": 10, "characters": len(text)}

core/tasks/summarize_feeds.py

@@ -109,6 +109,7 @@ def summarize_feed(
                     chunk_delimiter=chunk_delimiter,
                     max_chunks_per_entry=max_chunks_per_entry,
                     summary_detail=feed.summary_detail,
+                    user_prompt=feed.additional_prompt,
                 )
 
                 entry.ai_summary = summary
@@ -163,6 +164,7 @@ def _summarize_entry(
     chunk_delimiter: str,
     max_chunks_per_entry: int,
     summary_detail: float,
+    user_prompt: str = None,
 ) -> tuple[str, int]:
     """
     Summarize a single entry with memory optimizations.
@@ -210,6 +212,7 @@ def _summarize_entry(
             max_retries=3,
             text=text_chunks[0],
             target_language=target_language,
+            user_prompt=user_prompt,
         )
         summary = response.get("text", "")
         tokens = response.get("tokens", 0)
@@ -249,6 +252,7 @@ def _summarize_entry(
             text=prompt,
             target_language=target_language,
             max_tokens=max_context_tokens,
+            user_prompt=user_prompt,
         )
 
         chunk_summary = response.get("text", "")

core/tasks/translate_feeds.py

@@ -80,6 +80,7 @@ def translate_feed(feed: Feed, target_field: str = "title"):
                     entry=entry,
                     target_language=feed.target_language,
                     engine=feed.translator,
+                    user_prompt=feed.additional_prompt,
                 )
                 total_tokens += metrics["tokens"]
                 total_characters += metrics["characters"]
@@ -103,6 +104,7 @@ def translate_feed(feed: Feed, target_field: str = "title"):
                     entry=entry,
                     target_language=feed.target_language,
                     engine=feed.translator,
+                    user_prompt=feed.additional_prompt,
                 )
                 total_tokens += metrics["tokens"]
                 total_characters += metrics["characters"]
@@ -162,6 +164,7 @@ def _translate_entry_title(
     entry: Entry,
     target_language: str,
     engine: Agent,
+    user_prompt: str = None,
 ) -> dict:
     """Translate entry title with memory optimization."""
     total_tokens = 0
@@ -178,6 +181,7 @@ def _translate_entry_title(
         text=entry.original_title,
         target_language=target_language,
         text_type="title",
+        user_prompt=user_prompt,
     )
 
     if result:
@@ -193,6 +197,7 @@ def _translate_entry_content(
     entry: Entry,
     target_language: str,
     engine: Agent,
+    user_prompt: str = None,
 ) -> dict:
     """Translate entry content with memory optimization."""
     total_tokens = 0
@@ -230,6 +235,7 @@ def _translate_entry_content(
         text=processed_html,
         target_language=target_language,
         text_type="content",
+        user_prompt=user_prompt,
     )
 
     translated_content = result.get("text")

core/tests/test_tasks_consolidated.py

@@ -689,6 +689,52 @@ def test_handle_feeds_translation_integration(self, mockauto_retry):
         self.feed.refresh_from_db()
         self.assertIsNotNone(self.feed.last_translate)
 
+    @patch("core.tasks.translate_feeds.auto_retry")
+    def test_translate_feed_passes_additional_prompt_for_title(
+        self, mockauto_retry
+    ):
+        """测试标题翻译会透传feed.additional_prompt"""
+        self.feed.translator = self.agent
+        self.feed.translate_title = True
+        self.feed.additional_prompt = "Use glossary terms"
+        self.feed.save()
+        self._create_test_entry()
+
+        mockauto_retry.return_value = {
+            "text": "Translated Title",
+            "tokens": 10,
+            "characters": 15,
+        }
+
+        translate_feed(self.feed, target_field="title")
+
+        self.assertEqual(
+            mockauto_retry.call_args.kwargs["user_prompt"], "Use glossary terms"
+        )
+
+    @patch("core.tasks.translate_feeds.auto_retry")
+    def test_translate_feed_passes_additional_prompt_for_content(
+        self, mockauto_retry
+    ):
+        """测试正文翻译会透传feed.additional_prompt"""
+        self.feed.translator = self.agent
+        self.feed.translate_content = True
+        self.feed.additional_prompt = "Preserve HTML structure"
+        self.feed.save()
+        self._create_test_entry(content="<p>Test content</p>")
+
+        mockauto_retry.return_value = {
+            "text": "<p>Translated content</p>",
+            "tokens": 20,
+            "characters": 30,
+        }
+
+        translate_feed(self.feed, target_field="content")
+
+        self.assertEqual(
+            mockauto_retry.call_args.kwargs["user_prompt"], "Preserve HTML structure"
+        )
+
     @patch("core.tasks.fetch_feeds.handle_single_feed_fetch")
     def test_handle_feeds_summary(self, mock_handle_single):
         """测试批量摘要处理 - 批量操作验证"""
@@ -719,6 +765,35 @@ def test_handle_feeds_summary(self, mock_handle_single):
             # 如果出现异常，这是预期的，因为测试环境可能缺少某些依赖
             self.fail(f"Unexpected exception: {e}")
 
+    @patch("core.tasks.summarize_feeds.auto_retry")
+    @patch("core.tasks.summarize_feeds.text_handler.adaptive_chunking")
+    @patch("core.tasks.summarize_feeds.text_handler.get_token_count")
+    @patch("core.tasks.summarize_feeds.text_handler.clean_content")
+    def test_summarize_feed_passes_additional_prompt(
+        self,
+        mock_clean_content,
+        mock_get_token_count,
+        mock_adaptive_chunking,
+        mockauto_retry,
+    ):
+        """测试摘要会透传feed.additional_prompt"""
+        self.feed.summarizer = self.agent
+        self.feed.additional_prompt = "Keep blockchain terms in English"
+        self.feed.save()
+        self._create_test_entry(content="<p>Long content</p>")
+
+        mock_clean_content.return_value = "Long content"
+        mock_get_token_count.return_value = 100
+        mock_adaptive_chunking.return_value = ["Long content"]
+        mockauto_retry.return_value = {"text": "Summary", "tokens": 12}
+
+        summarize_feed(self.feed)
+
+        self.assertEqual(
+            mockauto_retry.call_args.kwargs["user_prompt"],
+            "Keep blockchain terms in English",
+        )
+
     # ==================== Edge Cases and Error Handling ====================
 
     def test_translate_title_with_empty_result(self):