Skip to content

Commit

Permalink
anthropic[patch]: remove retired model from tests (#27965)
Browse files Browse the repository at this point in the history
`claude-instant` was [retired
yesterday](https://docs.anthropic.com/en/docs/resources/model-deprecations).
  • Loading branch information
ccurme authored Nov 7, 2024
1 parent 2cb3927 commit a747dbd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions libs/partners/anthropic/tests/integration_tests/test_llms.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit a747dbd

Please sign in to comment.