import openai

client = openai.OpenAI(
    api_key=LITELLM_PROXY_API_KEY,
    base_url=LITELLM_PROXY_ENDPOINT
)

# WORKS ON ANTHROPIC
#MODEL="anthropic/claude-sonnet-4-5-20250929"

# DOES NOT WORK ON BEDROCK WITH SAME MODEL
# SAYS IT DOES NOT SEE IMAGE OR HALLUCINATES CONTENTS
MODEL="bedrock/global.anthropic.claude-sonnet-4-5-20250929-v1:0"

# test tool
tools=[{
    "type": "function",
    "function": {
        "name": "get_image",
        "description": "Returns an image to you",
        "parameters": {
            "type": "object",
            "properties": {}
        }
    }
}]

# user prompt
messages=[{
    "role": "user",
    "content": "please use tool get_image and describe the image contents"
}]

# send request
response = client.chat.completions.create(
    model=MODEL,
    tools=tools,
    messages=messages
)

# append tool call request
messages.append(response.choices[0].message)

# tool calls made by llm
tool_calls = response.choices[0].message.tool_calls

# append tool call response with image
messages.append({
    "tool_call_id": tool_calls[0].id,
    "role": "tool",
    "name": tool_calls[0].function.name,
    "content": [{
        "type": "image_url",
        "image_url": {
          "url": "https://upload.wikimedia.org/wikipedia/commons/3/3f/JPEG_example_flower.jpg"
        }
    }],
})

# send tool response
response = client.chat.completions.create(
    model=MODEL,
    tools=tools,
    messages=messages
)

# print assistant response
print(response.choices[0].message)