Merge pull request #201 from rstudio/trailing-slash

rstudio · Oct 27, 2023 · b9248cf · b9248cf
2 parents 6a88d23 + 31d3335
commit b9248cf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/vetiver/server.py b/vetiver/server.py
@@ -370,4 +370,8 @@ def vetiver_endpoint(url: str = "http://127.0.0.1:8000/predict") -> str:
     >>> import vetiver
     >>> endpoint = vetiver.vetiver_endpoint(url='http://127.0.0.1:8000/predict')
     """
+    # remove trailing backslash, if it exists
+    if url[-1] == "/":
+        url = url[:-1]
+
     return url
diff --git a/vetiver/tests/test_server.py b/vetiver/tests/test_server.py
@@ -4,6 +4,7 @@
     VetiverAPI,
     vetiver_create_prototype,
     InvalidPTypeError,
+    vetiver_endpoint,
 )
 from pydantic import BaseModel, conint
 from fastapi.testclient import TestClient
@@ -117,3 +118,10 @@ def test_complex_prototype(complex_prototype_model):
 
     with pytest.raises(InvalidPTypeError):
         vetiver_create_prototype(response.json())
+
+
+def test_vetiver_endpoint():
+    url_raw = "http://127.0.0.1:8000/predict/"
+    url = vetiver_endpoint(url_raw)
+
+    assert url == "http://127.0.0.1:8000/predict"