Updated Docs and Github Actions

Author: UtkarshTheDev

CHANGELOG.md

@@ -2,6 +2,12 @@
 
 All notable changes for version updates.
 
+## [0.1.3] - 2024-02-27
+
+### Changed
+
+- Updated GitHub Actions workflow to use the --no-cache-dir flag in pip install commands, which prevents disk space issues during dependency installation (e.g., for large packages like torch).
+
 ## [0.1.2] - 2024-02-25
 
 ### Changed

locallab/__init__.py

@@ -2,7 +2,7 @@
 LocalLab - A lightweight AI inference server
 """
 
-__version__ = "0.1.2" 
+__version__ = "0.1.3" 
 
 from typing import Dict, Any, Optional
 

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="locallab",
-    version="0.1.2",
+    version="0.1.3",
     packages=find_packages(include=["locallab", "locallab.*"]),
     install_requires=[
         "fastapi>=0.68.0,<1.0.0",

tests/conftest.py

@@ -1,34 +1,39 @@
 import pytest
 from fastapi.testclient import TestClient
-from unittest.mock import Mock, patch
-from server.locallab.main import app
-from server.locallab.model_manager import ModelManager
-from server.locallab.client import LocalLabClient
+from unittest.mock import Mock
+from locallab.main import app
+from locallab.model_manager import ModelManager
 
 @pytest.fixture
 def test_client():
+    """
+    Fixture to create a TestClient for the FastAPI app.
+    """
     return TestClient(app)
 
 @pytest.fixture
 def model_manager():
+    """
+    Fixture to create a new instance of the ModelManager.
+    """
     return ModelManager()
 
 @pytest.fixture
 def mock_tokenizer():
+    """
+    Fixture for a mocked tokenizer.
+    Returns a simple dictionary mimicking tokenizer output.
+    """
     tokenizer = Mock()
     tokenizer.return_value = {"input_ids": [1, 2, 3], "attention_mask": [1, 1, 1]}
     return tokenizer
 
 @pytest.fixture
 def mock_model():
+    """
+    Fixture for a mocked model.
+    Mocks the generate method.
+    """
     model = Mock()
     model.generate.return_value = [[1, 2, 3, 4]]
     return model
-
-@pytest.fixture
-def locallab_client():
-    with patch('locallab.client.requests') as mock_requests:
-        client = LocalLabClient("http://localhost:8000")
-        mock_requests.get.return_value.json.return_value = {"status": "ok"}
-        mock_requests.post.return_value.json.return_value = {"response": "test response"}
-        yield client