From 694956b44e89cce385da30b1a1a3dce63b2ecc5a Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 1 Jan 2024 11:12:57 +0530 Subject: [PATCH] (test) proxy - pass metadata to openai client --- litellm/proxy/tests/test_openai_request.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/litellm/proxy/tests/test_openai_request.py b/litellm/proxy/tests/test_openai_request.py index f49fce7bb..16755db0b 100644 --- a/litellm/proxy/tests/test_openai_request.py +++ b/litellm/proxy/tests/test_openai_request.py @@ -5,11 +5,22 @@ # ) # # request sent to model set on litellm proxy, `litellm --model` -# response = client.chat.completions.create(model="gpt-3.5-turbo", messages = [ -# { -# "role": "user", -# "content": "this is a test request, write a short poem" +# response = client.chat.completions.create( +# model="azure/chatgpt-v-2", +# messages = [ +# { +# "role": "user", +# "content": "this is a test request, write a short poem" +# } +# ], +# extra_body={ +# "metadata": { +# "generation_name": "ishaan-generation-openai-client", +# "generation_id": "openai-client-gen-id22", +# "trace_id": "openai-client-trace-id22", +# "trace_user_id": "openai-client-user-id2" +# } # } -# ]) +# ) # print(response)