phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_completion.py): fix merge error

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-07 20:31:38 -07:00
parent e1610d37b9
commit a03860b1cf
1 changed files with 40 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

81
litellm/tests/test_completion.py
View file

@ -892,51 +892,50 @@ def test_completion_claude_3_base64():
"model", ["gemini/gemini-1.5-flash"] # "claude-3-sonnet-20240229", "model", ["gemini/gemini-1.5-flash"] # "claude-3-sonnet-20240229",
) )
def test_completion_function_plus_image(model): def test_completion_function_plus_image(model):
try: litellm.set_verbose = True
litellm.set_verbose = True
image_content = [ image_content = [
{"type": "text", "text": "What’s in this image?"}, {"type": "text", "text": "What’s in this image?"},
{ {
"type": "image_url", "type": "image_url",
"image_url": { "image_url": {
"url": "https://litellm-listing.s3.amazonaws.com/litellm_logo.png" "url": "https://litellm-listing.s3.amazonaws.com/litellm_logo.png"
},
},
]
image_message = {"role": "user", "content": image_content}
tools = [
{
"type": "function",
"function": {
"name": "get_current_weather",
"description": "Get the current weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"unit": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
},
},
"required": ["location"],
}, },
}, },
] }
image_message = {"role": "user", "content": image_content} ]
tools = [ tool_choice = {"type": "function", "function": {"name": "get_current_weather"}}
{ messages = [
"type": "function", {
"function": { "role": "user",
"name": "get_current_weather", "content": "What's the weather like in Boston today in Fahrenheit?",
"description": "Get the current weather in a given location", }
"parameters": { ]
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"unit": {
"type": "string",
"enum": ["celsius", "fahrenheit"],
},
},
"required": ["location"],
},
},
}
]
tool_choice = {"type": "function", "function": {"name": "get_current_weather"}}
messages = [
{
"role": "user",
"content": "What's the weather like in Boston today in Fahrenheit?",
}
]
try: try:
response = completion( response = completion(