From 877ca407b65177e33be45ca211b966b89fa4a236 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Wed, 20 Dec 2023 12:39:15 -0800 Subject: [PATCH] fix lint --- ollama/__init__.py | 9 +++++++++ ollama/client.py | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) 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. '''