Improve query logging

Author: labrenbe

stacks/opensearch-rag/opensearch-rag.ipynb

@@ -452,14 +452,12 @@
     "    \n",
     "    # Log the query if requested (with truncated vector for readability)\n",
     "    if log_query:\n",
-    "        query_json = json.dumps(search_body, indent=2)\n",
-    "        # Replace full vector with truncated version: first 3 ... last 3\n",
-    "        vector_str = json.dumps(query_embedding)\n",
-    "        truncated = f\"[{query_embedding[0]:.4f}, {query_embedding[1]:.4f}, {query_embedding[2]:.4f}, \\\"... ({len(query_embedding)} dims) ...\\\", {query_embedding[-3]:.4f}, {query_embedding[-2]:.4f}, {query_embedding[-1]:.4f}]\"\n",
-    "        query_json = query_json.replace(vector_str, truncated)\n",
-    "        \n",
+    "    # Create display version with truncated vector\n",
+    "        log_body = json.loads(json.dumps(search_body))  # Deep copy\n",
+    "        vector = log_body['query']['knn']['embedding']['vector']\n",
+    "        log_body['query']['knn']['embedding']['vector'] = [vector[0], vector[1], vector[2], \"[...]\", vector[-3], vector[-2], vector[-1]]\n",
     "        print(\"OpenSearch Query:\")\n",
-    "        print(query_json)\n",
+    "        print(json.dumps(log_body, indent=2))\n",
     "        print()\n",
     "    \n",
     "    # Step 6: Execute search\n",