fix : correct type checking against db column for embedding_column variable (#243)

NanoClem · c.decoopman · web-flow · commit 6148b100fe00 · 2025-08-11T11:32:13.000-07:00
See #241 for reference. --------- Co-authored-by: c.decoopman <c.decoopman@axeal.com>
diff --git a/langchain_postgres/v2/async_vectorstore.py b/langchain_postgres/v2/async_vectorstore.py
@@ -210,7 +210,7 @@ async def create(
                 hybrid_search_config.tsv_column = ""
         if embedding_column not in columns:
             raise ValueError(f"Embedding column, {embedding_column}, does not exist.")
-        if columns[embedding_column] != "USER-DEFINED":
+        if columns[embedding_column] not in ["USER-DEFINED", "vector"]:
             raise ValueError(
                 f"Embedding column, {embedding_column}, is not type Vector."
             )

Original file line number	Diff line number	Diff line change
`@@ -210,7 +210,7 @@ async def create(`
`210`	`210`	`hybrid_search_config.tsv_column = ""`
`211`	`211`	`if embedding_column not in columns:`
`212`	`212`	`raise ValueError(f"Embedding column, {embedding_column}, does not exist.")`
`213`		`- if columns[embedding_column] != "USER-DEFINED":`
	`213`	`+ if columns[embedding_column] not in ["USER-DEFINED", "vector"]:`
`214`	`214`	`raise ValueError(`
`215`	`215`	`f"Embedding column, {embedding_column}, is not type Vector."`
`216`	`216`	`)`