Skip to content

Commit a90e75a

Browse files
authored
fix: errors table is now based on Features when running a Vectorizer (#757)
1 parent 49be5ca commit a90e75a

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

projects/pgai/pgai/vectorizer/vectorizer.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,13 @@ async def run(
163163
Returns:
164164
Number of items processed
165165
"""
166+
167+
if (
168+
self.errors_table == DEFAULT_VECTORIZER_ERRORS_TABLE
169+
and not features.has_vectorizer_errors_view
170+
):
171+
self.errors_table = "vectorizer_errors"
172+
166173
concurrency = concurrency or self.config.processing.concurrency
167174
tasks = [
168175
asyncio.create_task(

projects/pgai/pgai/vectorizer/worker.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from .. import __version__
1616
from .embeddings import ApiKeyMixin
1717
from .features import Features
18-
from .vectorizer import DEFAULT_VECTORIZER_ERRORS_TABLE, Vectorizer
18+
from .vectorizer import Vectorizer
1919
from .worker_tracking import WorkerTracking
2020

2121
if sys.version_info >= (3, 11):
@@ -125,12 +125,6 @@ def _get_vectorizer(self, vectorizer_id: int, features: Features) -> Vectorizer:
125125
embedding = vectorizer["config"]["embedding"]
126126
vectorizer = Vectorizer.model_validate(vectorizer)
127127

128-
if (
129-
vectorizer.errors_table == DEFAULT_VECTORIZER_ERRORS_TABLE
130-
and not features.has_vectorizer_errors_view
131-
):
132-
vectorizer.errors_table = "vectorizer_errors"
133-
134128
# The Ollama API doesn't need a key, so `api_key_name` may be unset
135129
if "api_key_name" in embedding:
136130
api_key_name = embedding["api_key_name"]

0 commit comments

Comments
 (0)