From 3615139ea099d2e4b130df060ad802de555908c0 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Wed, 4 Jun 2025 11:44:11 -0700 Subject: [PATCH 1/2] types: add capabilities to list response --- examples/list.py | 1 + ollama/_types.py | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/list.py b/examples/list.py index 00d6243c..577e1b84 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 f86c06d2..ef81569d 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.' From fdba575aeb38bff6b4ea8245ecef99c6a4429300 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Wed, 4 Jun 2025 11:51:39 -0700 Subject: [PATCH 2/2] fix formatting --- examples/list.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/list.py b/examples/list.py index 577e1b84..efc5e899 100644 --- a/examples/list.py +++ b/examples/list.py @@ -4,7 +4,7 @@ for model in response.models: print('Name:', model.model) - print(' Capabilities:', model.capabilities) + print(' Capabilities:', model.capabilities) print(' Size (MB):', f'{(model.size.real / 1024 / 1024):.2f}') if model.details: print(' Format:', model.details.format)