Browse Source

[BugFix] Skip checking thread when making sqlite connection (#978)

Deshraj Yadav 1 year ago
parent
commit
141a23fb1e
5 changed files with 6 additions and 5 deletions
  1. 2 1
      embedchain/apps/app.py
  2. 2 1
      embedchain/embedchain.py
  3. 1 1
      embedchain/memory/base.py
  4. 0 1
      embedchain/pipeline.py
  5. 1 1
      pyproject.toml

+ 2 - 1
embedchain/apps/app.py

@@ -2,7 +2,8 @@ from typing import Optional
 
 import yaml
 
-from embedchain.config import AppConfig, BaseEmbedderConfig, BaseLlmConfig, ChunkerConfig
+from embedchain.config import (AppConfig, BaseEmbedderConfig, BaseLlmConfig,
+                               ChunkerConfig)
 from embedchain.config.vectordb.base import BaseVectorDbConfig
 from embedchain.embedchain import EmbedChain
 from embedchain.embedder.base import BaseEmbedder

+ 2 - 1
embedchain/embedchain.py

@@ -16,7 +16,8 @@ from embedchain.embedder.base import BaseEmbedder
 from embedchain.helpers.json_serializable import JSONSerializable
 from embedchain.llm.base import BaseLlm
 from embedchain.loaders.base_loader import BaseLoader
-from embedchain.models.data_type import DataType, DirectDataType, IndirectDataType, SpecialDataType
+from embedchain.models.data_type import (DataType, DirectDataType,
+                                         IndirectDataType, SpecialDataType)
 from embedchain.telemetry.posthog import AnonymousTelemetry
 from embedchain.utils import detect_datatype, is_valid_json_string
 from embedchain.vectordb.base import BaseVectorDB

+ 1 - 1
embedchain/memory/base.py

@@ -23,7 +23,7 @@ CHAT_MESSAGE_CREATE_TABLE_QUERY = """
 
 class ECChatMemory:
     def __init__(self) -> None:
-        with sqlite3.connect(SQLITE_PATH) as self.connection:
+        with sqlite3.connect(SQLITE_PATH, check_same_thread=False) as self.connection:
             self.cursor = self.connection.cursor()
 
             self.cursor.execute(CHAT_MESSAGE_CREATE_TABLE_QUERY)

+ 0 - 1
embedchain/pipeline.py

@@ -23,7 +23,6 @@ from embedchain.utils import validate_yaml_config
 from embedchain.vectordb.base import BaseVectorDB
 from embedchain.vectordb.chroma import ChromaDB
 
-
 # Setup the user directory if doesn't exist already
 Client.setup_dir()
 

+ 1 - 1
pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "embedchain"
-version = "0.1.21"
+version = "0.1.22"
 description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data"
 authors = [
     "Taranjeet Singh <taranjeet@embedchain.ai>",