readd gif lfs pointers for models

.github/workflows/image-publish.yml

@@ -9,8 +9,8 @@ on:
     branches:
       - main
   schedule:
-    # Weekdays at noon GMT
-    - cron: '00 12 * * 1-5'
+    # Once weekly on fridays at noon
+    - cron: '00 12 * * 5'
   # Allow for manually triggering the workflow
   workflow_dispatch:
 jobs:

README.md

@@ -120,6 +120,10 @@ source venv/bin/activate  # On Windows: venv\Scripts\activate
 
 # Install dev dependencies
 pip install -e ".[dev]"
+
+# Download models
+git lfs install
+git lfs pull
 ```
 
 ### Testing

tests/test_storage.py

@@ -34,23 +34,21 @@ def mock_inference_engine():
 
 @pytest.mark.asyncio
 async def test_search(mock_weaviate_client, mock_inference_engine):
-    # Patch the WeaviateClient and LlamaCppInferenceEngine inside the test function
-    with (
-        patch("weaviate.WeaviateClient", return_value=mock_weaviate_client),
-        patch(
-            "codegate.inference.inference_engine.LlamaCppInferenceEngine",
-            return_value=mock_inference_engine,
-        ),
-    ):
-
-        # Initialize StorageEngine
-        storage_engine = StorageEngine(data_path="./weaviate_data")
-
-        # Invoke the search method
-        results = await storage_engine.search("test query", 5, 0.3)
-
-        # Assertions to validate the expected behavior
-        assert len(results) == 1  # Assert that one result is returned
-        assert results[0]["properties"]["name"] == "test"
-        mock_weaviate_client.connect.assert_called()
-        mock_weaviate_client.close.assert_called()
+    # Patch the LlamaCppInferenceEngine.embed method (not the entire class)
+    with patch("codegate.inference.inference_engine.LlamaCppInferenceEngine.embed",
+               mock_inference_engine.embed):
+
+        # Mock the WeaviateClient as before
+        with patch("weaviate.WeaviateClient", return_value=mock_weaviate_client):
+
+            # Initialize StorageEngine
+            storage_engine = StorageEngine(data_path="./weaviate_data")
+
+            # Invoke the search method
+            results = await storage_engine.search("test query", 5, 0.3)
+
+            # Assertions to validate the expected behavior
+            assert len(results) == 1  # Assert that one result is returned
+            assert results[0]["properties"]["name"] == "test"
+            mock_weaviate_client.connect.assert_called()
+            mock_weaviate_client.close.assert_called()