Skip to content

Commit

Permalink
auto model pricing
Browse files Browse the repository at this point in the history
  • Loading branch information
slava-vishnyakov committed Aug 6, 2024
1 parent 918ab79 commit b056833
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/chatlet/test_chatlet.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ def test_stream_cancellation():
@pytest.mark.flaky(reruns=5, reruns_delay=0)
def test_url_context():
url = "https://example.com/article"
chat = Chatlet()
chat = Chatlet(model="anthropic/claude-3-haiku")
with responses.RequestsMock(assert_all_requests_are_fired=False) as rsps:
rsps.add(responses.GET, url,
body="# title: Test Article\n\ncontent: This is a test article content.",
Expand All @@ -155,7 +155,7 @@ def test_url_context():

rsps.add_passthru('https://')

response = chat("Write the title and exact content I provided from text above (no need to go to internet! just copy the text)", urls=[url], temperature=0.0)
response = chat("Write the title and exact content I provided from text above", urls=[url], temperature=0.0)

assert "Test Article" in response.lower() or "test article" in response.lower()
assert "content" in response.lower()
Expand Down

0 comments on commit b056833

Please sign in to comment.