fix: align run_code with AWS docs pattern (invoke+executeCode, not execute_code wrapper)

agentclaw/app/agent_claw_main/tools/code_interpreter.py

@@ -1,68 +1,42 @@
-"""Code interpreter tool — runs Python code in AgentCore managed sandbox."""
+"""Code interpreter tool — runs Python code in AgentCore managed sandbox.
+
+Follows the AWS-recommended pattern:
+  https://docs.aws.amazon.com/bedrock-agentcore/latest/devguide/code-interpreter-building-agents.html
+"""
+import json
 import os
-import base64
 from strands import tool
-
+from bedrock_agentcore.tools.code_interpreter_client import code_session
-
-def _parse_stream(result: dict) -> str:
-    """Parse the streaming response from invoke_code_interpreter."""
-    parts = []
-    if "stream" not in result:
-        return str(result)
-
-    for event in result["stream"]:
-        if "result" not in event:
-            continue
-        for item in event["result"].get("content", []):
-            item_type = item.get("type", "")
-            if item_type == "text":
-                text = item.get("text", "")
-                if text:
-                    parts.append(text)
-            elif item_type == "resource":
-                resource = item.get("resource", {})
-                if "text" in resource:
-                    parts.append(resource["text"])
-                elif "blob" in resource:
-                    try:
-                        parts.append(base64.b64decode(resource["blob"]).decode("utf-8"))
-                    except Exception:
-                        parts.append(f"<binary resource: {resource.get('uri', '?')}>")
-            elif item_type == "image":
-                # Base64-encoded image
-                image_data = item.get("source", {}).get("data", "")
-                mime = item.get("source", {}).get("mediaType", "image/png")
-                parts.append(f"<image: {mime}, {len(image_data)} bytes base64>")
-
-    return "\n".join(parts) if parts else "(no output)"
 
 
 @tool
-def run_code(code: str, packages: list[str] | None = None) -> str:
+def run_code(code: str, description: str = '') -> str:
-    """Execute Python code in a secure managed sandbox and return the output.
+    """Execute Python code in a secure AgentCore managed sandbox and return the output.
-    Optionally install pip packages before running (e.g. ['pandas', 'numpy']).
+    State is maintained within a single session but not across separate calls.
+    Supports data analysis, calculations, file I/O, and most common Python libraries.
 
     Args:
         code: Python code to execute.
-        packages: Optional list of pip packages to install first.
+        description: Optional description prepended as a comment.
 
     Returns:
-        Execution output (stdout, results, errors).
+        JSON result with stdout, stderr, exitCode, and executionTime.
     """
-    try:
+    if description:
-        from bedrock_agentcore.tools import CodeInterpreter, code_session
+        code = f'# {description}\n{code}'
+
+    print(f'[run_code] executing {len(code)}c')
 
     region = os.environ.get('AWS_REGION', 'us-east-1')
 
     with code_session(region) as client:
-            if packages:
+        response = client.invoke('executeCode', {
-                install_raw = client.install_packages(packages)
+            'code': code,
-                install_out = _parse_stream(install_raw) if isinstance(install_raw, dict) else str(install_raw)
+            'language': 'python',
-                print(f'[code_interpreter] install: {install_out[:200]}')
+            'clearContext': False,
+        })
 
-            raw = client.execute_code(code)
+        for event in response['stream']:
-            return _parse_stream(raw)
+            return json.dumps(event['result'])
 
-    except Exception as e:
+    return json.dumps({'isError': True, 'content': [{'type': 'text', 'text': 'No output from code interpreter'}]})
-        import traceback
-        return f'Code interpreter error: {type(e).__name__}: {e}\n{traceback.format_exc()[-500:]}'