Enforce trailing slash in the client (#126)

Would resolve the error in #125 --------- Co-authored-by: Eugene Yurtsev <[email protected]>
langchain-ai · Oct 30, 2023 · a94b666 · a94b666
1 parent 7be2927
commit a94b666
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/langserve/client.py b/langserve/client.py
@@ -365,6 +365,11 @@ def _batch(
 
         return outputs
 
+    def _enforce_trailing_slash(self, url: str) -> str:
+        if url.endswith("/"):
+            return url
+        return url + "/"
+
     def batch(
         self,
         inputs: List[Input],
@@ -459,7 +464,7 @@ def stream(
             "config": _without_callbacks(config),
             "kwargs": kwargs,
         }
-        endpoint = urljoin(self.url, "stream")
+        endpoint = urljoin(self._enforce_trailing_slash(self.url), "stream")
 
         try:
             from httpx_sse import connect_sse
@@ -522,7 +527,7 @@ async def astream(
             "config": _without_callbacks(config),
             "kwargs": kwargs,
         }
-        endpoint = urljoin(self.url, "stream")
+        endpoint = urljoin(self._enforce_trailing_slash(self.url), "stream")
 
         try:
             from httpx_sse import aconnect_sse