Update langchain_postgres/vectorstores.py

Co-authored-by: Martín Gotelli Ferenaz <[email protected]>

Author: 0xrushi

langchain_postgres/vectorstores.py

@@ -222,25 +222,16 @@ class EmbeddingStore(Base):
         document = sqlalchemy.Column(sqlalchemy.String, nullable=True)
         cmetadata = sqlalchemy.Column(JSONB, nullable=True)
 
-        if extend_existing:
-            __table_args__ = (
-                sqlalchemy.Index(
-                    "ix_cmetadata_gin",
-                    "cmetadata",
-                    postgresql_using="gin",
-                    postgresql_ops={"cmetadata": "jsonb_path_ops"},
-                ),
-                {'extend_existing': True}
-            )
-        else:
-            __table_args__ = (
+        __table_args__ = (
                 sqlalchemy.Index(
                     "ix_cmetadata_gin",
                     "cmetadata",
                     postgresql_using="gin",
                     postgresql_ops={"cmetadata": "jsonb_path_ops"},
                 ),
             )
+        if extend_existing:
+            __table_args__ = __table_args__ + ({'extend_existing': True}, )
 
     _classes = (EmbeddingStore, CollectionStore)