fix idk

Helicone · May 16, 2024 · 0b7d9ff · 0b7d9ff
1 parent 04e3f4a
commit 0b7d9ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/helicone-python/tests/test_async.py b/helicone-python/tests/test_async.py
@@ -6,6 +6,7 @@
 import requests
 from dotenv import load_dotenv
 from psycopg2.extras import DictCursor
+import json
 
 from helicone.globals import helicone_global
 from helicone.globals.helicone import helicone_global
@@ -89,8 +90,7 @@ def test_openai_async():
 
     time.sleep(3)  # Give some time for the async logging to complete
 
-    query = "SELECT * FROM properties INNER JOIN request ON properties.request_id = request.id WHERE key = 'requestid' AND value = %s LIMIT 1"
-    request_data = fetch_from_db(query, (requestId,))
+    request_data = fetch_from_db("SELECT * FROM public.request WHERE properties @> %s", (json.dumps({"requestid": requestId}),))
     assert request_data, "Request data not found in the database for the given property request id"
 
     latest_request = request_data[0]