diff --git a/ollama/__init__.py b/ollama/__init__.py index 8e7dc22..99b0adf 100644 --- a/ollama/__init__.py +++ b/ollama/__init__.py @@ -2,29 +2,38 @@ from ollama.client import Client _default_client = Client() + def generate(*args, **kwargs): return _default_client.generate(*args, **kwargs) + def chat(*args, **kwargs): return _default_client.chat(*args, **kwargs) + def pull(*args, **kwargs): return _default_client.pull(*args, **kwargs) + def push(*args, **kwargs): return _default_client.push(*args, **kwargs) + def create(*args, **kwargs): return _default_client.create(*args, **kwargs) + def delete(*args, **kwargs): return _default_client.delete(*args, **kwargs) + def list(*args, **kwargs): return _default_client.list(*args, **kwargs) + def copy(*args, **kwargs): return _default_client.copy(*args, **kwargs) + def show(*args, **kwargs): return _default_client.show(*args, **kwargs) diff --git a/ollama/client.py b/ollama/client.py index 578757f..d80fd56 100644 --- a/ollama/client.py +++ b/ollama/client.py @@ -137,9 +137,9 @@ class Client(BaseClient): def _encode_image(image): ''' _encode_images takes a list of images and returns a generator of base64 encoded images. - if the image is a bytes object, it is assumed to be the raw bytes of an image. if the image is a string, it is assumed to be a path to a file. if the image is a Path object, it is assumed to be a path to a file. + if the image is a bytes object, it is assumed to be the raw bytes of an image. if the image is a file-like object, it is assumed to be a container to the raw bytes of an image. '''