Retry session and tenant (#4822)

langchain/callbacks/tracers/langchain.py

@@ -8,6 +8,7 @@ from typing import Any, Dict, List, Optional
 from uuid import UUID
 
 import requests
+from tenacity import retry, stop_after_attempt, wait_fixed
 
 from langchain.callbacks.tracers.base import BaseTracer
 from langchain.callbacks.tracers.schemas import (
@@ -33,6 +34,7 @@ def get_endpoint() -> str:
     return os.getenv("LANGCHAIN_ENDPOINT", "http://localhost:8000")
 
 
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(0.5))
 def _get_tenant_id(
     tenant_id: Optional[str], endpoint: Optional[str], headers: Optional[dict]
 ) -> str:
@@ -106,6 +108,7 @@ class LangChainTracer(BaseTracer):
         self.tenant_id = tenant_id
         return tenant_id
 
+    @retry(stop=stop_after_attempt(3), wait=wait_fixed(0.5))
     def ensure_session(self) -> TracerSession:
         """Upsert a session."""
         if self.session is not None: