From e7bcbdb888e42d1068ae11d1bc4572be34f72744 Mon Sep 17 00:00:00 2001 From: "Th3.m3ss14h@gmail.com" <58686863+th3-m3ss14h@users.noreply.github.com> Date: Sat, 7 Jun 2025 16:46:15 -0400 Subject: [PATCH] Update llm.py --- src/llm/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llm/llm.py b/src/llm/llm.py index 24f20d59..55d210ae 100644 --- a/src/llm/llm.py +++ b/src/llm/llm.py @@ -68,7 +68,7 @@ def __init__(self, model_id: str = None): } if ollama.client: - self.models["OLLAMA"] = [(model["name"], model["name"]) for model in ollama.models] + self.models["OLLAMA"] = [(model.model, model.model) for model in ollama.models] def list_models(self) -> dict: return self.models