vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2025-09-14 16:19:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

feat: Updated the extension for SurfSense v0.0.6

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-03-26 20:02:53 -07:00
parent 24fd873ca7
commit 8cd1264d3f
4 changed files with 74 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

32
surfsense_backend/app/routes/search_source_connectors_routes.py
View file

@ -255,8 +255,7 @@ async def index_connector_content(
# Add the indexing task to background tasks
if background_tasks:
background_tasks.add_task(
run_slack_indexing,
session,
run_slack_indexing_with_new_session,
connector_id,
search_space_id
)
@ -292,8 +291,7 @@ async def index_connector_content(
# Add the indexing task to background tasks
if background_tasks:
background_tasks.add_task(
run_notion_indexing,
session,
run_notion_indexing_with_new_session,
connector_id,
search_space_id
)
@ -355,6 +353,19 @@ async def update_connector_last_indexed(
logger.error(f"Failed to update last_indexed_at for connector {connector_id}: {str(e)}")
await session.rollback()
async def run_slack_indexing_with_new_session(
connector_id: int,
search_space_id: int
):
"""
Create a new session and run the Slack indexing task.
This prevents session leaks by creating a dedicated session for the background task.
"""
from app.db import async_session_maker
async with async_session_maker() as session:
await run_slack_indexing(session, connector_id, search_space_id)
async def run_slack_indexing(
session: AsyncSession,
connector_id: int,
@ -386,6 +397,19 @@ async def run_slack_indexing(
except Exception as e:
logger.error(f"Error in background Slack indexing task: {str(e)}")
async def run_notion_indexing_with_new_session(
connector_id: int,
search_space_id: int
):
"""
Create a new session and run the Notion indexing task.
This prevents session leaks by creating a dedicated session for the background task.
"""
from app.db import async_session_maker
async with async_session_maker() as session:
await run_notion_indexing(session, connector_id, search_space_id)
async def run_notion_indexing(
session: AsyncSession,
connector_id: int,