fix: add adapters if not set

langchain-ai · Mar 12, 2024 · 78eab21 · 78eab21
1 parent 0aec28e
commit 78eab21
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/python/langsmith/client.py b/python/langsmith/client.py
@@ -440,10 +440,12 @@ def __init__(
         else:
             self.tracing_queue = None
 
-        # Mount the HTTPAdapter with the retry configuration
+        # Mount the HTTPAdapter with the retry configuration. Don't overwrite if session already has it
         adapter = requests_adapters.HTTPAdapter(max_retries=self.retry_config)
-        self.session.mount("http://", adapter)
-        self.session.mount("https://", adapter)
+        if not self.session.adapters.get("http://"):
+            self.session.mount("http://", adapter)
+        if not self.session.adapters.get("https://"):
+            self.session.mount("https://", adapter)
         self._get_data_type_cached = functools.lru_cache(maxsize=10)(
             self._get_data_type
         )

diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langsmith"
-version = "0.1.23"
+version = "0.1.24"
 description = "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform."
 authors = ["LangChain <[email protected]>"]
 license = "MIT"