Show inferred document search queries in intermediate chat step on Web app

This commit is contained in:
Debanjum Singh Solanky 2024-04-11 19:20:04 +05:30
parent 997741119a
commit 653681967e
2 changed files with 5 additions and 2 deletions

View file

@ -5,7 +5,7 @@ import math
import os import os
import time import time
import uuid import uuid
from typing import Any, List, Optional, Union from typing import Any, Callable, List, Optional, Union
from asgiref.sync import sync_to_async from asgiref.sync import sync_to_async
from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile
@ -274,6 +274,7 @@ async def extract_references_and_questions(
d: float, d: float,
conversation_commands: List[ConversationCommand] = [ConversationCommand.Default], conversation_commands: List[ConversationCommand] = [ConversationCommand.Default],
location_data: LocationData = None, location_data: LocationData = None,
send_status_func: Optional[Callable] = None,
): ):
user = request.user.object if request.user.is_authenticated else None user = request.user.object if request.user.is_authenticated else None
@ -345,6 +346,8 @@ async def extract_references_and_questions(
with timer("Searching knowledge base took", logger): with timer("Searching knowledge base took", logger):
result_list = [] result_list = []
logger.info(f"🔍 Searching knowledge base with queries: {inferred_queries}") logger.info(f"🔍 Searching knowledge base with queries: {inferred_queries}")
if send_status_func:
await send_status_func(f"**🔍 Searching Documents for:** {'\n- ' + '\n- '.join(inferred_queries)}")
for query in inferred_queries: for query in inferred_queries:
n_items = min(n, 3) if using_offline_chat else n n_items = min(n, 3) if using_offline_chat else n
result_list.extend( result_list.extend(

View file

@ -371,7 +371,7 @@ async def websocket_endpoint(
q = q.replace(f"/{cmd.value}", "").strip() q = q.replace(f"/{cmd.value}", "").strip()
compiled_references, inferred_queries, defiltered_query = await extract_references_and_questions( compiled_references, inferred_queries, defiltered_query = await extract_references_and_questions(
websocket, None, meta_log, q, 7, 0.18, conversation_commands, location websocket, None, meta_log, q, 7, 0.18, conversation_commands, location, send_status_update
) )
if compiled_references: if compiled_references: