diff --git a/ollama/__init__.py b/ollama/__init__.py index 92bba280..c0bb46db 100644 --- a/ollama/__init__.py +++ b/ollama/__init__.py @@ -1,4 +1,5 @@ from ollama._client import AsyncClient, Client +from importlib.metadata import version, PackageNotFoundError from ollama._types import ( ChatResponse, EmbeddingsResponse, @@ -57,3 +58,9 @@ ps = _client.ps web_search = _client.web_search web_fetch = _client.web_fetch + + +try: + __version__ = version("ollama") +except PackageNotFoundError: + __version__ = "unknown" \ No newline at end of file