feat: Updated agents-api/agents_api/models/session

julep-ai · Apr 13, 2024 · 5ebfd60 · 5ebfd60
1 parent ddd39cc
commit 5ebfd60
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/agents-api/agents_api/models/session/update_session.py b/agents-api/agents_api/models/session/update_session.py
@@ -21,12 +21,14 @@ def update_session_query(
     developer_id: UUID,
     **update_data,
 ) -> pd.DataFrame:
+    # Process the update data to prepare it for the query.
     session_update_cols, session_update_vals = cozo_process_mutate_data(
         {
             **{k: v for k, v in update_data.items() if v is not None},
         }
     )
 
+        # Prepare lists of columns for the query.
     session_update_cols_lst = session_update_cols.split(",")
     all_fields_lst = list(set(session_update_cols_lst).union(set(_fields)))
     all_fields = ", ".join(all_fields_lst)
@@ -37,6 +39,7 @@ def update_session_query(
         )
     )
 
+    # Construct the datalog query for updating session information.
     session_update_query = f"""
     {{
         input[{session_update_cols}] <- $session_update_vals