Move from test to supported claude-instant-1 model (#9066)

Moves from "test" model to "claude-instant-1" model which is supported
and has actual capacity
This commit is contained in:
Mike Lambert 2023-08-10 14:57:28 -04:00 committed by GitHub
parent b7bc8ec87f
commit e94a5d753f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -11,14 +11,14 @@ from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler
def test_anthropic_call() -> None:
"""Test valid call to anthropic."""
llm = Anthropic(model="test")
llm = Anthropic(model="claude-instant-1")
output = llm("Say foo:")
assert isinstance(output, str)
def test_anthropic_streaming() -> None:
"""Test streaming tokens from anthropic."""
llm = Anthropic(model="test")
llm = Anthropic(model="claude-instant-1")
generator = llm.stream("I'm Pickle Rick")
assert isinstance(generator, Generator)