fix: honor explicit threads source requests
This commit is contained in:
committed by
Trevin Chow
parent
7bda02169d
commit
6a5a122195
@@ -124,7 +124,9 @@ def available_sources(config: dict[str, Any], requested_sources: list[str] | Non
|
||||
available.append("perplexity")
|
||||
if requested_sources and "xiaohongshu" in requested_sources and env.is_xiaohongshu_available(config):
|
||||
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")
|
||||
if requested_sources and "pinterest" in requested_sources and env.is_pinterest_available(config):
|
||||
available.append("pinterest")
|
||||
|
||||
@@ -77,6 +77,13 @@ class CliV3Tests(unittest.TestCase):
|
||||
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):
|
||||
with self.assertRaises(SystemExit):
|
||||
cli.parse_search_flag("unknown")
|
||||
|
||||
Reference in New Issue
Block a user