github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Retry session and tenant (#4822)

Browse Source
This commit is contained in:
Zander Chase 2023-05-16 18:54:40 -07:00 committed by GitHub
parent 49e4aaf673
commit d1b6839d97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/callbacks/tracers/langchain.py
View File

@ -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: