chore: rename VectorizerField to Vectorizer

timescale · Dec 4, 2024 · 36cf4d9 · 36cf4d9
1 parent 60a3327
commit 36cf4d9
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 11 deletions.
diff --git a/projects/pgai/pgai/sqlalchemy/__init__.py b/projects/pgai/pgai/sqlalchemy/__init__.py
@@ -25,7 +25,7 @@ class EmbeddingModel(DeclarativeBase, Generic[T]):
     parent: T  # Type of the parent model
 
 
-class VectorizerField:
+class Vectorizer:
     def __init__(
         self,
         dimensions: int,

diff --git a/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy.py b/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy.py
@@ -5,7 +5,7 @@
 from testcontainers.postgres import PostgresContainer  # type: ignore
 
 from pgai.cli import vectorizer_worker
-from pgai.sqlalchemy import VectorizerField
+from pgai.sqlalchemy import Vectorizer
 
 
 def test_sqlalchemy(postgres_container: PostgresContainer, initialized_engine: Engine):
@@ -20,7 +20,7 @@ class BlogPost(Base):
         title = Column(Text, nullable=False)
         content = Column(Text, nullable=False)
 
-        content_embeddings = VectorizerField(
+        content_embeddings = Vectorizer(
             dimensions=768,
             add_relationship=True,
         )

diff --git a/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_composite_primary.py b/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_composite_primary.py
@@ -6,7 +6,7 @@
 from testcontainers.postgres import PostgresContainer  # type: ignore
 
 from pgai.cli import vectorizer_worker
-from pgai.sqlalchemy import EmbeddingModel, VectorizerField
+from pgai.sqlalchemy import EmbeddingModel, Vectorizer
 
 
 class Base(DeclarativeBase):
@@ -18,7 +18,7 @@ class Author(Base):
     first_name = Column(Text, primary_key=True)
     last_name = Column(Text, primary_key=True)
     bio = Column(Text, nullable=False)
-    bio_embeddings = VectorizerField(
+    bio_embeddings = Vectorizer(
         dimensions=768,
         add_relationship=True,
     )

diff --git a/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_large_embeddings.py b/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_large_embeddings.py
@@ -6,7 +6,7 @@
 from testcontainers.postgres import PostgresContainer  # type: ignore
 
 from pgai.cli import vectorizer_worker
-from pgai.sqlalchemy import VectorizerField
+from pgai.sqlalchemy import Vectorizer
 
 
 class Base(DeclarativeBase):
@@ -18,7 +18,7 @@ class BlogPost(Base):
     id = Column(Integer, primary_key=True)
     title = Column(Text, nullable=False)
     content = Column(Text, nullable=False)
-    content_embeddings = VectorizerField(dimensions=1536)
+    content_embeddings = Vectorizer(dimensions=1536)
 
 
 def run_vectorizer_worker(db_url: str, vectorizer_id: int) -> None:

diff --git a/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_no_relationship.py b/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_no_relationship.py
@@ -6,7 +6,7 @@
 from testcontainers.postgres import PostgresContainer  # type: ignore
 
 from pgai.cli import vectorizer_worker
-from pgai.sqlalchemy import VectorizerField
+from pgai.sqlalchemy import Vectorizer
 
 
 class Base(DeclarativeBase):
@@ -18,7 +18,7 @@ class BlogPost(Base):
     id = Column(Integer, primary_key=True)
     title = Column(Text, nullable=False)
     content = Column(Text, nullable=False)
-    content_embeddings = VectorizerField(
+    content_embeddings = Vectorizer(
         dimensions=768,
         add_relationship=False,
     )

diff --git a/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_relationship.py b/projects/pgai/tests/vectorizer/extensions/test_sqlalchemy_relationship.py
@@ -6,7 +6,7 @@
 from testcontainers.postgres import PostgresContainer  # type: ignore
 
 from pgai.cli import vectorizer_worker
-from pgai.sqlalchemy import EmbeddingModel, VectorizerField
+from pgai.sqlalchemy import EmbeddingModel, Vectorizer
 
 
 class Base(DeclarativeBase):
@@ -18,7 +18,7 @@ class BlogPost(Base):
     id = Column(Integer, primary_key=True)
     title = Column(Text, nullable=False)
     content = Column(Text, nullable=False)
-    content_embeddings = VectorizerField(
+    content_embeddings = Vectorizer(
         dimensions=768,
         add_relationship=True,
     )