From 21fa2bc2fb762d129be0fedf0f913db025b89bce Mon Sep 17 00:00:00 2001 From: Krisztina Zsihovszki Date: Sat, 3 Aug 2024 15:09:00 +0200 Subject: [PATCH 1/2] NIFI-13628 Use NumPy 1.26.4 for Chroma processors --- src/extensions/vectorstores/PutChroma.py | 1 + src/extensions/vectorstores/QueryChroma.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/extensions/vectorstores/PutChroma.py b/src/extensions/vectorstores/PutChroma.py index 5554a82..247850e 100644 --- a/src/extensions/vectorstores/PutChroma.py +++ b/src/extensions/vectorstores/PutChroma.py @@ -29,6 +29,7 @@ class ProcessorDetails: "text", "LLM", ] + dependencies = ["numpy==1.26.4"] STORE_TEXT = PropertyDescriptor( name="Store Document Text", diff --git a/src/extensions/vectorstores/QueryChroma.py b/src/extensions/vectorstores/QueryChroma.py index 3d28ccf..7205bc9 100644 --- a/src/extensions/vectorstores/QueryChroma.py +++ b/src/extensions/vectorstores/QueryChroma.py @@ -30,6 +30,7 @@ class ProcessorDetails: "text", "LLM", ] + dependencies = ["numpy==1.26.4"] QUERY = PropertyDescriptor( name="Query", From abb13343c82db1cc4d86c2c758334c35408a6dab Mon Sep 17 00:00:00 2001 From: Krisztina Zsihovszki Date: Sun, 4 Aug 2024 15:38:42 +0200 Subject: [PATCH 2/2] NIFI-13628 Review comments --- src/extensions/vectorstores/PutChroma.py | 1 - src/extensions/vectorstores/QueryChroma.py | 1 - src/extensions/vectorstores/requirements.txt | 1 + 3 files changed, 1 insertion(+), 2 deletions(-) diff --git a/src/extensions/vectorstores/PutChroma.py b/src/extensions/vectorstores/PutChroma.py index 247850e..5554a82 100644 --- a/src/extensions/vectorstores/PutChroma.py +++ b/src/extensions/vectorstores/PutChroma.py @@ -29,7 +29,6 @@ class ProcessorDetails: "text", "LLM", ] - dependencies = ["numpy==1.26.4"] STORE_TEXT = PropertyDescriptor( name="Store Document Text", diff --git a/src/extensions/vectorstores/QueryChroma.py b/src/extensions/vectorstores/QueryChroma.py index 7205bc9..3d28ccf 100644 --- a/src/extensions/vectorstores/QueryChroma.py +++ b/src/extensions/vectorstores/QueryChroma.py @@ -30,7 +30,6 @@ class ProcessorDetails: "text", "LLM", ] - dependencies = ["numpy==1.26.4"] QUERY = PropertyDescriptor( name="Query", diff --git a/src/extensions/vectorstores/requirements.txt b/src/extensions/vectorstores/requirements.txt index 31e215a..c2131d4 100644 --- a/src/extensions/vectorstores/requirements.txt +++ b/src/extensions/vectorstores/requirements.txt @@ -7,6 +7,7 @@ langchain==0.1.11 # Chroma requirements chromadb==0.4.22 +numpy==1.26.4 onnxruntime tokenizers tqdm