diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index d17b03b882..7320d77816 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -663,16 +663,16 @@ async def chat_completion_files_handler( if len(queries) == 0: queries = [get_last_user_message(body["messages"])] - # await __event_emitter__( - # { - # "type": "status", - # "data": { - # "action": "queries_generated", - # "queries": queries, - # "done": True, - # }, - # } - # ) + await __event_emitter__( + { + "type": "status", + "data": { + "action": "queries_generated", + "queries": queries, + "done": False, + }, + } + ) try: # Offload get_sources_from_items to a separate thread @@ -710,6 +710,21 @@ async def chat_completion_files_handler( log.debug(f"rag_contexts:sources: {sources}") + sources_count = 0 + for source in sources: + sources_count += len(source.get("document", [])) + + await __event_emitter__( + { + "type": "status", + "data": { + "action": "sources_retrieved", + "count": sources_count, + "done": True, + }, + } + ) + return body, {"sources": sources} diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte index 271a49e940..35aaf111e2 100644 --- a/src/lib/components/chat/Messages/ResponseMessage.svelte +++ b/src/lib/components/chat/Messages/ResponseMessage.svelte @@ -643,7 +643,10 @@