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
@@ -53,16 +54,13 @@ def __init__(self, data_path='./weaviate_data'):
             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")
+        time.sleep(1)
 
     async def search(self, query: str, limit=5, distance=0.3) -> list[object]:
         """