How to go about supporting pg_vector postgres extension - supporting the vector type

[alexpusch]

Hi, pg_vector postgres extension offers sqlx support.

When I try to create a column of the pgvector::Vector type the following error occurs

#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "example")]
pub struct Model {
    #[sea_orm(primary_key)]
    pub id: i32,
    // does not help - #[sea_orm(column_type = "custom(\"vector\")")]
    pub vector: pgvector::Vector,
}

the trait bound `sea_orm::Value: From<Vector>` is not satisfied

Is it possible to support types with sqlX support?
Is it possible to add support for external types, or are we limited to the existing variants in the Value enum?

[billy1624]

Hey @alexpusch, is it possible to define it as pub vector: Vec<i32>?

[alexpusch]

@billy1624 trying to just use it with Vec<f32> naively fails when using it as binded parameter. The interesting question is - if SeaOrm uses sqlx under the hood. Is it possible to use custom types with sqlx support out of the box?

[billy1624]

We will need to update the internal of SeaQuery and SeaORM

[holmofy]

How is this issue now?

[alexpusch]

@billy1624 I've noticed that some support was added sea-query in SeaQL/sea-query#774
but it seems that sea orm migration still generates the entities as a string with custom type attribute. I can't seem to find a supported Vector type that I manually could replace the String with.

Are there plans to improve vector column support?

#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "example")]
pub struct Model {
    #[sea_orm(primary_key)]
    pub id: i32,
    #[sea_orm(column_type = "custom(\"vector\")")]
    pub vector: String,
}

[cosmincatalin]

I have written a short article on how to use pgVector with SeaORM. Maybe this is useful to people that end up here 🙂

https://cosminsanda.com/posts/using-pgvector-with-seaorm-in-rust/

[tyt2y3]

thank you so much for the article! very well written. I will work on integrating pgvector natively into SeaORM.
we've already added support in SeaQuery SeaQL/sea-query#774

[tyt2y3]

Please find this #2500

[tyt2y3]

Released! https://github.com/SeaQL/sea-orm/releases/tag/1.1.6

https://github.com/SeaQL/sea-orm/blob/master/tests/embedding_tests.rs

[cosmincatalin]

Would be nice to demonstrate some pgVector specific operations as well.

[tyt2y3]

yeah sad no examples, but there're a few operators https://github.com/SeaQL/sea-query/blob/master/src/extension/postgres/mod.rs#L18