fix: Update sessions with given IDs only

julep-ai · Apr 22, 2024 · a71ad5a · a71ad5a
1 parent 32ecc5a
commit a71ad5a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/agents-api/agents_api/models/session/patch_session.py b/agents-api/agents_api/models/session/patch_session.py
@@ -66,9 +66,11 @@ def patch_session_query(
     session_update_query = f"""
     {{
         input[{session_update_cols}] <- $session_update_vals
+        ids[session_id, developer_id] <- [[to_uuid($session_id), to_uuid($developer_id)]]
         
         ?[{all_fields}, metadata, updated_at] :=
             input[{session_update_cols}],
+            ids[session_id, developer_id],
             *sessions{{
                 {rest_fields}, metadata: md, @ "NOW"
             }},

diff --git a/agents-api/agents_api/models/session/update_session.py b/agents-api/agents_api/models/session/update_session.py
@@ -53,9 +53,11 @@ def update_session_query(
     session_update_query = f"""
     {{
         input[{session_update_cols}] <- $session_update_vals
+        ids[session_id, developer_id] <- [[to_uuid($session_id), to_uuid($developer_id)]]
         
         ?[{all_fields}, updated_at] :=
             input[{session_update_cols}],
+            ids[session_id, developer_id],
             *sessions{{
                 {rest_fields}, @ "NOW"
             }},
@@ -69,6 +71,8 @@ def update_session_query(
     }}
     """
 
+    print(session_update_query)
+
     combined_query = "{" + assertion_query + "}" + session_update_query
 
     return (