Merge branch 'ag2ai:main' into neo4j_rag_update

ag2ai · Dec 13, 2024 · 926ed4e · 926ed4e
2 parents a1510fb + 18a5e7f
commit 926ed4e
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/notebook/agentchat_swarm_graphrag_telemetry_trip_planner.ipynb b/notebook/agentchat_swarm_graphrag_telemetry_trip_planner.ipynb
@@ -10,8 +10,8 @@
     "\n",
     "The following diagram outlines the key components of the Swarm, with highlights being:\n",
     "\n",
+    "- **Arize Phoenix to provide transparency using the OpenTelemetry standard**\n",
     "- FalkorDB agent using a GraphRAG database of restaurants and attractions\n",
-    "- Arize Phoenix to provide transparency using the OpenTelemetry standard\n",
     "- Structured Output agent that will enforce a strict format for the accepted itinerary\n",
     "- Routing agent that utilises the Google Maps API to calculate distances between activites\n",
     "- Swarm orchestration utilising context variables"
@@ -42,9 +42,7 @@
    "source": [
     "## Install Docker Containers\n",
     "\n",
-    "**Note:** This likely requires a docker compose file to get it to work reliably.  It didn't require it on my system, but plan to come back to it at some point.\n",
-    "\n",
-    "For now I'm more interested in instrumenting GraphRAG solutions to get better visibility to key interations with the LLM -- in particular for entity and link detection."
+    "**Note:** This may require a Docker Compose file to get it to work reliably."
    ]
   },
   {
@@ -71,10 +69,7 @@
     "import subprocess\n",
     "\n",
     "# Run the Docker container without interactive mode\n",
-    "subprocess.Popen([\n",
-    "    \"docker\", \"run\", \"-p\", \"6379:6379\", \"-p\", \"3000:3000\",\n",
-    "    \"--rm\", \"falkordb/falkordb:latest\"\n",
-    "])"
+    "subprocess.Popen([\"docker\", \"run\", \"-p\", \"6379:6379\", \"-p\", \"3000:3000\", \"--rm\", \"falkordb/falkordb:latest\"])"
    ]
   },
   {
@@ -98,10 +93,7 @@
     "import subprocess\n",
     "\n",
     "# Run the Docker container without interactive mode\n",
-    "subprocess.Popen([\n",
-    "    \"docker\", \"run\", \"-p\", \"6006:6006\", \"-p\", \"4317:4317\",\n",
-    "    \"--rm\", \"arizephoenix/phoenix:latest\"\n",
-    "])"
+    "subprocess.Popen([\"docker\", \"run\", \"-p\", \"6006:6006\", \"-p\", \"4317:4317\", \"--rm\", \"arizephoenix/phoenix:latest\"])"
    ]
   },
   {
@@ -130,8 +122,8 @@
     "\n",
     "# defaults to endpoint=\"http://localhost:4317\"\n",
     "tracer_provider = register(\n",
-    "  project_name=\"ag2-swarm-graphrag\", # Default is 'default'\n",
-    "  endpoint=\"http://localhost:4317\",  # Sends traces using gRPC\n",
+    "    project_name=\"ag2-swarm-graphrag\",  # Default is 'default'\n",
+    "    endpoint=\"http://localhost:4317\",  # Sends traces using gRPC\n",
     ")"
    ]
   },