diff --git a/backend/glowpath-backend.spec b/backend/glowpath-backend.spec index 067432d..a38cd5e 100644 --- a/backend/glowpath-backend.spec +++ b/backend/glowpath-backend.spec @@ -32,6 +32,14 @@ additional_packages = [ 'websockets', 'sse_starlette', 'chromadb', + 'sentence_transformers', + 'transformers', + 'torch', + 'numpy', + 'sklearn', + 'scipy', + 'tokenizers', + 'huggingface_hub', ] for package in additional_packages: @@ -66,6 +74,27 @@ hiddenimports += [ 'chromadb.db', 'chromadb.db.impl', 'onnxruntime', + # Sentence transformers specific imports + 'sentence_transformers.models.Transformer', + 'sentence_transformers.models.Pooling', + 'sentence_transformers.models.Normalize', + 'sentence_transformers.cross_encoder', + 'sentence_transformers.util', + 'sentence_transformers.evaluation', + 'transformers.models.auto.modeling_auto', + 'transformers.models.auto.tokenization_auto', + 'transformers.models.auto.configuration_auto', + 'transformers.models.bert', + 'transformers.models.distilbert', + 'transformers.models.roberta', + 'transformers.models.mpnet', + 'torch._C', + 'torch.nn.functional', + 'torch.utils.data', + 'sklearn.metrics.pairwise', + 'sklearn.feature_extraction.text', + 'numpy.core._methods', + 'numpy.lib.format', ] # Include SSL certificates diff --git a/backend/hooks/hook-open_webui.py b/backend/hooks/hook-open_webui.py index efa64ef..cd58f5d 100644 --- a/backend/hooks/hook-open_webui.py +++ b/backend/hooks/hook-open_webui.py @@ -30,4 +30,25 @@ hiddenimports += [ "uvicorn.protocols.websockets.auto", "uvicorn.lifespan.on", "uvicorn.loops.auto", + # Sentence transformers and embedding related imports + "sentence_transformers", + "sentence_transformers.models", + "sentence_transformers.models.Transformer", + "sentence_transformers.models.Pooling", + "sentence_transformers.models.Normalize", + "sentence_transformers.util", + "sentence_transformers.cross_encoder", + "transformers", + "transformers.models", + "transformers.models.auto", + "transformers.models.bert", + "transformers.tokenization_utils", + "torch", + "torch.nn", + "torch.optim", + "tokenizers", + "tokenizers.implementations", + "huggingface_hub", + "huggingface_hub.file_download", + "sklearn.metrics.pairwise", ]