mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-10 05:39:11 +00:00
Add better handling for empty responses
This commit is contained in:
@@ -43,6 +43,7 @@ from khoj.routers.helpers import (
|
|||||||
construct_automation_created_message,
|
construct_automation_created_message,
|
||||||
create_automation,
|
create_automation,
|
||||||
get_conversation_command,
|
get_conversation_command,
|
||||||
|
is_query_empty,
|
||||||
is_ready_to_chat,
|
is_ready_to_chat,
|
||||||
text_to_image,
|
text_to_image,
|
||||||
update_telemetry_state,
|
update_telemetry_state,
|
||||||
@@ -480,6 +481,14 @@ async def websocket_endpoint(
|
|||||||
await send_rate_limit_message(e.detail)
|
await send_rate_limit_message(e.detail)
|
||||||
break
|
break
|
||||||
|
|
||||||
|
if is_query_empty(q):
|
||||||
|
await send_message("start_llm_response")
|
||||||
|
await send_message(
|
||||||
|
"It seems like your query is incomplete. Could you please provide more details or specify what you need help with?"
|
||||||
|
)
|
||||||
|
await send_message("end_llm_response")
|
||||||
|
continue
|
||||||
|
|
||||||
user_message_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
user_message_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
conversation_commands = [get_conversation_command(query=q, any_references=True)]
|
conversation_commands = [get_conversation_command(query=q, any_references=True)]
|
||||||
|
|
||||||
@@ -707,6 +716,12 @@ async def chat(
|
|||||||
) -> Response:
|
) -> Response:
|
||||||
user: KhojUser = request.user.object
|
user: KhojUser = request.user.object
|
||||||
q = unquote(q)
|
q = unquote(q)
|
||||||
|
if is_query_empty(q):
|
||||||
|
return Response(
|
||||||
|
content="It seems like your query is incomplete. Could you please provide more details or specify what you need help with?",
|
||||||
|
media_type="text/plain",
|
||||||
|
status_code=400,
|
||||||
|
)
|
||||||
user_message_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
user_message_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
logger.info(f"Chat request by {user.username}: {q}")
|
logger.info(f"Chat request by {user.username}: {q}")
|
||||||
|
|
||||||
|
|||||||
@@ -84,6 +84,10 @@ logger = logging.getLogger(__name__)
|
|||||||
executor = ThreadPoolExecutor(max_workers=1)
|
executor = ThreadPoolExecutor(max_workers=1)
|
||||||
|
|
||||||
|
|
||||||
|
def is_query_empty(query: str) -> bool:
|
||||||
|
return is_none_or_empty(query.strip())
|
||||||
|
|
||||||
|
|
||||||
def validate_conversation_config():
|
def validate_conversation_config():
|
||||||
default_config = ConversationAdapters.get_default_conversation_config()
|
default_config = ConversationAdapters.get_default_conversation_config()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user