apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-08 20:25:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #454 from MODSetter/dev

Browse source 
fix: podcast tasks for Windows compatibility
This commit is contained in:
Rohan Verma 2025-10-30 12:06:40 -07:00 • committed by GitHub
commit 5654f6c78f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
surfsense_backend/app/tasks/celery_tasks/podcast_tasks.py
View file

@ -1,6 +1,8 @@
"""Celery tasks for podcast generation.""" """Celery tasks for podcast generation."""
import asyncio
import logging import logging
import sys
from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
from sqlalchemy.pool import NullPool from sqlalchemy.pool import NullPool
@ -11,6 +13,14 @@ from app.tasks.podcast_tasks import generate_chat_podcast
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
if sys.platform.startswith("win"):
try:
asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
except AttributeError:
logger.warning(
"WindowsProactorEventLoopPolicy is unavailable; async subprocess support may fail."
)
def get_celery_session_maker(): def get_celery_session_maker():
""" """
@ -39,8 +49,6 @@ def generate_chat_podcast_task(
podcast_title: Title for the podcast podcast_title: Title for the podcast
user_id: ID of the user user_id: ID of the user
""" """
import asyncio
loop = asyncio.new_event_loop() loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop) asyncio.set_event_loop(loop)
@ -48,7 +56,9 @@ def generate_chat_podcast_task(
loop.run_until_complete( loop.run_until_complete(
_generate_chat_podcast(chat_id, search_space_id, podcast_title, user_id) _generate_chat_podcast(chat_id, search_space_id, podcast_title, user_id)
) )
loop.run_until_complete(loop.shutdown_asyncgens())
finally: finally:
asyncio.set_event_loop(None)
loop.close() loop.close()