Merge pull request #525 from d1mashok/trace-logging-fix

Agent Trace logging - fix

Author: mttanke

agents-and-function-calling/bedrock-agents/features-examples/06-prompt-and-session-attributes/06-prompt-and-session-attributes.ipynb

@@ -903,6 +903,16 @@
    "outputs": [],
    "source": [
     "%%time\n",
+    "import json\n",
+    "from datetime import datetime\n",
+    "\n",
+    "class DateTimeEncoder(json.JSONEncoder):\n",
+    "    def default(self, obj):\n",
+    "        if isinstance(obj, datetime):\n",
+    "            return obj.isoformat()\n",
+    "        return super().default(obj)\n",
+    "\n",
+    "\n",
     "event_stream = agentResponse['completion']\n",
     "try:\n",
     "    for event in event_stream:        \n",
@@ -913,7 +923,7 @@
     "            end_event_received = True\n",
     "            # End event indicates that the request finished successfully\n",
     "        elif 'trace' in event:\n",
-    "            logger.info(json.dumps(event['trace'], indent=2))\n",
+    "            logger.info(json.dumps(event['trace'], indent=2, cls=DateTimeEncoder))\n",
     "        else:\n",
     "            raise Exception(\"unexpected event.\", event)\n",
     "except Exception as e:\n",