File size: 727 Bytes
7db0ae4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# import openai
# client = openai.OpenAI(
#     api_key="anything",
#     base_url="http://0.0.0.0:8000"
# )

# # request sent to model set on litellm proxy, `litellm --model`
# 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)