From 9e196cb4709ab44a82e9e16170f6b322d9a06567 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Fri, 1 Sep 2023 17:14:06 -0700 Subject: [PATCH] rm sqlite3 import (#10115) --- libs/langchain/langchain/vectorstores/sqlitevss.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/vectorstores/sqlitevss.py b/libs/langchain/langchain/vectorstores/sqlitevss.py index d311ffafcb..93b1d67649 100644 --- a/libs/langchain/langchain/vectorstores/sqlitevss.py +++ b/libs/langchain/langchain/vectorstores/sqlitevss.py @@ -2,9 +2,9 @@ from __future__ import annotations import json import logging -import sqlite3 import warnings from typing import ( + TYPE_CHECKING, Any, Iterable, List, @@ -17,6 +17,9 @@ from langchain.docstore.document import Document from langchain.embeddings.base import Embeddings from langchain.vectorstores.base import VectorStore +if TYPE_CHECKING: + import sqlite3 + logger = logging.getLogger(__name__) @@ -203,6 +206,8 @@ class SQLiteVSS(VectorStore): @staticmethod def create_connection(db_file: str) -> sqlite3.Connection: + import sqlite3 + import sqlite_vss connection = sqlite3.connect(db_file)