Refactoring part 1
Some changes for workers, but stopped due to refactoring
This commit is contained in:
@@ -1,15 +1,17 @@
|
||||
import os
|
||||
from datetime import datetime as dt, timezone as tz
|
||||
from flask import request, redirect, url_for, flash, render_template, Blueprint, session, current_app
|
||||
from flask_security import hash_password, roles_required, roles_accepted, current_user
|
||||
from flask_security import roles_accepted, current_user
|
||||
from sqlalchemy import desc
|
||||
from sqlalchemy.orm import joinedload
|
||||
from werkzeug.utils import secure_filename
|
||||
|
||||
from ..models.document import Document, DocumentLanguage, DocumentVersion
|
||||
from ..extensions import db
|
||||
from common.models import Document, DocumentLanguage, DocumentVersion
|
||||
from common.extensions import db
|
||||
from .document_forms import AddDocumentForm
|
||||
from ..utils.middleware import mw_before_request
|
||||
from common.utils.middleware import mw_before_request
|
||||
from eveai_workers.tasks import create_embeddings
|
||||
|
||||
|
||||
document_bp = Blueprint('document_bp', __name__, url_prefix='/document')
|
||||
|
||||
@@ -68,11 +70,17 @@ def add_document():
|
||||
if error is None:
|
||||
flash('Document added successfully.', 'success')
|
||||
upload_file_for_version(new_doc_vers, file, extension)
|
||||
create_embeddings.delay(tenant_id=session['tenant']['id'],
|
||||
document_version_id=new_doc_vers.id,
|
||||
default_embedding_model=session['default_embedding_model'])
|
||||
current_app.logger.info(f'Document processing started for tenant {session["tenant"]["id"]}, '
|
||||
f'Document Version {new_doc_vers.id}')
|
||||
print('Processing should start soon')
|
||||
else:
|
||||
flash('Error adding document.', 'error')
|
||||
current_app.logger.error(f'Error adding document for tenant {session["tenant"]["id"]}: {error}')
|
||||
|
||||
# return render_template('document/add_document.html', form=form)
|
||||
return render_template('document/add_document.html', form=form)
|
||||
|
||||
|
||||
@document_bp.route('/documents', methods=['GET', 'POST'])
|
||||
|
||||
Reference in New Issue
Block a user