fix breaking imports

Author: dishaprakash

docs/document_loader.ipynb

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

docs/vector_store.ipynb

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

samples/index_tuning_sample/create_vector_embeddings.py

@@ -19,9 +19,8 @@
 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
+from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore, Column
 
 EMBEDDING_COUNT = 100000
 VECTOR_SIZE = 768

samples/index_tuning_sample/index_search.py

@@ -32,7 +32,7 @@
     vector_table_name,
 )
 from langchain_google_vertexai import VertexAIEmbeddings
-from langchain_postgres.v2.indexes import (
+from langchain_google_alloydb_pg.indexes import (
     HNSWIndex,
     HNSWQueryOptions,
     IVFFlatIndex,

samples/langchain_quick_start.ipynb

@@ -601,8 +601,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBLoader\n",
-    "from langchain_postgres import Column\n",
+    "from langchain_google_alloydb_pg import AlloyDBEngine, AlloyDBLoader, Column\n",
     "\n",
     "engine = AlloyDBEngine.from_instance(\n",
     "    project_id=project_id,\n",
@@ -709,8 +708,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_google_alloydb_pg import AlloyDBEngine\n",
-    "from langchain_postgres import Column\n",
+    "from langchain_google_alloydb_pg import AlloyDBEngine, 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_postgres import Column
+    from langchain_google_alloydb_pg import Column
 
     await alloydb_engine.ainit_vectorstore_table(
         table_name=alloydb_table,

src/langchain_google_alloydb_pg/__init__.py

@@ -15,11 +15,12 @@
 from .chat_message_history import AlloyDBChatMessageHistory
 from .checkpoint import AlloyDBSaver
 from .embeddings import AlloyDBEmbeddings
-from .engine import AlloyDBEngine, Column
+from .engine import AlloyDBEngine
 from .loader import AlloyDBDocumentSaver, AlloyDBLoader
 from .model_manager import AlloyDBModel, AlloyDBModelManager
 from .vectorstore import AlloyDBVectorStore
 from .version import __version__
+from langchain_postgres import Column
 
 __all__ = [
     "AlloyDBEngine",

src/langchain_google_alloydb_pg/async_vectorstore.py

@@ -28,7 +28,7 @@
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VectorStore, utils
-from langchain_postgres.v2.indexes import (
+from langchain_google_alloydb_pg.indexes import (
     DEFAULT_DISTANCE_STRATEGY,
     DEFAULT_INDEX_NAME_SUFFIX,
     BaseIndex,

src/langchain_google_alloydb_pg/indexes.py

@@ -15,7 +15,19 @@
 import warnings
 from dataclasses import dataclass, field
 
-from langchain_postgres.v2.indexes import BaseIndex, DistanceStrategy, QueryOptions
+from langchain_postgres.v2.indexes import (
+  BaseIndex,
+  DistanceStrategy,
+  QueryOptions,
+  StrategyMixin,
+  DEFAULT_DISTANCE_STRATEGY,
+  DEFAULT_INDEX_NAME_SUFFIX,
+  ExactNearestNeighbor,
+  HNSWIndex,
+  HNSWQueryOptions,
+  IVFFlatIndex,
+  IVFFlatQueryOptions,
+)
 
 
 @dataclass

src/langchain_google_alloydb_pg/vectorstore.py

@@ -20,7 +20,7 @@
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VectorStore
-from langchain_postgres.v2.indexes import (
+from langchain_google_alloydb_pg.indexes import (
     DEFAULT_DISTANCE_STRATEGY,
     BaseIndex,
     DistanceStrategy,