Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add default async client aliases #3

Merged
merged 1 commit into from
Jan 4, 2024
Merged

add default async client aliases #3

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 21 additions & 36 deletions ollama/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,41 +16,26 @@
'show',
]


_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)
generate = _default_client.generate
chat = _default_client.chat
pull = _default_client.pull
push = _default_client.push
create = _default_client.create
delete = _default_client.delete
list = _default_client.list
copy = _default_client.copy
show = _default_client.show

_async_default_client = AsyncClient()

async_generate = _async_default_client.generate
async_chat = _async_default_client.chat
async_pull = _async_default_client.pull
async_push = _async_default_client.push
async_create = _async_default_client.create
async_delete = _async_default_client.delete
async_list = _async_default_client.list
async_copy = _async_default_client.copy
async_show = _async_default_client.show