From 14f8c151c814fa8e00cd4ff8982d7572290ef88c Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Fri, 10 Nov 2023 17:48:54 -0800
Subject: [PATCH] Fix return type of the generate_chat_response method

---
 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..a5263d56 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, Union, Tuple, Dict
 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[str, str]]:
     def _save_to_conversation_log(
         q: str,
         chat_response: str,