From cee20695ebbc83be4dc3801bb4f1f0b8e3667704 Mon Sep 17 00:00:00 2001
From: Lunik <lunik@tiwabbit.fr>
Date: Mon, 26 Feb 2024 17:52:42 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix:=20Ollama=20vision=20models?=
 =?UTF-8?q?=20call=20arguments=20(like=20:=20llava)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Lunik <lunik@tiwabbit.fr>
---
 litellm/llms/ollama.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/litellm/llms/ollama.py b/litellm/llms/ollama.py
index 9339deb78..3611ccd8b 100644
--- a/litellm/llms/ollama.py
+++ b/litellm/llms/ollama.py
@@ -147,6 +147,7 @@ def get_ollama_response(
 
     stream = optional_params.pop("stream", False)
     format = optional_params.pop("format", None)
+    images = optional_params.pop("images", None)
     data = {
         "model": model,
         "prompt": prompt,
@@ -155,6 +156,8 @@ def get_ollama_response(
     }
     if format is not None:
         data["format"] = format
+    if images is not None:
+        data["images"] = images
 
     ## LOGGING
     logging_obj.pre_call(