From 6960fb097c08e18c26631dbc5e5a0a9ea1482014 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Wed, 9 Oct 2024 16:04:39 -0700 Subject: [PATCH] update types of prev iterations response --- src/khoj/routers/research.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/khoj/routers/research.py b/src/khoj/routers/research.py index f55fc024..7fb25498 100644 --- a/src/khoj/routers/research.py +++ b/src/khoj/routers/research.py @@ -27,9 +27,7 @@ logger = logging.getLogger(__name__) class InformationCollectionIteration: - def __init__( - self, data_source: str, query: str, context: str = None, onlineContext: str = None, result: Any = None - ): + def __init__(self, data_source: str, query: str, context: str = None, onlineContext: dict = None): self.data_source = data_source self.query = query self.context = context @@ -260,7 +258,7 @@ async def execute_information_collection( if isinstance(response, dict) and ChatEvent.STATUS in response: yield response[ChatEvent.STATUS] else: - response_log = response + response_log = response # type: ignore previous_iterations.append( InformationCollectionIteration( data_source=this_iteration.data_source,