From 12b7bc8ac64e5dd88cc60bde8b53e6c585c01c37 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Sirieix?= <clem.sirieix@gmail.com>
Date: Fri, 25 Oct 2024 10:11:44 +0200
Subject: [PATCH 1/2] feat: update prompt version definition

---
 literalai/api/gql.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/literalai/api/gql.py b/literalai/api/gql.py
index 7ad4713..1aab01e 100644
--- a/literalai/api/gql.py
+++ b/literalai/api/gql.py
@@ -1024,17 +1024,20 @@
 GET_PROMPT_VERSION = """
 query GetPrompt($id: String, $name: String, $version: Int) {
     promptVersion(id: $id, name: $name, version: $version) {
-        id
         createdAt
-        updatedAt
-        type
+        id
+        label
+        settings
+        status
+        tags
         templateMessages
         tools
-        settings
+        type
+        updatedAt
+        url
         variables
         variablesDefaultValues
         version
-        url
         lineage {
             name
         }

From 1bf029d3c9c10bbfdad82f6deffbee959eabec1a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Sirieix?= <clem.sirieix@gmail.com>
Date: Fri, 25 Oct 2024 18:17:58 +0200
Subject: [PATCH 2/2] fix: time sleep

---
 tests/e2e/test_e2e.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py
index 992c3a5..aede802 100644
--- a/tests/e2e/test_e2e.py
+++ b/tests/e2e/test_e2e.py
@@ -176,6 +176,7 @@ async def test_step(self, client: LiteralClient, async_client: AsyncLiteralClien
         await async_client.flush()
         is_deleted = await async_client.api.delete_step(id=step.id)
         assert is_deleted is True, "Step should be deleted"
+        time.sleep(1)
 
         deleted_step = await async_client.api.get_step(id=step.id)