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 732d23b commit 918ab79
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 @@ -143,7 +143,7 @@ def test_stream_cancellation():
assert False
assert len(chunks) <= 6

@pytest.mark.flaky(reruns=3, reruns_delay=0)
@pytest.mark.flaky(reruns=5, reruns_delay=0)
def test_url_context():
url = "https://example.com/article"
chat = Chatlet()
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", urls=[url], temperature=0.0)
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)

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

0 comments on commit 918ab79

Please sign in to comment.