diff --git a/examples/list.py b/examples/list.py index 00d6243..efc5e89 100644 --- a/examples/list.py +++ b/examples/list.py @@ -4,6 +4,7 @@ for model in response.models: print('Name:', model.model) + print(' Capabilities:', model.capabilities) print(' Size (MB):', f'{(model.size.real / 1024 / 1024):.2f}') if model.details: print(' Format:', model.details.format) diff --git a/ollama/_types.py b/ollama/_types.py index f86c06d..ef81569 100644 --- a/ollama/_types.py +++ b/ollama/_types.py @@ -464,6 +464,7 @@ class Model(SubscriptableBaseModel): digest: Optional[str] = None size: Optional[ByteSize] = None details: Optional[ModelDetails] = None + capabilities: Optional[List[str]] = None models: Sequence[Model] 'List of models.'