Skip to content

Commit 16a947f

Browse files
committed
return entire doc group instead of name
1 parent 9a1beb5 commit 16a947f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ai_ta_backend/service/retrieval_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ async def getTopContexts(self,
141141
disabled_doc_groups_response, public_doc_groups_response, user_query_embedding = await asyncio.gather(*tasks)
142142

143143
disabled_doc_groups = [doc_group['name'] for doc_group in disabled_doc_groups_response["data"]]
144-
public_doc_groups = [doc_group['name'] for doc_group in public_doc_groups_response["data"]]
144+
public_doc_groups = public_doc_groups_response["data"]
145145

146146
time_for_parallel_operations = time.monotonic() - start_time_overall
147147
start_time_vector_search = time.monotonic()

0 commit comments

Comments
 (0)