(feat:ingest_connectors) spread config params

This commit is contained in:
ManishMadan2882
2025-08-26 00:56:39 +05:30
parent e25b988dc8
commit 15a9e97a1e
3 changed files with 66 additions and 21 deletions

View File

@@ -913,10 +913,14 @@ class UploadRemote(Resource):
from application.api.user.tasks import ingest_connector_task
task = ingest_connector_task.delay(
source_config=config,
job_name=data["name"],
user=decoded_token.get("sub"),
source_type="google_drive"
source_type="google_drive",
session_token=session_token,
file_ids=file_ids,
folder_ids=folder_ids,
recursive=config.get("recursive", False),
retriever=config.get("retriever", "classic")
)
return make_response(jsonify({"success": True, "task_id": task.id}), 200)
task = ingest_remote.delay(

View File

@@ -48,9 +48,29 @@ def process_agent_webhook(self, agent_id, payload):
@celery.task(bind=True)
def ingest_connector_task(self, source_config, job_name, user, source_type, retriever="classic"):
def ingest_connector_task(
self,
job_name,
user,
source_type,
session_token=None,
file_ids=None,
folder_ids=None,
recursive=True,
retriever="classic"
):
from application.worker import ingest_connector
resp = ingest_connector(self, job_name, user, source_type, source_config, retriever)
resp = ingest_connector(
self,
job_name,
user,
source_type,
session_token=session_token,
file_ids=file_ids,
folder_ids=folder_ids,
recursive=recursive,
retriever=retriever
)
return resp