From 9057705bc081b16d3cc91e91d6ec8acd083b3945 Mon Sep 17 00:00:00 2001
From: Aarni Koskela <akx@iki.fi>
Date: Tue, 17 Dec 2024 13:31:25 +0200
Subject: [PATCH] fix: export Image model

Allows construction of `Message(..., images=Image(value=...))` without having to import from `._types`
---
 ollama/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ollama/__init__.py b/ollama/__init__.py
index 23d736a8..2517b50d 100644
--- a/ollama/__init__.py
+++ b/ollama/__init__.py
@@ -2,6 +2,7 @@
 from ollama._types import (
   Options,
   Message,
+  Image,
   Tool,
   GenerateResponse,
   ChatResponse,
@@ -21,6 +22,7 @@
   'AsyncClient',
   'Options',
   'Message',
+  'Image',
   'Tool',
   'GenerateResponse',
   'ChatResponse',