Skip to content

trading_agent.doctor.checks.ollama_backend_check

Checks that an Ollama backend's server answers and has the model.

Typical usage example:

result = OllamaBackendCheck('gemma', backend_configuration).run()

OllamaBackendCheck 

OllamaBackendCheck(backend_name, backend_configuration)

Bases: DoctorCheck

Checks one backend of kind "ollama".

Creates the check.

Parameters:

Name Type Description Default
backend_name str

The backend's name in trading_agent.toml.

 required
backend_configuration BackendConfiguration

Its settings.

 required
Source code in src/trading_agent/doctor/checks/ollama_backend_check.py 
19
20
21
22
23
24
25
26
27
28
29
30
31
def __init__(
    self,
    backend_name: str,
    backend_configuration: agent_configuration.BackendConfiguration,
):
    """Creates the check.

    Args:
        backend_name (str): The backend's name in trading_agent.toml.
        backend_configuration (agent_configuration.BackendConfiguration): Its settings.
    """
    self.name = f'Backend {backend_name} (Ollama)'
    self._backend_configuration = backend_configuration

run 

run()

Lists Ollama's installed models and looks for the configured one.

Returns:

Type Description
CheckResult

OK when the model is installed, failed when Ollama is unreachable or lacks the model.
Source code in src/trading_agent/doctor/checks/ollama_backend_check.py 
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
def run(self) -> doctor_check.CheckResult:
    """Lists Ollama's installed models and looks for the configured one.

    Returns:
        doctor_check.CheckResult: OK when the model is installed, failed when Ollama is unreachable or lacks the model.
    """
    base_url = self._backend_configuration.base_url
    model = self._backend_configuration.model
    try:
        response = httpx.get(
            f'{base_url}/api/tags', timeout=_TIMEOUT_SECONDS
        )
        response.raise_for_status()
    except httpx.HTTPError as error:
        return doctor_check.CheckResult(
            doctor_check.CheckStatus.FAILED,
            f'Ollama at {base_url} did not answer: {error!r}',
        )
    installed_models = []
    for entry in response.json().get('models', []):
        installed_models.append(entry.get('name'))
    if model not in installed_models:
        return doctor_check.CheckResult(
            doctor_check.CheckStatus.FAILED,
            f'Model {model} is not installed; run: ollama pull {model}',
        )
    return doctor_check.CheckResult(
        doctor_check.CheckStatus.OK,
        f'{model} is installed at {base_url}.',
    )