26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
|
import asyncio
|
||
|
|
from celery_app import celery_app
|
||
|
|
|
||
|
|
@celery_app.task
|
||
|
|
def text_query_task(question: str):
|
||
|
|
from endpoints.text import text_query # Import inside the function to avoid circular import
|
||
|
|
loop = asyncio.new_event_loop()
|
||
|
|
asyncio.set_event_loop(loop)
|
||
|
|
return loop.run_until_complete(text_query(question=question)) # ✅ Correct way to call async functions in Celery
|
||
|
|
|
||
|
|
@celery_app.task
|
||
|
|
def image_query_task(file_path: str, question: str):
|
||
|
|
from endpoints.image import image_query # Import inside the function
|
||
|
|
with open(file_path, "rb") as file:
|
||
|
|
loop = asyncio.new_event_loop()
|
||
|
|
asyncio.set_event_loop(loop)
|
||
|
|
return loop.run_until_complete(image_query(file=file, question=question)) # ✅ Use event loop
|
||
|
|
|
||
|
|
@celery_app.task
|
||
|
|
def video_query_task(file_path: str, question: str):
|
||
|
|
from endpoints.video import video_query # Import inside the function
|
||
|
|
with open(file_path, "rb") as file:
|
||
|
|
loop = asyncio.new_event_loop()
|
||
|
|
asyncio.set_event_loop(loop)
|
||
|
|
return loop.run_until_complete(video_query(file=file, question=question)) # ✅ Use event loop
|