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

fix langsmith logging test

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-17 18:34:49 -07:00
parent 21b8fdf8bf
commit e49001c5f2
3 changed files with 9 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/langsmith.py
View file

@ -8,6 +8,7 @@ from datetime import datetime
from typing import Any, List, Optional, Union
import dotenv # type: ignore
import httpx
import requests # type: ignore
from pydantic import BaseModel # type: ignore
@ -59,7 +60,9 @@ class LangsmithLogger(CustomLogger):
self.langsmith_base_url = os.getenv(
"LANGSMITH_BASE_URL", "https://api.smith.langchain.com"
)
self.async_httpx_client = AsyncHTTPHandler()
self.async_httpx_client = AsyncHTTPHandler(
timeout=httpx.Timeout(timeout=600.0, connect=5.0)
)
def _prepare_log_data(self, kwargs, response_obj, start_time, end_time):
import datetime