sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-24 07:55:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 33

Fix return type of the generate_chat_response method

Browse source
This commit is contained in:
sabaimran 2023-11-10 17:48:54 -08:00
parent 54d4fd0e08
commit 14f8c151c8
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 import asyncio
from datetime import datetime from datetime import datetime
from functools import partial 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 concurrent.futures import ThreadPoolExecutor
from fastapi import HTTPException, Request from fastapi import HTTPException, Request
@ -109,7 +109,7 @@ def generate_chat_response(
inferred_queries: List[str] = [], inferred_queries: List[str] = [],
conversation_command: ConversationCommand = ConversationCommand.Default, conversation_command: ConversationCommand = ConversationCommand.Default,
user: KhojUser = None, user: KhojUser = None,
) -> Union[ThreadedGenerator, Iterator[str]]: ) -> Tuple[Union[ThreadedGenerator, Iterator[str]], Dict[str, str]]:
def _save_to_conversation_log( def _save_to_conversation_log(
q: str, q: str,
chat_response: str, chat_response: str,