fix tests

Author: yrobla

src/codegate/storage/storage_engine.py

@@ -1,3 +1,4 @@
+import time
 from codegate.codegate_logging import setup_logging
 from codegate.inference.inference_engine import LlamaCppInferenceEngine
 from weaviate.classes.config import DataType
@@ -45,24 +46,22 @@ def __init__(self, data_path='./weaviate_data'):
                 weaviate_client.connect()
                 self.setup_schema(weaviate_client)
             except Exception as e:
-                self.__logger.error(f"Failed to connect or setup schema: {str(e)}")                
+                self.__logger.error(f"Failed to connect or setup schema: {str(e)}")
             finally:
-                print("i close")
-                weaviate_client.close()
+                try:
+                    weaviate_client.close()
+                except Exception as e:
+                    self.__logger.info(f"Failed to close client: {str(e)}")
         else:
             self.__logger.error("Could not find client, skipping schema setup.")
 
     def setup_schema(self, client):
-        print("in setup schema")
         for class_config in self.schema_config:
-            print("i get config")
             if not client.collections.exists(class_config['name']):
-                print("i create")
                 client.collections.create(class_config['name'],
                                           properties=class_config['properties'])
                 self.__logger.info(
                     f"Weaviate schema for class {class_config['name']} setup complete.")
-        print("finish schema")
 
     async def search(self, query: str, limit=5, distance=0.3) -> list[object]:
         """
@@ -100,4 +99,7 @@ async def search(self, query: str, limit=5, distance=0.3) -> list[object]:
             self.__logger.error(f"Error during search: {str(e)}")
             return []
         finally:
-            weaviate_client.close()
+            try:
+                weaviate_client.close()
+            except Exception as e:
+                self.__logger.info(f"Failed to close client: {str(e)}")