mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-15 18:25:18 +02:00
fix: Docs & Chats in other search spaces
This commit is contained in:
parent
73623aa37e
commit
2008b07304
6 changed files with 32 additions and 21 deletions
|
|
@ -88,16 +88,19 @@ async def create_chat(
|
|||
async def read_chats(
|
||||
skip: int = 0,
|
||||
limit: int = 100,
|
||||
search_space_id: int = None,
|
||||
session: AsyncSession = Depends(get_async_session),
|
||||
user: User = Depends(current_active_user)
|
||||
):
|
||||
try:
|
||||
query = select(Chat).join(SearchSpace).filter(SearchSpace.user_id == user.id)
|
||||
|
||||
# Filter by search_space_id if provided
|
||||
if search_space_id is not None:
|
||||
query = query.filter(Chat.search_space_id == search_space_id)
|
||||
|
||||
result = await session.execute(
|
||||
select(Chat)
|
||||
.join(SearchSpace)
|
||||
.filter(SearchSpace.user_id == user.id)
|
||||
.offset(skip)
|
||||
.limit(limit)
|
||||
query.offset(skip).limit(limit)
|
||||
)
|
||||
return result.scalars().all()
|
||||
except OperationalError:
|
||||
|
|
|
|||
|
|
@ -170,16 +170,19 @@ async def process_file_in_background(
|
|||
async def read_documents(
|
||||
skip: int = 0,
|
||||
limit: int = 300,
|
||||
search_space_id: int = None,
|
||||
session: AsyncSession = Depends(get_async_session),
|
||||
user: User = Depends(current_active_user)
|
||||
):
|
||||
try:
|
||||
query = select(Document).join(SearchSpace).filter(SearchSpace.user_id == user.id)
|
||||
|
||||
# Filter by search_space_id if provided
|
||||
if search_space_id is not None:
|
||||
query = query.filter(Document.search_space_id == search_space_id)
|
||||
|
||||
result = await session.execute(
|
||||
select(Document)
|
||||
.join(SearchSpace)
|
||||
.filter(SearchSpace.user_id == user.id)
|
||||
.offset(skip)
|
||||
.limit(limit)
|
||||
query.offset(skip).limit(limit)
|
||||
)
|
||||
db_documents = result.scalars().all()
|
||||
|
||||
|
|
|
|||
|
|
@ -116,16 +116,18 @@ async def create_search_source_connector(
|
|||
async def read_search_source_connectors(
|
||||
skip: int = 0,
|
||||
limit: int = 100,
|
||||
search_space_id: int = None,
|
||||
session: AsyncSession = Depends(get_async_session),
|
||||
user: User = Depends(current_active_user)
|
||||
):
|
||||
"""List all search source connectors for the current user."""
|
||||
try:
|
||||
query = select(SearchSourceConnector).filter(SearchSourceConnector.user_id == user.id)
|
||||
|
||||
# No need to filter by search_space_id as connectors are user-owned, not search space specific
|
||||
|
||||
result = await session.execute(
|
||||
select(SearchSourceConnector)
|
||||
.filter(SearchSourceConnector.user_id == user.id)
|
||||
.offset(skip)
|
||||
.limit(limit)
|
||||
query.offset(skip).limit(limit)
|
||||
)
|
||||
return result.scalars().all()
|
||||
except Exception as e:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue