Just pass user rather than whole request object to doc search func

This commit is contained in:
Debanjum
2025-04-19 20:49:10 +05:30
parent b2390fa977
commit 21d19163ba
3 changed files with 3 additions and 7 deletions

View File

@@ -352,7 +352,7 @@ def set_user_name(
async def extract_references_and_questions(
request: Request,
user: KhojUser,
meta_log: dict,
q: str,
n: int,
@@ -367,8 +367,6 @@ async def extract_references_and_questions(
query_files: str = None,
tracer: dict = {},
):
user = request.user.object if request.user.is_authenticated else None
# Initialize Variables
compiled_references: List[dict[str, str]] = []
inferred_queries: List[str] = []

View File

@@ -889,7 +889,6 @@ async def chat(
if conversation_commands == [ConversationCommand.Research]:
async for research_result in execute_information_collection(
request=request,
user=user,
query=defiltered_query,
conversation_id=conversation_id,
@@ -1025,7 +1024,7 @@ async def chat(
if not ConversationCommand.Research in conversation_commands:
try:
async for result in extract_references_and_questions(
request,
user,
meta_log,
q,
(n or 7),

View File

@@ -193,7 +193,6 @@ async def apick_next_tool(
async def execute_information_collection(
request: Request,
user: KhojUser,
query: str,
conversation_id: str,
@@ -251,7 +250,7 @@ async def execute_information_collection(
c["query"] for iteration in previous_iterations if iteration.context for c in iteration.context
}
async for result in extract_references_and_questions(
request,
user,
construct_tool_chat_history(previous_iterations, ConversationCommand.Notes),
this_iteration.query,
7,