phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix langsmith tenacity

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-11 13:48:44 -07:00
parent f339f9614a
commit d84fa05161
1 changed files with 0 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/integrations/langsmith.py
View file

@ -14,12 +14,6 @@ import dotenv # type: ignore
import httpx import httpx
import requests # type: ignore import requests # type: ignore
from pydantic import BaseModel # type: ignore from pydantic import BaseModel # type: ignore
from tenacity import (
retry,
retry_if_exception_type,
stop_after_attempt,
wait_exponential,
)
import litellm import litellm
from litellm._logging import verbose_logger from litellm._logging import verbose_logger
@ -303,11 +297,6 @@ class LangsmithLogger(CustomLogger):
self.log_queue.clear() self.log_queue.clear()
self.last_flush_time = time.time() self.last_flush_time = time.time()
@retry(
stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10),
retry=retry_if_exception_type((httpx.HTTPStatusError, Exception)),
)
async def _async_send_batch(self): async def _async_send_batch(self):
""" """
sends runs to /batch endpoint sends runs to /batch endpoint