hotfix: Apply a temp hotfix for sessions.chat

agents-api/agents_api/models/entry/get_history.py

@@ -42,6 +42,9 @@
             }
             for r in d.pop("relations")
         ],
+        # TODO: Remove this once we sort the entries in the cozo query
+        # Sort entries by created_at
+        "entries": sorted(d.pop("entries"), key=lambda entry: entry["created_at"]),
         **d,
     },
 )

agents-api/agents_api/routers/sessions/chat.py

@@ -72,6 +72,17 @@ async def chat(
         # messages = messages[-settings["max_tokens"] :]
         raise NotImplementedError("Truncation is not yet implemented")
 
+    # FIXME: Hotfix for datetime not serializable. Needs investigation
+    messages = [
+        msg.model_dump() if hasattr(msg, "model_dump") else msg
+        for msg in messages
+    ]
+
+    messages = [
+        dict(role=m["role"], content=m["content"], user=m.get("user"))
+        for m in messages
+    ]
+
     # Get the response from the model
     model_response = await litellm.acompletion(
         messages=messages,
@@ -90,6 +101,12 @@ async def chat(
             for msg in new_messages
         ]
 
+        # Add the response to the new entries
+        new_entries.append(
+            CreateEntryRequest.from_model_input(
+                model=settings["model"], **model_response.choices[0].model_dump()['message'], source="api_response"
+            )
+        )
         background_tasks.add_task(
             create_entries,
             developer_id=developer.id,