Run embedding (Insert) SQL stuck forever when using pgrag

I am testing the pgrag capability on neon following this https://neon.tech/docs/extensions/pgrag#end-to-end-rag-example official guide. I give the db 2 CU (8G RAM) that should be surfficent enough for RAG with only a few documents.

The image is the table schema using vector index. If I directly insert array of number to this table, it works without any issue.

But if I use

rag_bge_small_en_v15.embedding_for_passage

rag_bge_small_en_v15.embedding_for_passage

rag_bge_small_en_v15.embedding_for_passage

rag_bge_small_en_v15.embedding_for_passage function, then the db connection stuck forever (in pycharm). What could went wrong? I already tried asking NEON AI.

insert into documents (
    id, 
    embedding
)
values (
    'doc-1',
    rag_bge_small_en_v15.embedding_for_passage('hello world')
);

insert into documents (
    id, 
    embedding
)
values (
    'doc-1',
    rag_bge_small_en_v15.embedding_for_passage('hello world')
);

insert into documents (
    id, 
    embedding
)
values (
    'doc-1',
    rag_bge_small_en_v15.embedding_for_passage('hello world')
);

insert into documents (
    id, 
    embedding
)
values (
    'doc-1',
    rag_bge_small_en_v15.embedding_for_passage('hello world')
);

Neon

The pgrag extension - Neon Docs

What is RAG? What's included in a RAG pipeline? pgrag functions How to use pgrag The pgvector extension YouTube pgrag video demonstration pgrag GitHub repository The pgrag extension and its accompanyi...

Run embedding (Insert) SQL stuck forever when using pgrag

Similar Threads

Similar Threads

Similar Threads