🐛 fix: Ollama vision models call arguments (like : llava)

Signed-off-by: Lunik <lunik@tiwabbit.fr>
This commit is contained in:
Lunik
2024-02-26 17:52:55 +01:00
parent 4067d4f1e5
commit cee20695eb
+3
View File
@@ -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(