back to default values

backend/semantic_search.py

@@ -6,17 +6,19 @@ from sentence_transformers import SentenceTransformer
 
 db = lancedb.connect(".lancedb")
 
-#TABLE = db.open_table(os.getenv("TABLE_NAME"))
+TABLE = db.open_table(os.getenv("TABLE_NAME"))
 VECTOR_COLUMN = os.getenv("VECTOR_COLUMN", "vector")
 TEXT_COLUMN = os.getenv("TEXT_COLUMN", "text")
 BATCH_SIZE = int(os.getenv("BATCH_SIZE", 32))
 
-#retriever = SentenceTransformer(os.getenv("EMB_MODEL"))
+retriever = SentenceTransformer(os.getenv("EMB_MODEL"))
 
 
 def retrieve(query, k, table_name, emb_name):
-    TABLE = db.open_table(table_name)
-    retriever = SentenceTransformer(emb_name)
+    print(table_name)
+    print(emb_name)
+    #TABLE = db.open_table(table_name)
+    #retriever = SentenceTransformer(emb_name)
     query_vec = retriever.encode(query)
     try:
         documents = TABLE.search(query_vec, vector_column_name=VECTOR_COLUMN).limit(k).to_list()