- Add Catalog Functionality

eveai_workers/tasks.py

@@ -13,7 +13,7 @@ from langchain_core.runnables import RunnablePassthrough
 from sqlalchemy.exc import SQLAlchemyError
 
 from common.extensions import db, minio_client
-from common.models.document import DocumentVersion, Embedding
+from common.models.document import DocumentVersion, Embedding, Document
 from common.models.user import Tenant
 from common.utils.celery_utils import current_celery
 from common.utils.database import Database
@@ -50,8 +50,12 @@ def create_embeddings(tenant_id, document_version_id):
         if document_version is None:
             raise Exception(f'Document version {document_version_id} not found')
 
+        # Retrieve the Catalog ID
+        doc = Document.query.get_or_404(document_version.doc_id)
+        catalog_id = doc.catalog_id
+
         # Select variables to work with depending on tenant and model
-        model_variables = select_model_variables(tenant)
+        model_variables = select_model_variables(tenant, catalog_id=catalog_id)
         current_app.logger.debug(f'Model variables: {model_variables}')
 
     except Exception as e: