Merge branch 'main' into eugene/add_unique_names

langchain-ai · Oct 17, 2023 · 90c70ec · 90c70ec
2 parents 2992164 + 1902209
commit 90c70ec
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 3 deletions.
diff --git a/langserve/client.py b/langserve/client.py
@@ -6,6 +6,7 @@
 from typing import (
     Any,
     AsyncIterator,
+    Dict,
     Iterator,
     List,
     Optional,
@@ -15,6 +16,7 @@
 from urllib.parse import urljoin
 
 import httpx
+from httpx._types import AuthTypes, CertTypes, CookieTypes, HeaderTypes, VerifyTypes
 from langchain.callbacks.tracers.log_stream import RunLog, RunLogPatch
 from langchain.load.dump import dumpd
 from langchain.schema.runnable import Runnable
@@ -110,16 +112,48 @@ def __init__(
         url: str,
         *,
         timeout: Optional[float] = None,
+        auth: Optional[AuthTypes] = None,
+        headers: Optional[HeaderTypes] = None,
+        cookies: Optional[CookieTypes] = None,
+        verify: VerifyTypes = True,
+        cert: Optional[CertTypes] = None,
+        client_kwargs: Optional[Dict[str, Any]] = None,
     ) -> None:
         """Initialize the client.
 
         Args:
             url: The url of the server
             timeout: The timeout for requests
+            auth: Authentication class for requests
+            headers: Headers to send with requests
+            cookies: Cookies to send with requests
+            verify: Whether to verify SSL certificates
+            cert: SSL certificate to use for requests
+            client_kwargs: If provided will be unpacked as kwargs to both the sync
+                           and async httpx clients
         """
+        _client_kwargs = client_kwargs or {}
         self.url = url
-        self.sync_client = httpx.Client(base_url=url, timeout=timeout)
-        self.async_client = httpx.AsyncClient(base_url=url, timeout=timeout)
+        self.sync_client = httpx.Client(
+            base_url=url,
+            timeout=timeout,
+            auth=auth,
+            headers=headers,
+            cookies=cookies,
+            verify=verify,
+            cert=cert,
+            **_client_kwargs,
+        )
+        self.async_client = httpx.AsyncClient(
+            base_url=url,
+            timeout=timeout,
+            auth=auth,
+            headers=headers,
+            cookies=cookies,
+            verify=verify,
+            cert=cert,
+            **_client_kwargs,
+        )
 
         # Register cleanup handler once RemoteRunnable is garbage collected
         weakref.finalize(self, _close_clients, self.sync_client, self.async_client)

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langserve"
-version = "0.0.8"
+version = "0.0.9"
 description = ""
 readme = "README.md"
 authors = ["LangChain"]