diff --git a/tests/test_generate_text.py b/tests/test_generate_text.py new file mode 100644 index 0000000..55a9a13 --- /dev/null +++ b/tests/test_generate_text.py @@ -0,0 +1,23 @@ +import pytest + +from simplemind.providers import Anthropic, Gemini, OpenAI, Groq, Ollama + + +@pytest.mark.parametrize( + "provider_cls", + [ + Anthropic, + Gemini, + OpenAI, + Groq, + Ollama, + ], +) +def test_generate_text(provider_cls): + provider = provider_cls() + prompt = "What is 2+2?" + + response = provider.generate_text(prompt=prompt, llm_model=provider.DEFAULT_MODEL) + + assert isinstance(response, str) + assert len(response) > 0