ollama-models 0.1.0__tar.gz

ollama_models-0.1.0/.gitignore

@@ -0,0 +1,375 @@
+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+lerna-debug.log*
+
+# Diagnostic reports (https://nodejs.org/api/report.html)
+report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json
+
+# Runtime data
+pids
+*.pid
+*.seed
+*.pid.lock
+
+# Directory for instrumented libs generated by jscoverage/JSCover
+lib-cov
+
+# Coverage directory used by tools like istanbul
+coverage
+*.lcov
+
+# nyc test coverage
+.nyc_output
+
+# Grunt intermediate storage (https://gruntjs.com/creating-plugins#storing-task-files)
+.grunt
+
+# Bower dependency directory (https://bower.io/)
+bower_components
+
+# node-waf configuration
+.lock-wscript
+
+# Compiled binary addons (https://nodejs.org/api/addons.html)
+build/Release
+
+# Dependency directories
+node_modules/
+jspm_packages/
+
+# Snowpack dependency directory (https://snowpack.dev/)
+web_modules/
+
+# TypeScript cache
+*.tsbuildinfo
+
+# Optional npm cache directory
+.npm
+
+# Optional eslint cache
+.eslintcache
+
+# Optional stylelint cache
+.stylelintcache
+
+# Optional REPL history
+.node_repl_history
+
+# Output of 'npm pack'
+*.tgz
+
+# Yarn Integrity file
+.yarn-integrity
+
+# dotenv environment variable files
+.env
+.env.*
+!.env.example
+
+# parcel-bundler cache (https://parceljs.org/)
+.cache
+.parcel-cache
+
+# Next.js build output
+.next
+out
+
+# Nuxt.js build / generate output
+.nuxt
+dist
+.output
+
+# Gatsby files
+.cache/
+# Comment in the public line in if your project uses Gatsby and not Next.js
+# https://nextjs.org/blog/next-9-1#public-directory-support
+# public
+
+# vuepress build output
+.vuepress/dist
+
+# vuepress v2.x temp and cache directory
+.temp
+.cache
+
+# Sveltekit cache directory
+.svelte-kit/
+
+# vitepress build output
+**/.vitepress/dist
+
+# vitepress cache directory
+**/.vitepress/cache
+
+# Docusaurus cache and generated files
+.docusaurus
+
+# Serverless directories
+.serverless/
+
+# FuseBox cache
+.fusebox/
+
+# DynamoDB Local files
+.dynamodb/
+
+# Firebase cache directory
+.firebase/
+
+# TernJS port file
+.tern-port
+
+# Stores VSCode versions used for testing VSCode extensions
+.vscode-test
+
+# pnpm
+.pnpm-store
+
+# yarn v3
+.pnp.*
+.yarn/*
+!.yarn/patches
+!.yarn/plugins
+!.yarn/releases
+!.yarn/sdks
+!.yarn/versions
+
+# Vite files
+vite.config.js.timestamp-*
+vite.config.ts.timestamp-*
+.vite/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#   Usually these files are written by a python script from a template
+#   before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+# Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+# uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+# poetry.lock
+# poetry.toml
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-version-control
+# pdm.lock
+# pdm.toml
+.pdm-python
+.pdm-build/
+
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+# pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .venv directory. It is recommended not to include this directory in version control.
+.pixi
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# Redis
+*.rdb
+*.aof
+*.pid
+
+# RabbitMQ
+mnesia/
+rabbitmq/
+rabbitmq-data/
+
+# ActiveMQ
+activemq-data/
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#   JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#   be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#   and can be added to the global gitignore or merged into this file.  For a more nuclear
+#   option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# .idea/
+
+# Abstra
+#   Abstra is an AI-powered process automation framework.
+#   Ignore directories containing user credentials, local state, and settings.
+#   Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#   Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#   that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#   and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#   you could uncomment the following to ignore the entire vscode folder
+# .vscode/
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml
+
+# Cloudflare Wrangler
+.wrangler/
+
+# Rye virtual envs
+.venv/
+.python-version
+
+# tsup / tsc build output
+dist/
+
+# Temp (system)
+.temp/
+

ollama_models-0.1.0/PKG-INFO

@@ -0,0 +1,63 @@
+Metadata-Version: 2.4
+Name: ollama-models
+Version: 0.1.0
+Summary: Python client for the ollama-models API — search and list Ollama model weights
+Project-URL: Repository, https://github.com/devcomfort/ollama-models
+Author-email: devcomfort <im@devcomfort.me>
+License: MIT
+Keywords: llm,models,ollama,registry,search
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.27.0
+Description-Content-Type: text/markdown
+
+# ollama-models (Python)
+
+Python client for searching and listing models from the [Ollama](https://ollama.com) registry.
+
+## Installation
+
+```bash
+pip install ollama-models
+```
+
+## Usage
+
+```python
+from ollama_models import OllamaModelsClient
+
+# No base URL needed — defaults to the official hosted instance
+client = OllamaModelsClient()
+
+# Pass a base URL only if you self-host the API
+# client = OllamaModelsClient("https://your-own-instance.workers.dev")
+
+# Search models
+result = client.search("qwen3", page=1)
+for page in result.pages:
+    print(page.http_url)
+
+# Get all tags for a model
+model = client.get_model("qwen3")
+print(model.default_model_id)   # qwen3:latest
+for w in model.model_list:
+    print(w.id)                 # qwen3:latest, qwen3:4b, ...
+
+# Async usage
+import asyncio
+
+async def main():
+    result = await client.search_async("qwen3")
+    model  = await client.get_model_async("qwen3")
+
+asyncio.run(main())
+```

ollama_models-0.1.0/README.md

@@ -0,0 +1,41 @@
+# ollama-models (Python)
+
+Python client for searching and listing models from the [Ollama](https://ollama.com) registry.
+
+## Installation
+
+```bash
+pip install ollama-models
+```
+
+## Usage
+
+```python
+from ollama_models import OllamaModelsClient
+
+# No base URL needed — defaults to the official hosted instance
+client = OllamaModelsClient()
+
+# Pass a base URL only if you self-host the API
+# client = OllamaModelsClient("https://your-own-instance.workers.dev")
+
+# Search models
+result = client.search("qwen3", page=1)
+for page in result.pages:
+    print(page.http_url)
+
+# Get all tags for a model
+model = client.get_model("qwen3")
+print(model.default_model_id)   # qwen3:latest
+for w in model.model_list:
+    print(w.id)                 # qwen3:latest, qwen3:4b, ...
+
+# Async usage
+import asyncio
+
+async def main():
+    result = await client.search_async("qwen3")
+    model  = await client.get_model_async("qwen3")
+
+asyncio.run(main())
+```

ollama_models-0.1.0/pyproject.toml

@@ -0,0 +1,51 @@
+[project]
+name = "ollama-models"
+version = "0.1.0"
+description = "Python client for the ollama-models API — search and list Ollama model weights"
+authors = [
+    { name = "devcomfort", email = "im@devcomfort.me" }
+]
+license = { text = "MIT" }
+readme = "README.md"
+requires-python = ">= 3.8"
+keywords = ["ollama", "llm", "models", "search", "registry"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = [
+    "httpx>=0.27.0",
+]
+
+[project.urls]
+Repository = "https://github.com/devcomfort/ollama-models"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.rye]
+managed = true
+dev-dependencies = [
+    "pytest>=8.0.0",
+    "pytest-asyncio>=0.23.0",
+    "pytest-httpx>=0.21.0",
+    "build>=1.0.0",
+]
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/ollama_models"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"

ollama_models-0.1.0/requirements-dev.lock

@@ -0,0 +1,47 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+#   generate-hashes: false
+#   universal: false
+
+-e file:.
+anyio==4.13.0
+    # via httpx
+build==1.4.2
+certifi==2026.2.25
+    # via httpcore
+    # via httpx
+h11==0.16.0
+    # via httpcore
+httpcore==1.0.9
+    # via httpx
+httpx==0.28.1
+    # via ollama-models
+    # via pytest-httpx
+idna==3.11
+    # via anyio
+    # via httpx
+iniconfig==2.3.0
+    # via pytest
+packaging==26.0
+    # via build
+    # via pytest
+pluggy==1.6.0
+    # via pytest
+pygments==2.19.2
+    # via pytest
+pyproject-hooks==1.2.0
+    # via build
+pytest==9.0.2
+    # via pytest-asyncio
+    # via pytest-httpx
+pytest-asyncio==1.3.0
+pytest-httpx==0.36.0
+typing-extensions==4.15.0
+    # via anyio
+    # via pytest-asyncio

ollama_models-0.1.0/requirements.lock

@@ -0,0 +1,28 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+#   generate-hashes: false
+#   universal: false
+
+-e file:.
+anyio==4.13.0
+    # via httpx
+certifi==2026.2.25
+    # via httpcore
+    # via httpx
+h11==0.16.0
+    # via httpcore
+httpcore==1.0.9
+    # via httpx
+httpx==0.28.1
+    # via ollama-models
+idna==3.11
+    # via anyio
+    # via httpx
+typing-extensions==4.15.0
+    # via anyio

ollama_models-0.1.0/src/ollama_models/__init__.py

@@ -0,0 +1,10 @@
+from .client import OllamaModelsClient
+from .types import ModelList, ModelPage, ModelWeight, SearchResult
+
+__all__ = [
+    "OllamaModelsClient",
+    "SearchResult",
+    "ModelPage",
+    "ModelList",
+    "ModelWeight",
+]

ollama_models-0.1.0/src/ollama_models/client.py

@@ -0,0 +1,92 @@
+from __future__ import annotations
+
+from typing import Dict
+
+import httpx
+
+from .types import ModelList, ModelPage, ModelWeight, SearchResult
+
+DEFAULT_BASE_URL = "https://ollama-models-api.devcomfort.workers.dev"
+
+
+class OllamaModelsClient:
+    """Sync/async client for the ollama-models Cloudflare Workers API.
+
+    Usage (sync)::
+
+        client = OllamaModelsClient()
+        result = client.search("qwen3", page=1)
+        model  = client.get_model("qwen3")
+
+    Usage (async)::
+
+        result = await client.search_async("qwen3", page=1)
+        model  = await client.get_model_async("qwen3")
+    """
+
+    def __init__(self, base_url: str = DEFAULT_BASE_URL) -> None:
+        self._base_url = base_url.rstrip("/")
+
+    # ------------------------------------------------------------------
+    # Search
+    # ------------------------------------------------------------------
+
+    def search(self, keyword: str = "", page: int = 1) -> SearchResult:
+        params: Dict[str, str] = {"page": str(page)}
+        if keyword:
+            params["q"] = keyword
+        with httpx.Client() as client:
+            res = client.get(f"{self._base_url}/search", params=params)
+            res.raise_for_status()
+            data = res.json()
+        return _parse_search_result(data)
+
+    async def search_async(self, keyword: str = "", page: int = 1) -> SearchResult:
+        params: Dict[str, str] = {"page": str(page)}
+        if keyword:
+            params["q"] = keyword
+        async with httpx.AsyncClient() as client:
+            res = await client.get(f"{self._base_url}/search", params=params)
+            res.raise_for_status()
+            data = res.json()
+        return _parse_search_result(data)
+
+    # ------------------------------------------------------------------
+    # Model
+    # ------------------------------------------------------------------
+
+    def get_model(self, name: str) -> ModelList:
+        with httpx.Client() as client:
+            res = client.get(f"{self._base_url}/model", params={"name": name})
+            res.raise_for_status()
+            data = res.json()
+        return _parse_model_list(data)
+
+    async def get_model_async(self, name: str) -> ModelList:
+        async with httpx.AsyncClient() as client:
+            res = await client.get(f"{self._base_url}/model", params={"name": name})
+            res.raise_for_status()
+            data = res.json()
+        return _parse_model_list(data)
+
+
+# ------------------------------------------------------------------
+# Internal parsers
+# ------------------------------------------------------------------
+
+
+def _parse_search_result(data: dict) -> SearchResult:
+    return SearchResult(
+        pages=[ModelPage(http_url=p["http_url"]) for p in data["pages"]],
+        page_id=int(data["page_id"]),
+        keyword=str(data["keyword"]),
+    )
+
+
+def _parse_model_list(data: dict) -> ModelList:
+    return ModelList(
+        model_list=[
+            ModelWeight(http_url=w["http_url"], id=w["id"]) for w in data["model_list"]
+        ],
+        default_model_id=str(data["default_model_id"]),
+    )

ollama_models-0.1.0/src/ollama_models/types.py

@@ -0,0 +1,26 @@
+from dataclasses import dataclass
+from typing import List
+
+
+@dataclass
+class ModelPage:
+    http_url: str
+
+
+@dataclass
+class SearchResult:
+    pages: List[ModelPage]
+    page_id: int
+    keyword: str
+
+
+@dataclass
+class ModelWeight:
+    http_url: str
+    id: str
+
+
+@dataclass
+class ModelList:
+    model_list: List[ModelWeight]
+    default_model_id: str

ollama_models-0.1.0/tests/test_client.py

@@ -0,0 +1,160 @@
+from __future__ import annotations
+
+import pytest
+import httpx
+from pytest_httpx import HTTPXMock
+
+from ollama_models import OllamaModelsClient
+from ollama_models.client import DEFAULT_BASE_URL
+
+# ─── fixtures ─────────────────────────────────────────────────────────────────
+
+MOCK_SEARCH = {
+    "pages": [
+        {"http_url": "https://ollama.com/library/qwen3"},
+        {"http_url": "https://ollama.com/library/mistral"},
+    ],
+    "page_id": 1,
+    "keyword": "qwen3",
+}
+
+MOCK_MODEL = {
+    "model_list": [
+        {"http_url": "https://ollama.com/library/qwen3", "id": "qwen3:latest"},
+        {"http_url": "https://ollama.com/library/qwen3", "id": "qwen3:4b"},
+    ],
+    "default_model_id": "qwen3:latest",
+}
+
+
+# ─── DEFAULT_BASE_URL ─────────────────────────────────────────────────────────
+
+
+def test_default_base_url_is_official_instance():
+    assert DEFAULT_BASE_URL == "https://ollama-models-api.devcomfort.workers.dev"
+
+
+def test_client_uses_default_base_url(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    OllamaModelsClient().search("qwen3")
+    request = httpx_mock.get_requests()[0]
+    assert DEFAULT_BASE_URL in str(request.url)
+
+
+# ─── search() (sync) ──────────────────────────────────────────────────────────
+
+
+def test_search_returns_search_result(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    result = OllamaModelsClient().search("qwen3", page=1)
+    assert result.keyword == "qwen3"
+    assert result.page_id == 1
+    assert len(result.pages) == 2
+    assert result.pages[0].http_url == "https://ollama.com/library/qwen3"
+
+
+def test_search_sends_keyword_and_page_params(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    OllamaModelsClient().search("mistral", page=2)
+    request = httpx_mock.get_requests()[0]
+    assert "q=mistral" in str(request.url)
+    assert "page=2" in str(request.url)
+
+
+def test_search_omits_q_param_when_keyword_is_empty(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    OllamaModelsClient().search("", page=1)
+    request = httpx_mock.get_requests()[0]
+    assert "q=" not in str(request.url)
+
+
+def test_search_hits_search_endpoint(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    OllamaModelsClient().search("qwen3")
+    request = httpx_mock.get_requests()[0]
+    assert "/search" in str(request.url)
+
+
+def test_search_raises_on_http_error(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(status_code=500)
+    with pytest.raises(httpx.HTTPStatusError):
+        OllamaModelsClient().search("qwen3")
+
+
+# ─── search_async() ───────────────────────────────────────────────────────────
+
+
+async def test_search_async_returns_search_result(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    result = await OllamaModelsClient().search_async("qwen3", page=1)
+    assert result.keyword == "qwen3"
+    assert len(result.pages) == 2
+
+
+async def test_search_async_sends_keyword_and_page_params(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_SEARCH)
+    await OllamaModelsClient().search_async("qwen3", page=3)
+    request = httpx_mock.get_requests()[0]
+    assert "q=qwen3" in str(request.url)
+    assert "page=3" in str(request.url)
+
+
+async def test_search_async_raises_on_http_error(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(status_code=503)
+    with pytest.raises(httpx.HTTPStatusError):
+        await OllamaModelsClient().search_async("qwen3")
+
+
+# ─── get_model() (sync) ───────────────────────────────────────────────────────
+
+
+def test_get_model_returns_model_list(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_MODEL)
+    result = OllamaModelsClient().get_model("qwen3")
+    assert result.default_model_id == "qwen3:latest"
+    assert len(result.model_list) == 2
+    assert result.model_list[0].id == "qwen3:latest"
+    assert result.model_list[1].id == "qwen3:4b"
+
+
+def test_get_model_sends_name_param(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_MODEL)
+    OllamaModelsClient().get_model("qwen3")
+    request = httpx_mock.get_requests()[0]
+    assert "name=qwen3" in str(request.url)
+
+
+def test_get_model_hits_model_endpoint(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_MODEL)
+    OllamaModelsClient().get_model("qwen3")
+    request = httpx_mock.get_requests()[0]
+    assert "/model" in str(request.url)
+
+
+def test_get_model_raises_on_http_error(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(status_code=404)
+    with pytest.raises(httpx.HTTPStatusError):
+        OllamaModelsClient().get_model("nonexistent")
+
+
+# ─── get_model_async() ────────────────────────────────────────────────────────
+
+
+async def test_get_model_async_returns_model_list(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_MODEL)
+    result = await OllamaModelsClient().get_model_async("qwen3")
+    assert result.default_model_id == "qwen3:latest"
+    assert len(result.model_list) == 2
+
+
+async def test_get_model_async_sends_name_param(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(json=MOCK_MODEL)
+    await OllamaModelsClient().get_model_async("mistral")
+    request = httpx_mock.get_requests()[0]
+    assert "name=mistral" in str(request.url)
+
+
+async def test_get_model_async_raises_on_http_error(httpx_mock: HTTPXMock):
+    httpx_mock.add_response(status_code=404)
+    with pytest.raises(httpx.HTTPStatusError):
+        await OllamaModelsClient().get_model_async("nonexistent")

ollama_models-0.1.0/tests/test_types.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+from ollama_models.types import ModelList, ModelPage, ModelWeight, SearchResult
+
+
+def test_model_page_stores_url():
+    page = ModelPage(http_url="https://ollama.com/library/qwen3")
+    assert page.http_url == "https://ollama.com/library/qwen3"
+
+
+def test_search_result_stores_fields():
+    result = SearchResult(
+        pages=[ModelPage(http_url="https://ollama.com/library/qwen3")],
+        page_id=2,
+        keyword="qwen3",
+    )
+    assert result.page_id == 2
+    assert result.keyword == "qwen3"
+    assert len(result.pages) == 1
+    assert result.pages[0].http_url == "https://ollama.com/library/qwen3"
+
+
+def test_model_weight_stores_fields():
+    weight = ModelWeight(
+        http_url="https://ollama.com/library/qwen3",
+        id="qwen3:4b",
+    )
+    assert weight.http_url == "https://ollama.com/library/qwen3"
+    assert weight.id == "qwen3:4b"
+
+
+def test_model_list_stores_fields():
+    model_list = ModelList(
+        model_list=[
+            ModelWeight(http_url="https://ollama.com/library/qwen3", id="qwen3:latest"),
+            ModelWeight(http_url="https://ollama.com/library/qwen3", id="qwen3:4b"),
+        ],
+        default_model_id="qwen3:latest",
+    )
+    assert model_list.default_model_id == "qwen3:latest"
+    assert len(model_list.model_list) == 2
+    assert model_list.model_list[0].id == "qwen3:latest"
+    assert model_list.model_list[1].id == "qwen3:4b"