fix: use aws-opentelemetry-distro and add input/output logging for LangGraph agent (#552)

The LangGraph template was using the generic opentelemetry-distro which
doesn't integrate properly with AWS X-Ray and CloudWatch. This resulted
in only 1 span and no input/output logs in the runtime log group.

Switch to aws-opentelemetry-distro (matching the Strands template) to
enable proper AWS-native telemetry with botocore instrumentation that
captures Bedrock model calls, tool usage, and trace context. Also add
input/output logging to the agent entrypoint.

Fixes #551

Co-authored-by: Jesse Turner <ajesstur@amazon.com>

Author: tejaskash

src/assets/__tests__/__snapshots__/assets.snapshot.test.ts.snap

@@ -2904,12 +2904,15 @@ async def invoke(payload, context):
 
     # Process the user prompt
     prompt = payload.get("prompt", "What can you help me with?")
+    log.info(f"Agent input: {prompt}")
 
     # Run the agent
     result = await graph.ainvoke({"messages": [HumanMessage(content=prompt)]})
 
     # Return result
-    return {"result": result["messages"][-1].content}
+    output = result["messages"][-1].content
+    log.info(f"Agent output: {output}")
+    return {"result": output}
 
 
 if __name__ == "__main__":
@@ -3147,8 +3150,7 @@ description = "AgentCore Runtime Application using LangChain/LangGraph"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "opentelemetry-distro",
-    "opentelemetry-exporter-otlp",
+    "aws-opentelemetry-distro",
     "langgraph >= 1.0.2",
     "mcp >= 1.19.0",
     "langchain-mcp-adapters >= 0.1.11",

src/assets/python/http/langchain_langgraph/base/main.py

@@ -54,12 +54,15 @@ async def invoke(payload, context):
 
     # Process the user prompt
     prompt = payload.get("prompt", "What can you help me with?")
+    log.info(f"Agent input: {prompt}")
 
     # Run the agent
     result = await graph.ainvoke({"messages": [HumanMessage(content=prompt)]})
 
     # Return result
-    return {"result": result["messages"][-1].content}
+    output = result["messages"][-1].content
+    log.info(f"Agent output: {output}")
+    return {"result": output}
 
 
 if __name__ == "__main__":

src/assets/python/http/langchain_langgraph/base/pyproject.toml

@@ -9,8 +9,7 @@ description = "AgentCore Runtime Application using LangChain/LangGraph"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "opentelemetry-distro",
-    "opentelemetry-exporter-otlp",
+    "aws-opentelemetry-distro",
     "langgraph >= 1.0.2",
     "mcp >= 1.19.0",
     "langchain-mcp-adapters >= 0.1.11",