klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-06 13:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
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: class InformationCollectionIteration:
def __init__( def __init__(self, data_source: str, query: str, context: str = None, onlineContext: dict = None):
self, data_source: str, query: str, context: str = None, onlineContext: str = None, result: Any = None
):
self.data_source = data_source self.data_source = data_source
self.query = query self.query = query
self.context = context self.context = context
@@ -260,7 +258,7 @@ async def execute_information_collection(
if isinstance(response, dict) and ChatEvent.STATUS in response: if isinstance(response, dict) and ChatEvent.STATUS in response:
yield response[ChatEvent.STATUS] yield response[ChatEvent.STATUS]
else: else:
response_log = response response_log = response # type: ignore
previous_iterations.append( previous_iterations.append(
InformationCollectionIteration( InformationCollectionIteration(
data_source=this_iteration.data_source, data_source=this_iteration.data_source,