phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +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 19:32:22 -07:00
parent ac6370c88b
commit f52dd2bb35
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_langsmith.py
View file

@ -14,6 +14,7 @@ import litellm
from litellm import completion from litellm import completion
from litellm._logging import verbose_logger from litellm._logging import verbose_logger
from litellm.integrations.langsmith import LangsmithLogger from litellm.integrations.langsmith import LangsmithLogger
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler
verbose_logger.setLevel(logging.DEBUG) verbose_logger.setLevel(logging.DEBUG)
@ -74,6 +75,11 @@ async def test_async_langsmith_logging():
assert "user_api_key_user_id" in extra_fields_on_langsmith assert "user_api_key_user_id" in extra_fields_on_langsmith
assert "user_api_key_team_alias" in extra_fields_on_langsmith assert "user_api_key_team_alias" in extra_fields_on_langsmith
for cb in litellm.callbacks:
if isinstance(cb, LangsmithLogger):
await cb.async_httpx_client.client.aclose()
# test_langsmith_logger.async_httpx_client.close()
except Exception as e: except Exception as e:
print(e) print(e)
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
@ -95,6 +101,10 @@ def test_async_langsmith_logging_with_metadata():
print(response) print(response)
time.sleep(3) time.sleep(3)
for cb in litellm.callbacks:
if isinstance(cb, LangsmithLogger):
cb.async_httpx_client.close()
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
print(e) print(e)
@ -119,6 +129,9 @@ async def test_async_langsmith_logging_with_streaming_and_metadata(sync_mode):
stream=True, stream=True,
metadata={"id": run_id}, metadata={"id": run_id},
) )
for cb in litellm.callbacks:
if isinstance(cb, LangsmithLogger):
cb.async_httpx_client = AsyncHTTPHandler()
for chunk in response: for chunk in response:
continue continue
time.sleep(3) time.sleep(3)
@ -132,6 +145,9 @@ async def test_async_langsmith_logging_with_streaming_and_metadata(sync_mode):
stream=True, stream=True,
metadata={"id": run_id}, metadata={"id": run_id},
) )
for cb in litellm.callbacks:
if isinstance(cb, LangsmithLogger):
cb.async_httpx_client = AsyncHTTPHandler()
async for chunk in response: async for chunk in response:
continue continue
await asyncio.sleep(3) await asyncio.sleep(3)