From 8955d783d144a91ff8795f9ed0b3cf0bede9936d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 17 Jul 2024 11:29:29 -0700 Subject: [PATCH] test correct fields are logged on langsmith --- litellm/tests/test_langsmith.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_langsmith.py b/litellm/tests/test_langsmith.py index a4025c851f..8af0c9cbb5 100644 --- a/litellm/tests/test_langsmith.py +++ b/litellm/tests/test_langsmith.py @@ -26,7 +26,18 @@ def test_langsmith_logging(): messages=[{"role": "user", "content": "what llm are u"}], max_tokens=10, temperature=0.2, - metadata={"id": run_id}, + metadata={ + "id": run_id, + "user_api_key": "6eb81e014497d89f3cc1aa9da7c2b37bda6b7fea68e4b710d33d94201e68970c", + "user_api_key_alias": "ishaans-langmsith-key", + "user_api_end_user_max_budget": None, + "litellm_api_version": "1.40.19", + "global_max_parallel_requests": None, + "user_api_key_user_id": "admin", + "user_api_key_org_id": None, + "user_api_key_team_id": "dbe2f686-a686-4896-864a-4c3924458709", + "user_api_key_team_alias": "testing-team", + }, ) print(response) time.sleep(3) @@ -49,6 +60,11 @@ def test_langsmith_logging(): assert "api_key" not in input_fields_on_langsmith assert "api_key" not in extra_fields_on_langsmith + # assert user_api_key in extra_fields_on_langsmith + assert "user_api_key" 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 + except Exception as e: print(e)