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

Fix return type hint for generate_chat_response func

Browse source
This commit is contained in:
Debanjum Singh Solanky 2023-11-10 17:34:19 -08:00
parent c9c0ba67c6
commit 45b8670c25
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/khoj/routers/helpers.py
View file

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