pgmold 0.33.5

PostgreSQL schema-as-code management tool
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

mod common;
use common::*;

#[tokio::test]
async fn introspect_vector_type() {
    let (_container, url) = setup_postgres().await;
    let connection = PgConnection::new(&url).await.unwrap();

    // Simulate pgvector extension behavior without requiring the actual extension
    // pgvector stores dimension directly in atttypmod (no offset)
    // See: https://github.com/pgvector/pgvector/blob/master/src/vector.c
    sqlx::query("CREATE TYPE vector AS (placeholder int)")
        .execute(connection.pool())
        .await
        .unwrap();

    sqlx::query(
        r#"
        CREATE TABLE embeddings (
            id BIGINT PRIMARY KEY,
            embedding vector
        )
        "#,
    )
    .execute(connection.pool())
    .await
    .unwrap();

    // pgvector's vector_typmod_in returns dimension directly (atttypmod = dimension)
    sqlx::query(
        r#"
        UPDATE pg_attribute
        SET atttypmod = 1536
        WHERE attrelid = 'embeddings'::regclass
        AND attname = 'embedding'
        "#,
    )
    .execute(connection.pool())
    .await
    .unwrap();

    let schema = introspect_schema(&connection, &["public".to_string()], false)
        .await
        .unwrap();

    let table = schema
        .tables
        .get("public.embeddings")
        .expect("embeddings table should exist");

    let embedding_col = table
        .columns
        .get("embedding")
        .expect("embedding column should exist");

    match &embedding_col.data_type {
        pgmold::model::PgType::Vector(dim) => {
            assert_eq!(*dim, Some(1536), "Vector dimension should be 1536");
        }
        other => panic!("Expected Vector type, got {other:?}"),
    }
}

#[tokio::test]
async fn introspect_vector_type_unconstrained() {
    let (_container, url) = setup_postgres().await;
    let connection = PgConnection::new(&url).await.unwrap();

    // Test unconstrained vector type (no dimension specified)
    sqlx::query("CREATE TYPE vector AS (placeholder int)")
        .execute(connection.pool())
        .await
        .unwrap();

    sqlx::query(
        r#"
        CREATE TABLE embeddings (
            id BIGINT PRIMARY KEY,
            embedding vector
        )
        "#,
    )
    .execute(connection.pool())
    .await
    .unwrap();

    // Default atttypmod is -1 for unconstrained types
    // No need to update atttypmod, it should already be -1

    let schema = introspect_schema(&connection, &["public".to_string()], false)
        .await
        .unwrap();

    let table = schema
        .tables
        .get("public.embeddings")
        .expect("embeddings table should exist");

    let embedding_col = table
        .columns
        .get("embedding")
        .expect("embedding column should exist");

    match &embedding_col.data_type {
        pgmold::model::PgType::Vector(dim) => {
            assert_eq!(*dim, None, "Unconstrained vector should have no dimension");
        }
        other => panic!("Expected Vector type, got {other:?}"),
    }
}