Add test for generating text with different providers

This commit is contained in:
2024-10-31 16:22:05 -04:00
parent 25ee4ae32c
commit c41a3f00fb
+23
View File
@@ -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