From b6a919f0df1419fb2723f794a14348adcfc6d580 Mon Sep 17 00:00:00 2001 From: Disha Prakash Date: Tue, 13 May 2025 18:51:53 +0000 Subject: [PATCH] fix: Add support for a metadata column named id --- langchain_postgres/v2/async_vectorstore.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/langchain_postgres/v2/async_vectorstore.py b/langchain_postgres/v2/async_vectorstore.py index 11e5ff99..e8c7f5df 100644 --- a/langchain_postgres/v2/async_vectorstore.py +++ b/langchain_postgres/v2/async_vectorstore.py @@ -275,14 +275,14 @@ async def aadd_embeddings( ) insert_stmt = f'INSERT INTO "{self.schema_name}"."{self.table_name}"("{self.id_column}", "{self.content_column}", "{self.embedding_column}"{metadata_col_names}' values = { - "id": id, + "langchain_id": id, "content": content, "embedding": str([float(dimension) for dimension in embedding]), } - values_stmt = "VALUES (:id, :content, :embedding" + values_stmt = "VALUES (:langchain_id, :content, :embedding" if not embedding and can_inline_embed: - values_stmt = f"VALUES (:id, :content, {self.embedding_service.embed_query_inline(content)}" # type: ignore + values_stmt = f"VALUES (:langchain_id, :content, {self.embedding_service.embed_query_inline(content)}" # type: ignore # Add metadata extra = copy.deepcopy(metadata)