Update test_ollama_pt

2025-03-15 01:07:15 +07:00 · 2025-03-15 01:07:15 +07:00 · ff3f79e468
commit ff3f79e468
parent 51589364a1
1 changed files with 2 additions and 1 deletions
--- 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)