fix: route parallel web backend through grounding
This commit is contained in:
committed by
Trevin Chow
parent
7214dd6051
commit
5c802b0daa
@@ -203,7 +203,7 @@ def run(
|
|||||||
available = [source for source in available if source in requested_sources]
|
available = [source for source in available if source in requested_sources]
|
||||||
if web_backend == "none":
|
if web_backend == "none":
|
||||||
available = [s for s in available if s != "grounding"]
|
available = [s for s in available if s != "grounding"]
|
||||||
elif web_backend in ("brave", "exa", "serper") and "grounding" not in available:
|
elif web_backend in ("brave", "exa", "serper", "parallel") and "grounding" not in available:
|
||||||
available.append("grounding")
|
available.append("grounding")
|
||||||
if not available:
|
if not available:
|
||||||
raise RuntimeError("No sources are available for this run.")
|
raise RuntimeError("No sources are available for this run.")
|
||||||
|
|||||||
@@ -52,6 +52,32 @@ class PipelineV3Tests(unittest.TestCase):
|
|||||||
# At least one per-subquery line.
|
# At least one per-subquery line.
|
||||||
self.assertIn("[Planner] sq1 label=", output)
|
self.assertIn("[Planner] sq1 label=", output)
|
||||||
|
|
||||||
|
def test_parallel_web_backend_enables_grounding_source(self):
|
||||||
|
plan = {
|
||||||
|
"intent": "news",
|
||||||
|
"freshness_mode": "balanced_recent",
|
||||||
|
"cluster_mode": "timeline",
|
||||||
|
"subqueries": [
|
||||||
|
{
|
||||||
|
"label": "primary",
|
||||||
|
"search_query": "test topic",
|
||||||
|
"ranking_query": "What happened with test topic?",
|
||||||
|
"sources": ["grounding"],
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"source_weights": {"grounding": 1.0},
|
||||||
|
}
|
||||||
|
report = pipeline.run(
|
||||||
|
topic="test topic",
|
||||||
|
config={"LAST30DAYS_REASONING_PROVIDER": "auto"},
|
||||||
|
depth="quick",
|
||||||
|
requested_sources=["grounding"],
|
||||||
|
web_backend="parallel",
|
||||||
|
external_plan=plan,
|
||||||
|
)
|
||||||
|
self.assertIn("grounding", report.errors_by_source)
|
||||||
|
self.assertIn("PARALLEL_API_KEY", report.errors_by_source["grounding"])
|
||||||
|
|
||||||
|
|
||||||
class TestSourceFetchCap(unittest.TestCase):
|
class TestSourceFetchCap(unittest.TestCase):
|
||||||
"""X source fetch count must be capped by MAX_SOURCE_FETCHES."""
|
"""X source fetch count must be capped by MAX_SOURCE_FETCHES."""
|
||||||
|
|||||||
Reference in New Issue
Block a user