re-expose hybrid search configs from init

Author: dishaprakash

src/langchain_google_alloydb_pg/__init__.py

@@ -13,6 +13,11 @@
 # limitations under the License.
 
 from langchain_postgres import Column
+from langchain_postgres.v2.hybrid_search_config import (
+    HybridSearchConfig,
+    reciprocal_rank_fusion,
+    weighted_sum_ranking,
+)
 
 from .chat_message_history import AlloyDBChatMessageHistory
 from .checkpoint import AlloyDBSaver
@@ -34,5 +39,8 @@
     "AlloyDBModelManager",
     "AlloyDBModel",
     "AlloyDBSaver",
+    "HybridSearchConfig",
+    "reciprocal_rank_fusion",
+    "weighted_sum_ranking",
     "__version__",
 ]

src/langchain_google_alloydb_pg/hybrid_search_config.py

@@ -21,7 +21,7 @@
 from langchain_core.embeddings import Embeddings
 from langchain_postgres import PGVectorStore
 
-from langchain_google_alloydb_pg.hybrid_search_config import HybridSearchConfig
+from langchain_google_alloydb_pg import HybridSearchConfig
 from langchain_google_alloydb_pg.indexes import (
     DEFAULT_DISTANCE_STRATEGY,
     DistanceStrategy,

src/langchain_google_alloydb_pg/vectorstore.py

@@ -23,9 +23,8 @@
 from langchain_core.embeddings import DeterministicFakeEmbedding
 from sqlalchemy import text
 
-from langchain_google_alloydb_pg import AlloyDBEngine
+from langchain_google_alloydb_pg import AlloyDBEngine, HybridSearchConfig
 from langchain_google_alloydb_pg.async_vectorstore import AsyncAlloyDBVectorStore
-from langchain_google_alloydb_pg.hybrid_search_config import HybridSearchConfig
 from langchain_google_alloydb_pg.indexes import (
     DEFAULT_INDEX_NAME_SUFFIX,
     DistanceStrategy,

tests/test_async_vectorstore_index.py

@@ -23,13 +23,14 @@
 from PIL import Image
 from sqlalchemy import text
 
-from langchain_google_alloydb_pg import AlloyDBEngine, Column
-from langchain_google_alloydb_pg.async_vectorstore import AsyncAlloyDBVectorStore
-from langchain_google_alloydb_pg.hybrid_search_config import (
+from langchain_google_alloydb_pg import (
+    AlloyDBEngine,
+    Column,
     HybridSearchConfig,
     reciprocal_rank_fusion,
     weighted_sum_ranking,
 )
+from langchain_google_alloydb_pg.async_vectorstore import AsyncAlloyDBVectorStore
 from langchain_google_alloydb_pg.indexes import (
     DistanceStrategy,
     HNSWQueryOptions,