From d1b6839d97ea1b0c60f226633da34d97a130c183 Mon Sep 17 00:00:00 2001
From: Zander Chase <130414180+vowelparrot@users.noreply.github.com>
Date: Tue, 16 May 2023 18:54:40 -0700
Subject: [PATCH] Retry session and tenant (#4822)

---
 langchain/callbacks/tracers/langchain.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/langchain/callbacks/tracers/langchain.py b/langchain/callbacks/tracers/langchain.py
index e8603905..97d714ea 100644
--- a/langchain/callbacks/tracers/langchain.py
+++ b/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: