ishaq101 commited on
Commit
0e9263a
·
1 Parent(s): 079dea0

fix build error in importing packages langchain_postgres

Browse files
Files changed (1) hide show
  1. src/db/postgres/vector_store.py +0 -6
src/db/postgres/vector_store.py CHANGED
@@ -1,16 +1,10 @@
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,
 
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,