diff --git a/prompt-service/src/unstract/prompt_service/controllers/answer_prompt.py b/prompt-service/src/unstract/prompt_service/controllers/answer_prompt.py index fdf0deaae1..cb47c415a7 100644 --- a/prompt-service/src/unstract/prompt_service/controllers/answer_prompt.py +++ b/prompt-service/src/unstract/prompt_service/controllers/answer_prompt.py @@ -34,8 +34,8 @@ answer_prompt_bp = Blueprint("answer-prompt", __name__) -@AuthHelper.auth_required @answer_prompt_bp.route("/answer-prompt", methods=["POST"]) +@AuthHelper.auth_required def prompt_processor() -> Any: platform_key = AuthHelper.get_token_from_auth_header(request) payload: dict[Any, Any] = request.json diff --git a/prompt-service/src/unstract/prompt_service/controllers/extraction.py b/prompt-service/src/unstract/prompt_service/controllers/extraction.py index 516894f429..2749018b9b 100644 --- a/prompt-service/src/unstract/prompt_service/controllers/extraction.py +++ b/prompt-service/src/unstract/prompt_service/controllers/extraction.py @@ -18,8 +18,8 @@ ] -@AuthHelper.auth_required @extraction_bp.route("/extract", methods=["POST"]) +@AuthHelper.auth_required def extract() -> Any: platform_key = AuthHelper.get_token_from_auth_header(request) payload: dict[Any, Any] = request.json diff --git a/prompt-service/src/unstract/prompt_service/controllers/indexing.py b/prompt-service/src/unstract/prompt_service/controllers/indexing.py index f9416a9c92..f412e4fbeb 100644 --- a/prompt-service/src/unstract/prompt_service/controllers/indexing.py +++ b/prompt-service/src/unstract/prompt_service/controllers/indexing.py @@ -32,8 +32,8 @@ ] -@AuthHelper.auth_required @indexing_bp.route("/index", methods=["POST"]) +@AuthHelper.auth_required def index() -> Any: """Endpoint for indexing documents into the vector database.