ollama/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-02-04 02:54:08 -06:00
Code Issues Packages Projects Releases 30 Wiki Activity

fix lint

Browse Source
This commit is contained in:
Michael Yang 2023-12-20 12:39:15 -08:00
parent 717abfbaed
commit 877ca407b6
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ollama/__init__.py
View File

@ -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)

2
ollama/client.py
View File

@ -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.
'''