diff --git a/ollama/__init__.py b/ollama/__init__.py index 23d736a..2517b50 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',