yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-05 11:38:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix] MySQL connection 'check_same_thread' error (#1585)

Browse Source 
* [fix] sql mysql connection 'check_same_thread' error

* adjust sql connection if-block logic
This commit is contained in:
CandiceYu8 2021-10-15 16:29:36 +08:00 committed by GitHub
parent 451e51a224
commit 5cfdabda2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
haystack/document_store/sql.py
View File

@ -118,7 +118,10 @@ class SQLDocumentStore(BaseDocumentStore):
url=url, index=index, label_index=label_index, duplicate_documents=duplicate_documents, check_same_thread=check_same_thread
)
engine = create_engine(url,connect_args={'check_same_thread': check_same_thread})
if "sqlite" in url:
engine = create_engine(url, connect_args={'check_same_thread': check_same_thread})
else:
engine = create_engine(url)
ORMBase.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
self.session = Session()