Setup of documents view
This commit is contained in:
@@ -5,12 +5,14 @@ from flask_security import SQLAlchemyUserDatastore
|
||||
from flask_security.signals import user_authenticated
|
||||
from werkzeug.middleware.proxy_fix import ProxyFix
|
||||
import logging.config
|
||||
from celery import Celery
|
||||
|
||||
from .extensions import db, migrate, bootstrap, security, mail, login_manager, cors
|
||||
from .models.user import User, Tenant, Role
|
||||
from .models.document import Document, DocumentLanguage, DocumentVersion
|
||||
from .logging_config import LOGGING
|
||||
from .utils.security import set_tenant_session_data
|
||||
from .worker.celery_utils import init_celery
|
||||
|
||||
|
||||
def create_app(config_file=None):
|
||||
@@ -43,6 +45,9 @@ def create_app(config_file=None):
|
||||
mail_logger.setLevel(logging.DEBUG)
|
||||
security_logger = logging.getLogger('flask_security')
|
||||
security_logger.setLevel(logging.DEBUG)
|
||||
sqlalchemy_logger = logging.getLogger('sqlalchemy.engine')
|
||||
sqlalchemy_logger.setLevel(logging.DEBUG)
|
||||
# sqlalchemy_logger.addHandler(logging.StreamHandler())
|
||||
|
||||
# Register API
|
||||
register_api(app)
|
||||
@@ -73,3 +78,18 @@ def register_api(app):
|
||||
pass
|
||||
# from . import api
|
||||
# app.register_blueprint(api.bp, url_prefix='/api')
|
||||
|
||||
|
||||
def create_celery_app(config_file=None):
|
||||
app = Flask(__name__)
|
||||
if config_file is None:
|
||||
app.config.from_object('config.DevConfig')
|
||||
else:
|
||||
app.config.from_object(config_file)
|
||||
|
||||
celery = Celery(app.import_name)
|
||||
init_celery(celery, app)
|
||||
return celery
|
||||
|
||||
|
||||
celery = create_celery_app()
|
||||
|
||||
@@ -1,3 +0,0 @@
|
||||
from ..models.document import Document, DocumentLanguage, DocumentVersion
|
||||
|
||||
def process_document(file, name, language, valid_from):
|
||||
@@ -27,6 +27,7 @@ class DocumentLanguage(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
document_id = db.Column(db.Integer, db.ForeignKey(Document.id), nullable=False)
|
||||
language = db.Column(db.String(2), nullable=False)
|
||||
latest_version_id = db.Column(db.Integer, db.ForeignKey('document_version.id'), nullable=True)
|
||||
|
||||
# Versioning Information
|
||||
created_at = db.Column(db.DateTime, nullable=False, server_default=db.func.now())
|
||||
@@ -35,7 +36,17 @@ class DocumentLanguage(db.Model):
|
||||
updated_by = db.Column(db.Integer, db.ForeignKey(User.id))
|
||||
|
||||
# Relations
|
||||
versions = db.relationship('DocumentVersion', backref='document_language', lazy=True)
|
||||
versions = db.relationship(
|
||||
'DocumentVersion',
|
||||
backref='document_language',
|
||||
lazy='joined',
|
||||
foreign_keys='DocumentVersion.doc_lang_id'
|
||||
)
|
||||
latest_version = db.relationship(
|
||||
'DocumentVersion',
|
||||
uselist=False,
|
||||
foreign_keys=[latest_version_id]
|
||||
)
|
||||
|
||||
def __repr__(self):
|
||||
return f"<DocumentLanguage {self.document_id}.{self.language}>"
|
||||
|
||||
18
eveai_app/templates/document/documents.html
Normal file
18
eveai_app/templates/document/documents.html
Normal file
@@ -0,0 +1,18 @@
|
||||
{% extends 'base.html' %}
|
||||
{% from 'macros.html' import render_nested_table, render_pagination %}
|
||||
|
||||
{% block title %}Documents{% endblock %}
|
||||
|
||||
{% block content_title %}Documents{% endblock %}
|
||||
{% block content_description %}View Documents for Tenant{% endblock %}
|
||||
|
||||
{% block content %}
|
||||
<div class="container">
|
||||
<!-- Documents Table -->
|
||||
{{ render_nested_table(headers=["Name", "Created At", "Valid From", "Languages & Versions"], rows=rows) }}
|
||||
</div>
|
||||
{% endblock %}
|
||||
|
||||
{% block content_footer %}
|
||||
{{ render_pagination(pagination, 'document_bp.documents') }}
|
||||
{% endblock %}
|
||||
@@ -22,3 +22,151 @@
|
||||
{% endif %}
|
||||
{% endif %}
|
||||
{% endmacro %}
|
||||
|
||||
{% macro render_table(headers, rows) %}
|
||||
<div class="card">
|
||||
<div class="table-responsive">
|
||||
<table class="table align-items-center mb-0">
|
||||
<thead>
|
||||
<tr>
|
||||
{% for header in headers %}
|
||||
<th class="text-uppercase text-secondary text-xxs font-weight-bolder opacity-7">{{ header }}</th>
|
||||
{% endfor %}
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{% for row in rows %}
|
||||
<tr>
|
||||
{% for cell in row %}
|
||||
<td class="{{ cell.class }}">
|
||||
{% if cell.type == 'image' %}
|
||||
<div class="d-flex px-2 py-1">
|
||||
<div>
|
||||
<img src="{{ cell.value }}" class="avatar avatar-sm me-3">
|
||||
</div>
|
||||
</div>
|
||||
{% elif cell.type == 'text' %}
|
||||
<p class="text-xs {{ cell.text_class }}">{{ cell.value }}</p>
|
||||
{% elif cell.type == 'badge' %}
|
||||
<span class="badge badge-sm {{ cell.badge_class }}">{{ cell.value }}</span>
|
||||
{% elif cell.type == 'link' %}
|
||||
<a href="{{ cell.href }}" class="text-secondary font-weight-normal text-xs" data-toggle="tooltip" data-original-title="{{ cell.title }}">{{ cell.value }}</a>
|
||||
{% else %}
|
||||
{{ cell.value }}
|
||||
{% endif %}
|
||||
</td>
|
||||
{% endfor %}
|
||||
</tr>
|
||||
{% endfor %}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
</div>
|
||||
{% endmacro %}
|
||||
|
||||
{% macro render_accordion(accordion_id, accordion_items, header_title, header_description) %}
|
||||
<div class="accordion-1">
|
||||
<div class="container">
|
||||
<div class="row my-5">
|
||||
<div class="col-md-6 mx-auto text-center">
|
||||
<h2>{{ header_title }}</h2>
|
||||
<p>{{ header_description }}</p>
|
||||
</div>
|
||||
</div>
|
||||
<div class="row">
|
||||
<div class="col-md-10 mx-auto">
|
||||
<div class="accordion" id="{{ accordion_id }}">
|
||||
{% for item in accordion_items %}
|
||||
<div class="accordion-item mb-3">
|
||||
<h5 class="accordion-header" id="heading{{ loop.index }}">
|
||||
<button class="accordion-button border-bottom font-weight-bold collapsed" type="button" data-bs-toggle="collapse" data-bs-target="#collapse{{ loop.index }}" aria-expanded="false" aria-controls="collapse{{ loop.index }}">
|
||||
{{ item.title }}
|
||||
<i class="collapse-close fa fa-plus text-xs pt-1 position-absolute end-0 me-3" aria-hidden="true"></i>
|
||||
<i class="collapse-open fa fa-minus text-xs pt-1 position-absolute end-0 me-3" aria-hidden="true"></i>
|
||||
</button>
|
||||
</h5>
|
||||
<div id="collapse{{ loop.index }}" class="accordion-collapse collapse" aria-labelledby="heading{{ loop.index }}" data-bs-parent="#{{ accordion_id }}">
|
||||
<div class="accordion-body text-sm opacity-8">
|
||||
{{ item.content }}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% endfor %}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% endmacro %}
|
||||
|
||||
{% macro render_nested_table(headers, rows) %}
|
||||
<div class="">
|
||||
<div class="table-responsive">
|
||||
<table class="table align-items-center mb-0">
|
||||
<thead>
|
||||
<tr>
|
||||
{% for header in headers %}
|
||||
<th class="text-uppercase text-secondary text-xxs font-weight-bolder opacity-7">{{ header }}</th>
|
||||
{% endfor %}
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{% for row in rows %}
|
||||
<tr>
|
||||
{% for cell in row %}
|
||||
{% if cell.is_group %}
|
||||
<td colspan="{{ cell.colspan }}" class="{{ cell.class }}">
|
||||
{{ render_nested_table(cell.headers, cell.sub_rows) }}
|
||||
</td>
|
||||
{% else %}
|
||||
<td class="{{ cell.class }}">
|
||||
{% if cell.type == 'image' %}
|
||||
<div class="d-flex px-2 py-1">
|
||||
<div>
|
||||
<img src="{{ cell.value }}" class="avatar avatar-sm me-3">
|
||||
</div>
|
||||
</div>
|
||||
{% elif cell.type == 'text' %}
|
||||
<p class="text-xs {{ cell.text_class }}">{{ cell.value }}</p>
|
||||
{% elif cell.type == 'badge' %}
|
||||
<span class="badge badge-sm {{ cell.badge_class }}">{{ cell.value }}</span>
|
||||
{% elif cell.type == 'link' %}
|
||||
<a href="{{ cell.href }}" class="text-secondary font-weight-normal text-xs" data-toggle="tooltip" data-original-title="{{ cell.title }}">{{ cell.value }}</a>
|
||||
{% else %}
|
||||
{{ cell.value }}
|
||||
{% endif %}
|
||||
</td>
|
||||
{% endif %}
|
||||
{% endfor %}
|
||||
</tr>
|
||||
{% endfor %}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
</div>
|
||||
{% endmacro %}
|
||||
|
||||
{% macro render_pagination(pagination, endpoint) %}
|
||||
<nav aria-label="Page navigation">
|
||||
<ul class="pagination">
|
||||
<li class="page-item {{ 'disabled' if not pagination.has_prev }}">
|
||||
<a class="page-link" href="{{ url_for(endpoint, page=pagination.prev_num) if pagination.has_prev else 'javascript:;' }}" tabindex="-1">
|
||||
<i class="fa fa-angle-left"></i>
|
||||
<span class="sr-only">Previous</span>
|
||||
</a>
|
||||
</li>
|
||||
{% for page in pagination.iter_pages(left_edge=1, left_current=2, right_current=3, right_edge=1) %}
|
||||
<li class="page-item {{ 'active' if page == pagination.page }}">
|
||||
<a class="page-link" href="{{ url_for(endpoint, page=page) }}">{{ page }} {% if page == pagination.page %}<span class="sr-only">(current)</span>{% endif %}</a>
|
||||
</li>
|
||||
{% endfor %}
|
||||
<li class="page-item {{ 'disabled' if not pagination.has_next }}">
|
||||
<a class="page-link" href="{{ url_for(endpoint, page=pagination.next_num) if pagination.has_next else 'javascript:;' }}">
|
||||
<i class="fa fa-angle-right"></i>
|
||||
<span class="sr-only">Next</span>
|
||||
</a>
|
||||
</li>
|
||||
</ul>
|
||||
</nav>
|
||||
{% endmacro %}
|
||||
|
||||
|
||||
@@ -4,7 +4,7 @@ from sqlalchemy import text
|
||||
from sqlalchemy.schema import CreateSchema
|
||||
from sqlalchemy.exc import InternalError
|
||||
from sqlalchemy.orm import sessionmaker, scoped_session
|
||||
from flask_migrate import heads
|
||||
from flask import current_app
|
||||
|
||||
from ..extensions import db, migrate
|
||||
|
||||
@@ -37,9 +37,10 @@ class Database:
|
||||
db.session.execute(CreateSchema(self.schema))
|
||||
db.session.execute(text(f"CREATE EXTENSION IF NOT EXISTS pgvector SCHEMA {self.schema}"))
|
||||
db.session.commit()
|
||||
except InternalError:
|
||||
except InternalError as e:
|
||||
db.session.rollback()
|
||||
db.session.close()
|
||||
current_app.logger.error(f"Error creating schema {self.schema}: {e.args}")
|
||||
|
||||
def create_tables(self):
|
||||
"""create tables in for schema"""
|
||||
|
||||
@@ -16,7 +16,7 @@ class SessionDefaultsForm(FlaskForm):
|
||||
default_llm_model = SelectField('Default LLM Model', choices=[], validators=[DataRequired()])
|
||||
|
||||
def __init__(self):
|
||||
super(SessionDefaultsForm, self).__init__()
|
||||
super().__init__()
|
||||
self.user_name.data = current_user.user_name
|
||||
self.user_email.data = current_user.email
|
||||
self.tenant_name.data = session.get('tenant').get('name')
|
||||
|
||||
@@ -10,14 +10,20 @@ class AddDocumentForm(FlaskForm):
|
||||
file = FileField('File', validators=[FileAllowed(['pdf', 'txt']),
|
||||
FileRequired()])
|
||||
name = StringField('Name', validators=[Length(max=100)])
|
||||
language = StringField('Language', validators=[Length(max=2)])
|
||||
language = SelectField('Language', choices=[], validators=[Optional()])
|
||||
valid_from = DateField('Valid from', id='form-control datepicker', validators=[Optional()])
|
||||
doc_embedding_model = SelectField('Default Embedding Model', choices=[], validators=[DataRequired()])
|
||||
|
||||
submit = SubmitField('Submit')
|
||||
|
||||
def __init__(self):
|
||||
super(AddDocumentForm, self).__init__()
|
||||
super().__init__()
|
||||
self.language.choices = [(language, language) for language in
|
||||
session.get('tenant').get('allowed_languages')]
|
||||
self.language.data = session.get('default_language')
|
||||
self.doc_embedding_model.choices = [(model, model) for model in
|
||||
session.get('tenant').get('allowed_embedding_models')]
|
||||
self.doc_embedding_model.data = session.get('default_embedding_model')
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -2,12 +2,13 @@ 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 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 .document_forms import AddDocumentForm
|
||||
from ..utils.database import Database
|
||||
from ..utils.middleware import mw_before_request
|
||||
|
||||
document_bp = Blueprint('document_bp', __name__, url_prefix='/document')
|
||||
@@ -37,74 +38,177 @@ def add_document():
|
||||
else:
|
||||
new_doc.name = form.name.data
|
||||
|
||||
timestamp = dt.now(tz.utc)
|
||||
if form.valid_from.data or form.valid_from.data != '':
|
||||
new_doc.valid_from = form.valid_from.data
|
||||
else:
|
||||
new_doc.valid_from = timestamp
|
||||
|
||||
new_doc.valid_from = dt.now(tz.utc)
|
||||
new_doc.tenant_id = session['tenant']['id']
|
||||
|
||||
new_doc.created_at = timestamp
|
||||
new_doc.updated_at = timestamp
|
||||
new_doc.created_by = current_user.id
|
||||
new_doc.updated_by = current_user.id
|
||||
set_logging_information(new_doc, dt.now(tz.utc))
|
||||
|
||||
# Create the DocumentLanguage
|
||||
new_doc_lang = DocumentLanguage()
|
||||
if form.language.data == '':
|
||||
new_doc_lang.language = session['default_language']
|
||||
else:
|
||||
new_doc_lang.language = form.language.data
|
||||
|
||||
new_doc_lang.document = new_doc
|
||||
|
||||
new_doc_lang.created_at = timestamp
|
||||
new_doc_lang.updated_at = timestamp
|
||||
new_doc_lang.created_by = current_user.id
|
||||
new_doc_lang.updated_by = current_user.id
|
||||
new_doc_lang = create_language_for_document(new_doc, form.language.data)
|
||||
|
||||
# Create the DocumentVersion
|
||||
new_doc_vers = DocumentVersion()
|
||||
new_doc_vers.document_language = new_doc_lang
|
||||
|
||||
new_doc_vers.created_at = timestamp
|
||||
new_doc_vers.updated_at = timestamp
|
||||
new_doc_vers.created_by = current_user.id
|
||||
new_doc_vers.updated_by = current_user.id
|
||||
set_logging_information(new_doc_vers, dt.now(tz.utc))
|
||||
|
||||
try:
|
||||
db.session.add(new_doc)
|
||||
db.session.add(new_doc_lang)
|
||||
db.session.add(new_doc_vers)
|
||||
db.session.commit()
|
||||
new_doc_lang.latest_version = new_doc_vers
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
error = e.args
|
||||
|
||||
# Save the file and process the document
|
||||
if error is None:
|
||||
flash('Document added successfully.')
|
||||
new_doc_vers.file_type = extension
|
||||
new_doc_vers.file_name = new_doc_vers.calc_file_name()
|
||||
new_doc_vers.file_location = new_doc_vers.calc_file_location()
|
||||
|
||||
upload_path = os.path.join(current_app.config['UPLOAD_FOLDER'], new_doc_vers.file_location)
|
||||
if not os.path.exists(upload_path):
|
||||
os.makedirs(upload_path, exist_ok=True)
|
||||
file.save(os.path.join(upload_path, new_doc_vers.file_name))
|
||||
try:
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
error = e.args
|
||||
if error is None:
|
||||
flash('Document saved successfully.')
|
||||
# TODO: processing of document to embeddings (async)
|
||||
flash('Document processing started.')
|
||||
else:
|
||||
flash('Error saving document.')
|
||||
flash('Document added successfully.', 'success')
|
||||
upload_file_for_version(new_doc_vers, file, extension)
|
||||
else:
|
||||
flash('Error adding document.')
|
||||
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'])
|
||||
@roles_accepted('Super User', 'Tenant Admin')
|
||||
def documents():
|
||||
page = request.args.get('page', 1, type=int)
|
||||
per_page = request.args.get('per_page', 10, type=int)
|
||||
|
||||
query = Document.query.order_by(desc(Document.created_at)).options(
|
||||
joinedload(Document.languages).joinedload(DocumentLanguage.versions))
|
||||
|
||||
pagination = query.paginate(page=page, per_page=per_page, error_out=False)
|
||||
docs = pagination.items
|
||||
|
||||
rows = prepare_document_data(docs)
|
||||
|
||||
return render_template('document/documents.html', rows=rows, pagination=pagination)
|
||||
|
||||
|
||||
@document_bp.route('/process_version/<int:version_id>', methods=['POST'])
|
||||
@roles_accepted('Super User', 'Tenant Admin')
|
||||
def process_version(version_id):
|
||||
version = DocumentVersion.query.get_or_404(version_id)
|
||||
if not version.processing:
|
||||
print(f'Placeholder for processing version: {version_id}')
|
||||
|
||||
return redirect(url_for('documents'))
|
||||
|
||||
|
||||
def set_logging_information(obj, timestamp):
|
||||
obj.created_at = timestamp
|
||||
obj.updated_at = timestamp
|
||||
obj.created_by = current_user.id
|
||||
obj.updated_by = current_user.id
|
||||
|
||||
|
||||
def create_language_for_document(document, language):
|
||||
new_doc_lang = DocumentLanguage()
|
||||
if language == '':
|
||||
new_doc_lang.language = session['default_language']
|
||||
else:
|
||||
new_doc_lang.language = language
|
||||
|
||||
new_doc_lang.document = document
|
||||
|
||||
set_logging_information(new_doc_lang, dt.now(tz.utc))
|
||||
|
||||
return new_doc_lang
|
||||
|
||||
|
||||
def upload_file_for_version(doc_vers, file, extension):
|
||||
error = None
|
||||
doc_vers.file_type = extension
|
||||
doc_vers.file_name = doc_vers.calc_file_name()
|
||||
doc_vers.file_location = doc_vers.calc_file_location()
|
||||
|
||||
upload_path = os.path.join(current_app.config['UPLOAD_FOLDER'], doc_vers.file_location)
|
||||
if not os.path.exists(upload_path):
|
||||
os.makedirs(upload_path, exist_ok=True)
|
||||
file.save(os.path.join(upload_path, doc_vers.file_name))
|
||||
try:
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
error = e.args
|
||||
if error is None:
|
||||
flash('Document saved successfully.', 'success')
|
||||
current_app.logger.info(f'Starting Doucment processing for tenant {session['tenant']['id']} for document '
|
||||
f'version {doc_vers.id}')
|
||||
# TODO: processing of document to embeddings (async)
|
||||
flash('Document processing started.', 'info')
|
||||
else:
|
||||
flash('Error saving document.', 'error')
|
||||
current_app.logger.error(f'Error saving document for tenant {session["tenant"]["id"]}: {error}')
|
||||
|
||||
|
||||
# Sample code for adding or updating versions and ensuring latest_version is set in DocumentLanguage
|
||||
# def add_or_update_version(language_id, version_data):
|
||||
# new_version = Version(language_id=language_id, **version_data)
|
||||
# db.session.add(new_version)
|
||||
# db.session.flush() # Ensures new_version gets an ID assigned if it's new
|
||||
#
|
||||
# # Assuming we always call this when we know it's the latest
|
||||
# language = Language.query.get(language_id)
|
||||
# language.latest_version_id = new_version.id
|
||||
# db.session.commit()
|
||||
|
||||
# sample code for using latest_version in the application
|
||||
# @app.route('/language/<int:language_id>')
|
||||
# def show_language(language_id):
|
||||
# language = Language.query.get_or_404(language_id)
|
||||
# latest_version = language.latest_version # This is now a direct, efficient database access
|
||||
# return render_template('language_details.html', language=language, latest_version=latest_version)
|
||||
|
||||
|
||||
def prepare_document_data(docs):
|
||||
rows = []
|
||||
for doc in docs:
|
||||
doc_row = [{'value': doc.name, 'class': '', 'type': 'text'},
|
||||
{'value': doc.created_at.strftime("%Y-%m-%d %H:%M:%S"), 'class': '', 'type': 'text'}]
|
||||
# Document basic details
|
||||
if doc.valid_from:
|
||||
doc_row.append({'value': doc.valid_from.strftime("%Y-%m-%d"), 'class': '', 'type': 'text'})
|
||||
else:
|
||||
doc_row.append({'value': '', 'class': '', 'type': 'text'})
|
||||
|
||||
# Nested languages and versions
|
||||
languages_rows = []
|
||||
for lang in doc.languages:
|
||||
lang_row = [{'value': lang.language, 'class': '', 'type': 'text'}]
|
||||
|
||||
# Latest version details if available (should be available ;-) )
|
||||
if lang.latest_version:
|
||||
lang_row.append({'value': lang.latest_version.created_at.strftime("%Y-%m-%d %H:%M:%S"),
|
||||
'class': '', 'type': 'text'})
|
||||
if lang.latest_version.url:
|
||||
lang_row.append({'value': lang.latest_version.url,
|
||||
'class': '', 'type': 'link', 'href': lang.latest_version.url})
|
||||
else:
|
||||
lang_row.append({'value': '', 'class': '', 'type': 'text'})
|
||||
|
||||
if lang.latest_version.file_name:
|
||||
lang_row.append({'value': lang.latest_version.file_name, 'class': '', 'type': 'text'})
|
||||
else:
|
||||
lang_row.append({'value': '', 'class': '', 'type': 'text'})
|
||||
|
||||
if lang.latest_version.file_type:
|
||||
lang_row.append({'value': lang.latest_version.file_type, 'class': '', 'type': 'text'})
|
||||
else:
|
||||
lang_row.append({'value': '', 'class': '', 'type': 'text'})
|
||||
# Include other details as necessary
|
||||
|
||||
languages_rows.append(lang_row)
|
||||
|
||||
doc_row.append({'is_group': True, 'colspan': '4',
|
||||
'headers': ['Language', 'Latest Version', 'URL', 'File Name', 'Type'],
|
||||
'sub_rows': languages_rows})
|
||||
rows.append(doc_row)
|
||||
return rows
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
# from . import user_bp
|
||||
import uuid
|
||||
from datetime import datetime as dt, timezone as tz
|
||||
from flask import request, redirect, url_for, flash, render_template, Blueprint, session
|
||||
from flask import request, redirect, url_for, flash, render_template, Blueprint, session, current_app
|
||||
from flask_security import hash_password, roles_required, roles_accepted
|
||||
|
||||
from ..models.user import User, Tenant, Role
|
||||
@@ -83,9 +83,12 @@ def tenant():
|
||||
|
||||
# Create schema for new tenant
|
||||
if error is None:
|
||||
current_app.logger.info(f"Successfully created tenant {new_tenant.id} in Database")
|
||||
current_app.logger.info(f"Creating schema for tenant {new_tenant.id}")
|
||||
Database(new_tenant.id).create_tenant_schema()
|
||||
|
||||
flash(error) if error else flash('Tenant added successfully.')
|
||||
else:
|
||||
current_app.logger.error(f"Error creating tenant: {error}")
|
||||
flash(error) if error else flash('Tenant added successfully.')
|
||||
|
||||
form = TenantForm()
|
||||
return render_template('user/tenant.html', form=form)
|
||||
|
||||
9
eveai_app/worker/celery_utils.py
Normal file
9
eveai_app/worker/celery_utils.py
Normal file
@@ -0,0 +1,9 @@
|
||||
def init_celery(celery, app):
|
||||
celery.conf.update(app.config) # Load all configurations form Flask app including Queue settings
|
||||
|
||||
class ContextTask(celery.Task):
|
||||
def __call__(self, *args, **kwargs):
|
||||
with app.app_context():
|
||||
return self.run(*args, **kwargs)
|
||||
|
||||
celery.Task = ContextTask
|
||||
59
eveai_app/worker/tasks.py
Normal file
59
eveai_app/worker/tasks.py
Normal file
@@ -0,0 +1,59 @@
|
||||
from datetime import datetime as dt, timezone as tz
|
||||
from flask import current_app
|
||||
from langchain_mistralai import MistralAIEmbeddings
|
||||
from langchain_openai import OpenAIEmbeddings
|
||||
from langchain_community.document_loaders.pdf import PyPDFLoader
|
||||
from langchain_community.vectorstores.chroma import Chroma
|
||||
from langchain_text_splitters import CharacterTextSplitter
|
||||
import os
|
||||
|
||||
from eveai_app import celery
|
||||
from ..utils.database import Database
|
||||
from ..models.document import DocumentVersion, EmbeddingMistral, EmbeddingSmallOpenAI
|
||||
from .. import db
|
||||
|
||||
|
||||
@celery.task(name='create_embeddings', queue='embeddings')
|
||||
def create_embeddings(tenant_id, document_version_id, embedding_model_def):
|
||||
current_app.logger.info(f'Creating embeddings for tenant {tenant_id} on document version {document_version_id} '
|
||||
f'with model {embedding_model_def}')
|
||||
Database(tenant_id).switch_schema()
|
||||
document_version = DocumentVersion.query.get(document_version_id)
|
||||
if document_version is None:
|
||||
current_app.logger.error(f'Cannot create embeddings for tenant {tenant_id}. '
|
||||
f'Document version {document_version_id} not found')
|
||||
return
|
||||
db.session.add(document_version)
|
||||
|
||||
# start processing
|
||||
document_version.processing = True
|
||||
document_version.processing_started_at = dt.now(tz.utc)
|
||||
db.session.commit()
|
||||
|
||||
embedding_provider = embedding_model_def.rsplit('.', 1)[0]
|
||||
embedding_model = embedding_model_def.rsplit('.', 1)[1]
|
||||
# define embedding variables
|
||||
match (embedding_provider, embedding_model):
|
||||
case ('openai', 'text-embedding-3-small'):
|
||||
embedding_model = EmbeddingSmallOpenAI()
|
||||
case ('mistral', 'text-embedding-3-small'):
|
||||
embedding_model = EmbeddingMistral()
|
||||
|
||||
match document_version.file_type:
|
||||
case 'pdf':
|
||||
pdf_file = os.path.join(current_app.config['UPLOAD_FOLDER'],
|
||||
document_version.file_location,
|
||||
document_version.file_path)
|
||||
loader = PyPDFLoader(pdf_file)
|
||||
|
||||
# We
|
||||
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
|
||||
documents = text_splitter.split_documents(loader.load())
|
||||
|
||||
pass
|
||||
|
||||
|
||||
@celery.task(name='ask_eveAI', queue='llm_interactions')
|
||||
def ask_eve_ai(query):
|
||||
# Interaction logic with LLMs like GPT (Langchain API calls, etc.)
|
||||
pass
|
||||
Reference in New Issue
Block a user