ollama-python/ollama/__init__.py
2024-11-08 09:59:03 -08:00

52 lines
890 B
Python

from ollama._client import Client, AsyncClient
from ollama._types import (
Options,
Message,
Tool,
GenerateResponse,
ChatResponse,
EmbedResponse,
EmbeddingsResponse,
StatusResponse,
ProgressResponse,
ListResponse,
ShowResponse,
ProcessResponse,
RequestError,
ResponseError,
)
__all__ = [
'Client',
'AsyncClient',
'Options',
'Message',
'Tool',
'GenerateResponse',
'ChatResponse',
'EmbedResponse',
'EmbeddingsResponse',
'StatusResponse',
'ProgressResponse',
'ListResponse',
'ShowResponse',
'ProcessResponse',
'RequestError',
'ResponseError',
]
_client = Client()
generate = _client.generate
chat = _client.chat
embed = _client.embed
embeddings = _client.embeddings
pull = _client.pull
push = _client.push
create = _client.create
delete = _client.delete
list = _client.list
copy = _client.copy
show = _client.show
ps = _client.ps