Spaces:
Sleeping
Sleeping
File size: 306 Bytes
cfd3735 |
1 2 3 4 5 6 7 8 9 10 11 |
"""Test Prediction Guard API wrapper."""
from langchain.llms.predictionguard import PredictionGuard
def test_predictionguard_call() -> None:
"""Test valid call to prediction guard."""
llm = PredictionGuard(name="default-text-gen")
output = llm("Say foo:")
assert isinstance(output, str)
|