From 45b8670c25bb6c17c7ef022b22eab6d949acf3eb Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Fri, 10 Nov 2023 17:34:19 -0800
Subject: [PATCH] Fix return type hint for generate_chat_response func

---
 src/khoj/routers/helpers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py
index 99c1b281..e2e62d96 100644
--- a/src/khoj/routers/helpers.py
+++ b/src/khoj/routers/helpers.py
@@ -2,7 +2,7 @@ import logging
 import asyncio
 from datetime import datetime
 from functools import partial
-from typing import Iterator, List, Optional, Union
+from typing import Iterator, List, Optional, Tuple, Union
 from concurrent.futures import ThreadPoolExecutor
 
 from fastapi import HTTPException, Request
@@ -109,7 +109,7 @@ def generate_chat_response(
     inferred_queries: List[str] = [],
     conversation_command: ConversationCommand = ConversationCommand.Default,
     user: KhojUser = None,
-) -> Union[ThreadedGenerator, Iterator[str]]:
+) -> Tuple[Union[ThreadedGenerator, Iterator[str]], dict]:
     def _save_to_conversation_log(
         q: str,
         chat_response: str,