From 9aee5f372c97cc0d5e09736a8ced1dd0dcd44872 Mon Sep 17 00:00:00 2001 From: Adrian Cole Date: Mon, 23 Sep 2024 14:21:31 +1000 Subject: [PATCH] drift Signed-off-by: Adrian Cole --- tests/providers/openai/test_ollama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/providers/openai/test_ollama.py b/tests/providers/openai/test_ollama.py index a432d0c..0aa4646 100644 --- a/tests/providers/openai/test_ollama.py +++ b/tests/providers/openai/test_ollama.py @@ -1,5 +1,6 @@ from typing import Tuple +import os import pytest from exchange import Text @@ -26,7 +27,7 @@ def test_ollama_completion_integration(): def ollama_complete() -> Tuple[Message, Usage]: provider = OllamaProvider.from_env() - model = OLLAMA_MODEL + model = os.getenv("OLLAMA_MODEL", OLLAMA_MODEL) system = "You are a helpful assistant." messages = [Message.user("Hello")] return provider.complete(model=model, system=system, messages=messages, tools=None)