test: simplify cohere tests (#563)

Author: smoya

projects/extension/tests/test_cohere.py

@@ -1,5 +1,4 @@
 import os
-from time import sleep
 
 import psycopg
 import pytest
@@ -364,26 +363,19 @@ def test_cohere_rerank_simple(cur_with_api_key):
 
 
 def test_cohere_chat_complete(cur_with_api_key):
-    actual = None
-    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!"""
-    # apparently cohere doesn't always use the seed that we tell it to, so give it a few chances
-    for _ in range(5):
-        cur_with_api_key.execute("""
-            select ai.cohere_chat_complete
-            ( 'command-r'
-            , jsonb_build_array
-              ( jsonb_build_object
-                ( 'role', 'user'
-                , 'content', 'How much wood would a woodchuck chuck if a woodchuck could chuck wood?'
-                )
-              )
-            , seed=>42
-            , temperature=>0.0
-            )->'message'->'content'->0->>'text'
-        """)
-        actual = cur_with_api_key.fetchone()[0]
-        if actual == expected:
-            return
-        else:
-            sleep(0.1)
-    assert actual == expected
+    # even with temperature 0.0 and seed 42, the response can vary, so we only check if it's not empty
+    cur_with_api_key.execute("""
+        select ai.cohere_chat_complete
+        ( 'command-r'
+        , jsonb_build_array
+          ( jsonb_build_object
+            ( 'role', 'user'
+            , 'content', 'How much wood would a woodchuck chuck if a woodchuck could chuck wood?'
+            )
+          )
+        , seed=>42
+        , temperature=>0.0
+        )->'message'->'content'->0->>'text'
+    """)
+    result = cur_with_api_key.fetchone()[0]
+    assert result