GeezerVison is showing github.com — Add other input types for SQLAlchemy BIT model by jackrua · Pull Request #137 · pgvector/pgvector-python
Conversation
jackrua
commented
Jul 17, 2025
•
edited
Loading
Uh oh!
There was an error while loading. Please reload this page.
Hi,
First of all, thank you for the great library @ankane!
I'm opening this PR to address some issues that have been raised by users (#84, #110 and #112), that is to input types other than string in the BIT SQLAlchemy model (also when using the asyncpg dialect). In brief with this PR, given:
class Item(Base):
id = mapped_column(Integer, primary_key=True)
binary_embedding = mapped_column(BIT(3))
You can add it to your database using whatever input you like (and this will be transformed following the logic already present in the init of the Bit class):
with Session(engine) as session:
session.add(Item(id = 1,
binary_embedding = [True, False, True])) # or [1, 0, 1], or np.array([1, 0, 1])
Sorry, something went wrong.
Uh oh!
There was an error while loading. Please reload this page.
jackrua
changed the title
Add other input types for SQLAlchemy models
Add other input types for SQLAlchemy BIT model
Jul 17, 2025
|
Hi @jackrua, thanks for the PR. Added support for passing str objects in the commit above. I think the other changes are breaking to some degree, but may make sense for 0.5.0.
|
Sorry, something went wrong.
Uh oh!
There was an error while loading. Please reload this page.
jackrua
commented
Sep 22, 2025
•
edited
Loading
Uh oh!
There was an error while loading. Please reload this page.
|
Thank you @ankane for the feedback and sorry for the late response. I merged master in my branch and updated my PR, and now the CI passes too.
However, I was thinking that now when you read from the DB using the sqlalchemy model, we are returning str in some cases and BitString in others, wouldn't it be better to always return a Bit object (similar to what you do with HALFVEC for example), what do you think?
I wanted to implement the result_processor method to fix this in this PR but I wasn't sure what you thought about it, let me know.
|
Sorry, something went wrong.
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
Assignees
No one assigned
Milestone
No milestone
Development
Successfully merging this pull request may close these issues.
Uh oh!
There was an error while loading. Please reload this page.
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
|