From ff3f79e4680c1dcdc8ea9508db9f4091ce4f39f1 Mon Sep 17 00:00:00 2001 From: Brian Dev Date: Sat, 15 Mar 2025 01:07:15 +0700 Subject: [PATCH] Update test_ollama_pt --- tests/llm_translation/test_prompt_factory.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/llm_translation/test_prompt_factory.py b/tests/llm_translation/test_prompt_factory.py index 1844af2030..314aceac7d 100644 --- a/tests/llm_translation/test_prompt_factory.py +++ b/tests/llm_translation/test_prompt_factory.py @@ -752,4 +752,5 @@ def test_ollama_pt(): {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": "Hello!"}, ] - ollama_pt(model="ollama/llama3.1:8b", messages=messages) + prompt = ollama_pt(model="ollama/llama3.1", messages=messages) + print(prompt)