Merge pull request #384 from akx/export-image-model

mxyng · web-flow · commit 4daf4afdb63d · 2024-12-17T12:59:50.000-08:00
fix: export Image model
diff --git 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',
