diff --git a/libs/langchain/langchain/vectorstores/sqlitevss.py b/libs/langchain/langchain/vectorstores/sqlitevss.py index d311ffafcb9..93b1d676494 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)