fix: align system prompt and apply ruff format to agentcore example

Author: mesutoezdil

examples/zero-code-examples/agentcore/run.py

@@ -33,8 +33,9 @@
 
 load_dotenv(override=True)
 os.environ.setdefault("OTEL_SEMCONV_STABILITY_OPT_IN", "gen_ai_latest_experimental")
-os.environ.setdefault("OTEL_RESOURCE_ATTRIBUTES",
-    "agentevals.eval_set_id=agentcore_eval,agentevals.session_name=agentcore-zero-code")
+os.environ.setdefault(
+    "OTEL_RESOURCE_ATTRIBUTES", "agentevals.eval_set_id=agentcore_eval,agentevals.session_name=agentcore-zero-code"
+)
 
 _telemetry = StrandsTelemetry()
 _telemetry.tracer_provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter(), schedule_delay_millis=1000))
@@ -60,7 +61,7 @@ async def handler(payload):
     agent = Agent(
         model=BedrockModel(model_id="us.amazon.nova-pro-v1:0"),
         tools=[roll_die, check_prime],
-        system_prompt="Use roll_die when asked to roll dice. Use check_prime when asked about prime numbers.",
+        system_prompt="You are a helpful assistant. You can roll dice and check if numbers are prime.",
     )
     async for event in agent.stream_async(prompt):
         yield event

tests/integration/test_live_agents.py

@@ -40,6 +40,7 @@
     reason="GOOGLE_API_KEY not set",
 )
 
+
 def _run_agent(
     script: str,
     otlp_http_port: int,
@@ -72,10 +73,12 @@ def _run_agent(
 
 @contextlib.contextmanager
 def _agentcore_server(otlp_http_port: int, session_name: str, extra_env: dict | None = None):
-    env = {**os.environ,
-           "OTEL_EXPORTER_OTLP_ENDPOINT": f"http://127.0.0.1:{otlp_http_port}",
-           "OTEL_RESOURCE_ATTRIBUTES": f"agentevals.eval_set_id=e2e-test,agentevals.session_name={session_name}",
-           **(extra_env or {})}
+    env = {
+        **os.environ,
+        "OTEL_EXPORTER_OTLP_ENDPOINT": f"http://127.0.0.1:{otlp_http_port}",
+        "OTEL_RESOURCE_ATTRIBUTES": f"agentevals.eval_set_id=e2e-test,agentevals.session_name={session_name}",
+        **(extra_env or {}),
+    }
     proc = subprocess.Popen([sys.executable, os.path.join(REPO_ROOT, _AGENTCORE_SCRIPT)], env=env, cwd=REPO_ROOT)
     try:
         for _ in range(20):
@@ -406,7 +409,9 @@ def test_session_created_spans_only(self, live_servers):
         main_port, otlp_http_port, mgr = live_servers
         session_name = "e2e-agentcore"
         with _agentcore_server(otlp_http_port, session_name, extra_env=_AGENTCORE_ENV):
-            r = httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Roll a 20-sided die"}, timeout=60)
+            r = httpx.post(
+                f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Roll a 20-sided die"}, timeout=60
+            )
             assert r.status_code == 200
         wait_for_session_complete_sync(mgr, session_name, timeout=60)
         s = mgr.sessions[session_name]
@@ -416,7 +421,9 @@ def test_invocations_extracted(self, live_servers):
         main_port, otlp_http_port, mgr = live_servers
         session_name = "e2e-agentcore-inv"
         with _agentcore_server(otlp_http_port, session_name, extra_env=_AGENTCORE_ENV):
-            r = httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Is 17 prime?"}, timeout=60)
+            r = httpx.post(
+                f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Is 17 prime?"}, timeout=60
+            )
             assert r.status_code == 200
         wait_for_session_complete_sync(mgr, session_name, timeout=60)
         assert len(mgr.sessions[session_name].invocations) > 0