Skip to content

Commit 21d1916

Browse files
committed
Just pass user rather than whole request object to doc search func
1 parent b2390fa commit 21d1916

File tree

3 files changed

+3
-7
lines changed

3 files changed

+3
-7
lines changed

src/khoj/routers/api.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,7 @@ def set_user_name(
352352

353353

354354
async def extract_references_and_questions(
355-
request: Request,
355+
user: KhojUser,
356356
meta_log: dict,
357357
q: str,
358358
n: int,
@@ -367,8 +367,6 @@ async def extract_references_and_questions(
367367
query_files: str = None,
368368
tracer: dict = {},
369369
):
370-
user = request.user.object if request.user.is_authenticated else None
371-
372370
# Initialize Variables
373371
compiled_references: List[dict[str, str]] = []
374372
inferred_queries: List[str] = []

src/khoj/routers/api_chat.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -889,7 +889,6 @@ def collect_telemetry():
889889

890890
if conversation_commands == [ConversationCommand.Research]:
891891
async for research_result in execute_information_collection(
892-
request=request,
893892
user=user,
894893
query=defiltered_query,
895894
conversation_id=conversation_id,
@@ -1025,7 +1024,7 @@ def collect_telemetry():
10251024
if not ConversationCommand.Research in conversation_commands:
10261025
try:
10271026
async for result in extract_references_and_questions(
1028-
request,
1027+
user,
10291028
meta_log,
10301029
q,
10311030
(n or 7),

src/khoj/routers/research.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,6 @@ async def apick_next_tool(
193193

194194

195195
async def execute_information_collection(
196-
request: Request,
197196
user: KhojUser,
198197
query: str,
199198
conversation_id: str,
@@ -251,7 +250,7 @@ async def execute_information_collection(
251250
c["query"] for iteration in previous_iterations if iteration.context for c in iteration.context
252251
}
253252
async for result in extract_references_and_questions(
254-
request,
253+
user,
255254
construct_tool_chat_history(previous_iterations, ConversationCommand.Notes),
256255
this_iteration.query,
257256
7,

0 commit comments

Comments
 (0)