fix: honor explicit threads source requests

This commit is contained in:
Dinakar Sarbada
2026-05-06 12:39:08 -07:00
committed by Trevin Chow
parent 7bda02169d
commit 6a5a122195
2 changed files with 10 additions and 1 deletions
+3 -1
View File
@@ -124,7 +124,9 @@ def available_sources(config: dict[str, Any], requested_sources: list[str] | Non
available.append("perplexity") available.append("perplexity")
if requested_sources and "xiaohongshu" in requested_sources and env.is_xiaohongshu_available(config): if requested_sources and "xiaohongshu" in requested_sources and env.is_xiaohongshu_available(config):
available.append("xiaohongshu") available.append("xiaohongshu")
if env.is_threads_available(config): if env.is_threads_available(config) or (
requested_sources and "threads" in requested_sources and config.get("SCRAPECREATORS_API_KEY")
):
available.append("threads") available.append("threads")
if requested_sources and "pinterest" in requested_sources and env.is_pinterest_available(config): if requested_sources and "pinterest" in requested_sources and env.is_pinterest_available(config):
available.append("pinterest") available.append("pinterest")
+7
View File
@@ -77,6 +77,13 @@ class CliV3Tests(unittest.TestCase):
cli.parse_search_flag("threads, pinterest"), cli.parse_search_flag("threads, pinterest"),
) )
def test_explicit_threads_search_uses_scrapecreators_key_without_include_sources(self):
available = cli.pipeline.available_sources(
{"SCRAPECREATORS_API_KEY": "test-key", "INCLUDE_SOURCES": ""},
requested_sources=["threads"],
)
self.assertIn("threads", available)
def test_parse_search_flag_rejects_invalid_or_empty_inputs(self): def test_parse_search_flag_rejects_invalid_or_empty_inputs(self):
with self.assertRaises(SystemExit): with self.assertRaises(SystemExit):
cli.parse_search_flag("unknown") cli.parse_search_flag("unknown")