Skip to content

Commit bb46c02

Browse files
committed
fix: flaky cohere test
1 parent 82d40d6 commit bb46c02

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

projects/extension/tests/test_cohere.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,7 @@ def test_cohere_rerank_simple(cur_with_api_key):
365365

366366
def test_cohere_chat_complete(cur_with_api_key):
367367
actual = None
368-
expected = """According to a tongue-twister that originated in the United States, a woodchuck would chuck "as much wood as a woodchuck would, if a woodchuck could chuck wood." This phrase is meant to be difficult to say quickly and is often used to challenge someone\'s ability to speak clearly and accurately.\n\nThe amount of wood a woodchuck could actually chuck is unclear and somewhat subjective, as it depends on several factors such as the size of the woodchuck, the method of chucking, and the duration of said chucking. Woodchucks, also known as groundhogs, are capable of moving and stacking wood, but the notion of them chucking wood is more of a humorous idea than a practical reality."""
368+
expected = """According to a tongue-twister that originated in the United States, a woodchuck would chuck "as much wood as a woodchuck would, if a woodchuck could chuck wood."\n\nThis phrase is meant to be difficult to say quickly and can be fun to practice with family and friends!"""
369369
# apparently cohere doesn't always use the seed that we tell it to, so give it a few chances
370370
for _ in range(5):
371371
cur_with_api_key.execute("""

0 commit comments

Comments
 (0)