From f7e6f99a32f4bbf691b05db2f823e7e7db55126b Mon Sep 17 00:00:00 2001 From: sabaimran Date: Wed, 9 Oct 2024 16:05:34 -0700 Subject: [PATCH] add typing for extract document references --- src/khoj/routers/api_chat.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/khoj/routers/api_chat.py b/src/khoj/routers/api_chat.py index 64c50a1c..98598b36 100644 --- a/src/khoj/routers/api_chat.py +++ b/src/khoj/routers/api_chat.py @@ -6,7 +6,7 @@ import time import warnings from datetime import datetime from functools import partial -from typing import Dict, Optional +from typing import Any, Dict, List, Optional from urllib.parse import unquote from asgiref.sync import sync_to_async @@ -698,7 +698,9 @@ async def chat( researched_results = "" online_results: Dict = dict() ## Extract Document References - compiled_references, inferred_queries, defiltered_query = [], [], None + compiled_references: List[Any] = [] + inferred_queries: List[Any] = [] + defiltered_query: str = None if conversation_commands == [ConversationCommand.Default] or is_automated_task: async for research_result in execute_information_collection(