From 53a79d9b65184d35aaa796355f06a1448f6f3546 Mon Sep 17 00:00:00 2001 From: Utkarsh-Patel-13 Date: Mon, 21 Jul 2025 09:54:33 -0700 Subject: [PATCH] Replaced deprecated stream service sources update call --- surfsense_backend/app/agents/researcher/nodes.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/surfsense_backend/app/agents/researcher/nodes.py b/surfsense_backend/app/agents/researcher/nodes.py index cbcd44f83..032d1cbec 100644 --- a/surfsense_backend/app/agents/researcher/nodes.py +++ b/surfsense_backend/app/agents/researcher/nodes.py @@ -810,8 +810,7 @@ async def fetch_relevant_documents( # After all sources are collected and deduplicated, stream them if streaming_service and writer: - streaming_service.only_update_sources(deduplicated_sources) - writer({"yield_value": streaming_service._format_annotations()}) + writer({"yield_value": streaming_service.format_sources_delta(deduplicated_sources)}) # Deduplicate raw documents based on chunk_id or content seen_chunk_ids = set()