ollama-python/ollama/__init__.py
2024-01-10 10:52:50 -08:00

38 lines
560 B
Python

from ollama._client import Client, AsyncClient
from ollama._types import (
GenerateResponse,
ChatResponse,
Message,
Options,
)
__all__ = [
'Client',
'AsyncClient',
'GenerateResponse',
'ChatResponse',
'Message',
'Options',
'generate',
'chat',
'pull',
'push',
'create',
'delete',
'list',
'copy',
'show',
]
_client = Client()
generate = _client.generate
chat = _client.chat
pull = _client.pull
push = _client.push
create = _client.create
delete = _client.delete
list = _client.list
copy = _client.copy
show = _client.show