anthropic[patch]: remove retired model from tests (#27965)

`claude-instant` was [retired
yesterday](https://docs.anthropic.com/en/docs/resources/model-deprecations).
This commit is contained in:
ccurme 2024-11-07 11:16:29 -05:00 committed by GitHub
parent 2cb39270ec
commit a747dbd24b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -24,14 +24,14 @@ def test_anthropic_model_param() -> None:
def test_anthropic_call() -> None:
"""Test valid call to anthropic."""
llm = Anthropic(model="claude-instant-1") # type: ignore[call-arg]
llm = Anthropic(model="claude-2.1") # type: ignore[call-arg]
output = llm.invoke("Say foo:")
assert isinstance(output, str)
def test_anthropic_streaming() -> None:
"""Test streaming tokens from anthropic."""
llm = Anthropic(model="claude-instant-1") # type: ignore[call-arg]
llm = Anthropic(model="claude-2.1") # type: ignore[call-arg]
generator = llm.stream("I'm Pickle Rick")
assert isinstance(generator, Generator)