chore: e2e tests

janhq · Dec 27, 2024 · 5df12cf · 5df12cf
1 parent 26daec9
commit 5df12cf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 9 deletions.
diff --git a/engine/e2e-test/test_api_engine.py b/engine/e2e-test/test_api_engine.py
@@ -44,8 +44,7 @@ def test_engines_install_llamacpp_specific_version_and_null_variant(self):
     # engines uninstall
     @pytest.mark.asyncio
     async def test_engines_install_uninstall_llamacpp_should_be_successful(self):
-        data = {"version": "v0.1.43"}
-        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install", json=data)
+        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install")
         assert response.status_code == 200
         await wait_for_websocket_download_success_event(timeout=None)
         time.sleep(30)

diff --git a/engine/e2e-test/test_api_engine_install_nightly.py b/engine/e2e-test/test_api_engine_install_nightly.py
@@ -19,8 +19,7 @@ def setup_and_teardown(self):
         stop_server()
 
     def test_engines_install_llamacpp_should_be_successful(self):
-        data = {"version": "v0.1.43"}
-        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install", json=data)
+        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install")
         assert response.status_code == 200
 
     def test_engines_install_llamacpp_specific_version_and_variant(self):

diff --git a/engine/e2e-test/test_api_model.py b/engine/e2e-test/test_api_model.py
@@ -88,9 +88,7 @@ async def test_model_pull_with_direct_url_should_have_desired_name(self):
     @pytest.mark.asyncio
     async def test_models_start_stop_should_be_successful(self):
         print("Install engine")
-        # TODO(sang) Remove version after marking 0.1.43 as stable
-        data = {"version": "v0.1.43"}
-        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install", json=data)
+        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install")
         assert response.status_code == 200
         await wait_for_websocket_download_success_event(timeout=None)
         # TODO(sang) need to fix for cuda download

diff --git a/engine/e2e-test/test_cli_engine_uninstall.py b/engine/e2e-test/test_cli_engine_uninstall.py
@@ -24,8 +24,7 @@ def setup_and_teardown(self):
 
     @pytest.mark.asyncio
     async def test_engines_uninstall_llamacpp_should_be_successfully(self):
-        data = {"version": "v0.1.43"}
-        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install", json=data)
+        response = requests.post("http://localhost:3928/v1/engines/llama-cpp/install")
         await wait_for_websocket_download_success_event(timeout=None)
         exit_code, output, error = run(
             "Uninstall engine", ["engines", "uninstall", "llama-cpp"]