Adapt tests

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2025-01-17 21:40:30 +01:00
parent f536038f65
commit 9ec5a915fe
3 changed files with 3 additions and 4 deletions

View file

@ -100,8 +100,7 @@ jobs:
# The python3-grpc-tools package in 22.04 is too old # The python3-grpc-tools package in 22.04 is too old
pip install --user grpcio-tools pip install --user grpcio-tools
sudo rm -rfv /usr/bin/conda || true make -C backend/python/transformers
PATH=$PATH:/opt/conda/bin make -C backend/python/sentencetransformers
# Pre-build piper before we start tests in order to have shared libraries in place # Pre-build piper before we start tests in order to have shared libraries in place
make sources/go-piper && \ make sources/go-piper && \

View file

@ -822,7 +822,7 @@ var _ = Describe("API test", func() {
application, err := application.New( application, err := application.New(
append(commonOpts, append(commonOpts,
config.WithExternalBackend("huggingface", os.Getenv("HUGGINGFACE_GRPC")), config.WithExternalBackend("transformers", os.Getenv("HUGGINGFACE_GRPC")),
config.WithContext(c), config.WithContext(c),
config.WithModelPath(modelPath), config.WithModelPath(modelPath),
)...) )...)

View file

@ -1,5 +1,5 @@
name: code-search-ada-code-001 name: code-search-ada-code-001
backend: huggingface backend: sentencetransformers
embeddings: true embeddings: true
parameters: parameters:
model: all-MiniLM-L6-v2 model: all-MiniLM-L6-v2