Data path handling

artitw · Oct 16, 2024 · 192638d · 192638d
1 parent 1d1c5c0
commit 192638d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.7.9",
+  version="1.8.0",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",

diff --git a/text2text/rag_assistant.py b/text2text/rag_assistant.py
@@ -1,5 +1,6 @@
 import text2text as t2t
 
+import os
 import pickle
 import sqlite3
 import requests
@@ -89,7 +90,7 @@ def __init__(self, **kwargs):
 
     self.records["embedding"] = self.records["embedding"].apply(lambda x: pickle.dumps(x))
 
-    if sqlite_path:
+    if sqlite_path and os.path.exists(sqlite_path):
       conn = sqlite3.connect(sqlite_path)
       fields = ", ".join(db_fields)
       query = f"SELECT {fields} FROM {RAG_TABLE_NAME}"