update faiss optional import

llmware/embeddings.py

@@ -20,7 +20,6 @@
 
 
 import os
-import faiss
 import logging
 import numpy as np
 import re
@@ -30,6 +29,11 @@
 from pymilvus import connections, utility, FieldSchema, CollectionSchema, DataType, Collection
 from pymongo import MongoClient
 
+try:
+    import faiss
+except ImportError:
+    pass
+
 # note: update- adding psycopg and postgres to core llmware package in version 0.2.0
 try:
     from pgvector.psycopg import register_vector
@@ -636,9 +640,20 @@ def create_new_embedding(self, doc_ids=None, batch_size=100):
 
         if not self.index:
             if os.path.exists(self.embedding_file_path):
-                self.index = faiss.read_index(self.embedding_file_path)
-            else: 
-                self.index = faiss.IndexFlatL2(self.embedding_dims)
+
+                #   shifted faiss to optional dependency
+                #   note: there may be an edge case where this faiss command would fail even with
+                #   library installed, but we throw dependency not installed error as most likely cause
+
+                try:
+                    self.index = faiss.read_index(self.embedding_file_path)
+                except:
+                    raise DependencyNotInstalledException("faiss-cpu")
+            else:
+                try:
+                    self.index = faiss.IndexFlatL2(self.embedding_dims)
+                except:
+                    raise DependencyNotInstalledException("faiss-cpu")
 
         # get cursor for text collection with blocks requiring embedding
         all_blocks_cursor, num_of_blocks = self.utils.get_blocks_cursor(doc_ids=doc_ids)

llmware/exceptions.py

@@ -92,8 +92,7 @@ def __init__(self, prompt_name):
 class DependencyNotInstalledException(LLMWareException):
 
     def __init__(self, required_library_dependency):
-        message = f"'{required_library_dependency}' needs to be installed to use this function.  Please refer to the " \
-                  f"documentation with any questions. "
+        message = f"'{required_library_dependency}' needs to be installed to use this function."
         super().__init__(message)