From 84fbc903aa7d7fbaf480de0c4c623634ea536230 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 30 Dec 2023 20:19:22 +0530 Subject: [PATCH] (test) langfuse - set custom trace_id --- litellm/tests/test_alangfuse.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py index fded026e2..6a10528df 100644 --- a/litellm/tests/test_alangfuse.py +++ b/litellm/tests/test_alangfuse.py @@ -231,6 +231,10 @@ def test_langfuse_logging_custom_generation_name(): "langfuse/foo": "bar", "langsmith/fizz": "buzz", "prompt_hash": "asdf98u0j9131123", + "generation_name": "ishaan-test-generation", + "generation_id": "gen-id22", + "trace_id": "trace-id22", + "trace_user_id": "user-id2", }, ) print(response) @@ -241,7 +245,7 @@ def test_langfuse_logging_custom_generation_name(): print(e) -# test_langfuse_logging_custom_generation_name() +test_langfuse_logging_custom_generation_name() @pytest.mark.skip(reason="beta test - checking langfuse output")