ishaq101 commited on
Commit
079dea0
·
1 Parent(s): 8dcc782

Fix schema table langchain_pg_embedding

Browse files
Files changed (1) hide show
  1. src/db/postgres/vector_store.py +6 -0
src/db/postgres/vector_store.py CHANGED
@@ -1,10 +1,16 @@
1
  """PGVector store setup for document embeddings."""
2
 
3
  from langchain_postgres import PGVector
 
4
  from langchain_openai import AzureOpenAIEmbeddings
 
5
  from src.config.settings import settings
6
  from src.db.postgres.connection import _pgvector_engine
7
 
 
 
 
 
8
  # Initialize embeddings
9
  embeddings = AzureOpenAIEmbeddings(
10
  azure_deployment=settings.azureai_deployment_name_embedding,
 
1
  """PGVector store setup for document embeddings."""
2
 
3
  from langchain_postgres import PGVector
4
+ from langchain_postgres.vectorstores import EmbeddingStore
5
  from langchain_openai import AzureOpenAIEmbeddings
6
+ from sqlalchemy.orm import column_property
7
  from src.config.settings import settings
8
  from src.db.postgres.connection import _pgvector_engine
9
 
10
+ # Golang-managed schema uses 'uuid' as the PK column; langchain_postgres 0.0.17 expects 'id'.
11
+ # Remap the ORM attribute so generated SQL targets the correct column.
12
+ EmbeddingStore.id = column_property(EmbeddingStore.__table__.c["uuid"])
13
+
14
  # Initialize embeddings
15
  embeddings = AzureOpenAIEmbeddings(
16
  azure_deployment=settings.azureai_deployment_name_embedding,