sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update types of prev iterations response

Browse source
This commit is contained in:
sabaimran 2024-10-09 16:04:39 -07:00
parent 4978360852
commit 6960fb097c
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/khoj/routers/research.py
View file

@ -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,