Merge pull request #14 from networkandcode/strands-newrelic-demo

Strands newrelic demo

Author: networkandcode

strands-examples/strands-newrelic-demo/.env.example

@@ -0,0 +1,6 @@
+AWS_ACCESS_KEY_ID=""
+AWS_SECRET_ACCESS_KEY=""
+AWS_DEFAULT_REGION=us-west-2
+OTEL_EXPORTER_OTLP_ENDPOINT="https://otlp.nr-data.net"
+OTEL_EXPORTER_OTLP_HEADERS=api-key=
+OTEL_EXPORTER_OTLP_PROTOCOL=http/protobuf

strands-examples/strands-newrelic-demo/.python-version

@@ -0,0 +1 @@
+3.12

strands-examples/strands-newrelic-demo/README.md

@@ -0,0 +1,66 @@
+import re
+
+from mcp import stdio_client, StdioServerParameters
+from strands import Agent
+from strands.tools.mcp import MCPClient
+import streamlit as st
+
+from set_telemetry import set_telemetry
+import uuid
+
+def remove_html_tags(text_with_html):
+    text_with_out_html = re.sub(r"<[^>]+>", "", text_with_html)
+    return text_with_out_html
+
+async def stream_result(stream):
+    result = ""
+    placeholder = st.empty()
+    async for chunk in stream:
+        if 'data' in chunk:
+            result += chunk['data']
+            result = remove_html_tags(result)
+            placeholder.write(result)
+
+
+class KubernetesMCPAgent:
+    def __init__(self):
+        set_telemetry()
+        server_params = {
+            "command": "npx",
+            "args": [
+                "-y",
+                "kubernetes-mcp-server@latest"
+            ],
+            "env": {
+                "KUBECONFIG": "k3s.yaml"
+            }
+        }
+        
+        self.stdio_mcp_client = MCPClient(lambda: stdio_client(
+            StdioServerParameters(
+                **server_params
+            )
+        ))
+
+        with self.stdio_mcp_client:
+            # Get the tools from the MCP server
+            tools = self.stdio_mcp_client.list_tools_sync()
+
+            # Create an agent with these tools
+            self.agent = Agent(
+                callback_handler=None,
+                model="us.amazon.nova-micro-v1:0",
+                tools=tools,
+                trace_attributes={
+                    "session.id": str(uuid.uuid4()),
+                },
+            )
+
+    
+    async def send_prompt(self, prompt):
+        with self.stdio_mcp_client:
+            stream = self.agent.stream_async(prompt)
+            await stream_result(stream)
+
+
+kubernetes_mcp_agent = KubernetesMCPAgent()

strands-examples/strands-newrelic-demo/k8s_mcp_agent.py

@@ -0,0 +1,16 @@
+import asyncio
+
+import streamlit as st
+
+from k8s_mcp_agent import kubernetes_mcp_agent
+
+async def main():
+    st.title("Kubernetes MCP App")
+
+    if prompt := st.chat_input("Chat with the Kubernetes API through MCP"):
+        with st.chat_message("user"):
+            st.write(prompt)
+        with st.chat_message("assistant"):
+            await kubernetes_mcp_agent.send_prompt(prompt)
+
+asyncio.run(main())

strands-examples/strands-newrelic-demo/k8s_mcp_app.py

@@ -0,0 +1,6 @@
+def main():
+    print("Hello from strands-langfuse-demo!")
+
+
+if __name__ == "__main__":
+    main()

strands-examples/strands-newrelic-demo/main.py

@@ -0,0 +1,10 @@
+[project]
+name = "strands-newrelic-demo"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "strands-agents[otel]>=1.7.1",
+    "streamlit>=1.49.1",
+]

strands-examples/strands-newrelic-demo/pyproject.toml

@@ -0,0 +1,7 @@
+from dotenv import load_dotenv
+from strands.telemetry import StrandsTelemetry
+
+def set_telemetry():
+    load_dotenv()
+    strands_telemetry = StrandsTelemetry()
+    strands_telemetry.setup_otlp_exporter()