Add GenAI tool definitions to sample spans

Author: nagkumar91

lab/5-Observability/01-trace-agent-session.ipynb

@@ -77,6 +77,27 @@
     "    \"price\": 24.0,\n",
     "    \"available_stock\": 67\n",
     "}\n",
+    "tool_definitions = [\n",
+    "    {\n",
+    "        \"type\": \"function\",\n",
+    "        \"name\": \"product_search\",\n",
+    "        \"description\": \"Searches the catalog for matching Zava retail products.\",\n",
+    "        \"parameters\": {\n",
+    "            \"type\": \"object\",\n",
+    "            \"properties\": {\n",
+    "                \"sku\": {\n",
+    "                    \"type\": \"string\",\n",
+    "                    \"description\": \"Unique product SKU to search for.\"\n",
+    "                },\n",
+    "                \"query\": {\n",
+    "                    \"type\": \"string\",\n",
+    "                    \"description\": \"Free-form text describing the desired product.\"\n",
+    "                }\n",
+    "            },\n",
+    "            \"required\": [\"sku\"]\n",
+    "        }\n",
+    "    }\n",
+    "]\n",
     "input_messages = [\n",
     "    {\n",
     "        \"role\": \"system\",\n",
@@ -130,6 +151,7 @@
     "    span.set_attribute(\"gen_ai.agent.description\", \"Cora retail assistant for Zava DIY customers.\")\n",
     "    span.set_attribute(\"gen_ai.request.model\", \"gpt-4o-mini\")\n",
     "    span.set_attribute(\"gen_ai.system_instructions\", system_prompt)\n",
+    "    span.set_attribute(\"gen_ai.tool.definitions\", json.dumps(tool_definitions, ensure_ascii=False))\n",
     "    span.set_attribute(\"server.address\", \"cora-agents.eastus2.inference.ai.azure.com\")\n",
     "    span.set_attribute(\"server.port\", 443)\n",
     "    span.set_attribute(\"azure.resource_provider.namespace\", \"Microsoft.CognitiveServices\")\n",

lab/5-Observability/02-collect-span-snapshots.ipynb

@@ -71,6 +71,27 @@
     "        \"name\": \"Zava SwiftCoat Satin\",\n",
     "        \"inventory\": 142\n",
     "    }\n",
+    "    tool_definitions = [\n",
+    "        {\n",
+    "            \"type\": \"function\",\n",
+            "            \"name\": \"inventory_lookup\",\n",
+    "            \"description\": \"Fetches inventory details for a specific product category and finish.\",\n",
+    "            \"parameters\": {\n",
+    "                \"type\": \"object\",\n",
+    "                \"properties\": {\n",
+    "                    \"product_type\": {\n",
+    "                        \"type\": \"string\",\n",
+    "                        \"description\": \"Product category name to search.\"\n",
+    "                    },\n",
+    "                    \"finish\": {\n",
+    "                        \"type\": \"string\",\n",
+    "                        \"description\": \"Desired paint finish (e.g., satin, matte).\"\n",
+    "                    }\n",
+    "                },\n",
+    "                \"required\": [\"product_type\"]\n",
+    "            }\n",
+    "        }\n",
+    "    ]\n",
     "    input_messages = [\n",
     "        {\n",
     "            \"role\": \"system\",\n",
@@ -133,6 +154,7 @@
     "        span.set_attribute(\"gen_ai.usage.output_tokens\", 41)\n",
     "        span.set_attribute(\"gen_ai.conversation.id\", conversation_id)\n",
     "        span.set_attribute(\"gen_ai.system_instructions\", system_prompt)\n",
+        "        span.set_attribute(\"gen_ai.tool.definitions\", json.dumps(tool_definitions, ensure_ascii=False))\n",
         "        span.set_attribute(\"gen_ai.input.messages\", json.dumps(input_messages, ensure_ascii=False))\n",
         "        span.set_attribute(\"gen_ai.output.messages\", json.dumps(output_messages, ensure_ascii=False))\n",
     "        span.set_attribute(\"server.address\", \"cora-agents.eastus2.inference.ai.azure.com\")\n",