Merge pull request #2 from bdeboe/main

caretdev · May 8, 2024 · 742b488 · 742b488
2 parents a791f1b + 9ed4937
commit 742b488
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/langchain_iris/vectorstores.py b/langchain_iris/vectorstores.py
@@ -95,7 +95,15 @@ def __init__(
         self.logger = logger or logging.getLogger(__name__)
         self.override_relevance_score_fn = relevance_score_fn
         self.engine_args = engine_args or {}
-        self._conn = connection if connection else self.connect()
+        if connection:
+            self._conn = connection 
+            try:
+                if connection.dialect.supports_vectors:
+                    self.native_vector = True
+            except:  # noqa
+                pass
+        else: 
+            self._conn = self.connect()
 
         self.__post_init__()