feat: models alias

Author: vansangpfiev

engine/e2e-test/main.py

@@ -11,7 +11,8 @@
 # from test_api_model_delete import TestApiModelDelete
 # from test_api_model_list import TestApiModelList
 # from test_api_model_get import TestApiModelGet
-from test_api_model_update import TestApiModelUpdate
+# from test_api_model_update import TestApiModelUpdate
+from test_api_model_alias import TestApiModelAlias
 ###
 # from test_cli_engine_get import TestCliEngineGet
 # from test_cli_engine_install import TestCliEngineInstall

engine/e2e-test/test_api_model_alias.py

@@ -0,0 +1,30 @@
+import pytest
+import requests
+from test_runner import popen, run
+from test_runner import start_server, stop_server
+
+
+class TestApiModelAlias:
+
+    @pytest.fixture(autouse=True)
+    def setup_and_teardown(self):
+        # Setup
+        success = start_server()
+        if not success:
+            raise Exception("Failed to start server")
+
+        # TODO: using pull with branch for easy testing tinyllama:gguf for example
+        popen(["pull", "tinyllama"], "1\n")
+
+        yield
+
+        # Teardown
+        # Clean up
+        run("Delete model", ["models", "delete", "tinyllama:gguf"])
+        stop_server()
+
+    def test_models_set_alias_should_be_successful(self):
+        body_json = {'modelId': 'tinyllama:gguf',
+                     'modelAlias': 'tg'}
+        response = requests.post("http://localhost:3928/models/alias", json = body_json)        
+        assert response.status_code == 200