change imports

Author: dishaprakash

docs/document_loader.ipynb

@@ -415,7 +415,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain_google_alloydb_pg import Column\n",
+    "from langchain_postgres import Column\n",
     "\n",
     "await engine.ainit_document_table(\n",
     "    table_name=TABLE_NAME,\n",

docs/vector_store.ipynb

@@ -606,7 +606,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain_google_alloydb_pg import Column\n",
+    "from langchain_postgres import Column\n",
+    "\n",
     "\n",
     "# Set table name\n",
     "TABLE_NAME = \"vectorstore_custom\"\n",

samples/index_tuning_sample/create_vector_embeddings.py

@@ -19,8 +19,9 @@
 import sqlalchemy
 from langchain_community.document_loaders import CSVLoader
 from langchain_google_vertexai import VertexAIEmbeddings
+from langchain_postgres import Column
 
-from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore, Column
+from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore
 
 EMBEDDING_COUNT = 100000
 VECTOR_SIZE = 768

samples/index_tuning_sample/index_search.py

@@ -32,12 +32,14 @@
     vector_table_name,
 )
 from langchain_google_vertexai import VertexAIEmbeddings
-
-from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore
-from langchain_google_alloydb_pg.indexes import (
+from langchain_postgres.v2.indexes import (
     HNSWIndex,
     HNSWQueryOptions,
     IVFFlatIndex,
+)
+
+from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore
+from langchain_google_alloydb_pg.indexes import (
     IVFIndex,
     ScaNNIndex,
 )

samples/langchain_quick_start.ipynb

@@ -601,7 +601,8 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_google_alloydb_pg import AlloyDBEngine, Column, AlloyDBLoader\n",
+    "from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBLoader\n",
+    "from langchain_postgres.v2 import Column\n",
     "\n",
     "engine = AlloyDBEngine.from_instance(\n",
     "    project_id=project_id,\n",
@@ -708,7 +709,8 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_google_alloydb_pg import AlloyDBEngine, Column\n",
+    "from langchain_google_alloydb_pg import AlloyDBEngine\n",
+    "from langchain_postgres import Column\n",
     "\n",
     "sample_vector_table_name = \"movie_vector_table_samples\"\n",
     "\n",

samples/migrations/migrate_pinecone_vectorstore_to_alloydb.py

@@ -175,7 +175,7 @@ async def main(
     print("Langchain AlloyDB client initiated.")
 
     # [START pinecone_vectorstore_alloydb_migration_create_table]
-    from langchain_google_alloydb_pg import Column
+    from langchain_postgres import Column
 
     await alloydb_engine.ainit_vectorstore_table(
         table_name=alloydb_table,