bedrock-agentcore-starter-toolkit 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

bedrock_agentcore_starter_toolkit/services/import_agent/scripts/base_bedrock_translate.py

@@ -1031,6 +1031,10 @@ class BaseBedrockTranslator:
 
             return code_1p
 
+    def _get_url_regex_pattern(self) -> str:
+        """Get the URL regex pattern for source extraction."""
+        return r"(?:https?://|www\.)(?:[a-zA-Z0-9-]+\.)+[a-zA-Z]{2,}(?:/[^/\s]*)*"
+
     def generate_entrypoint_code(self, platform: str) -> str:
         """Generate entrypoint code for the agent."""
         entrypoint_code = ""
@@ -1059,7 +1063,8 @@ class BaseBedrockTranslator:
             else "tools_used.update([msg.name for msg in agent_result if isinstance(msg, ToolMessage)])"
         )
         response_content_code = "str(agent_result)" if platform == "strands" else "agent_result[-1].content"
-
+        url_pattern = self._get_url_regex_pattern()
+        
         entrypoint_code += f"""
     def endpoint(payload, context):
         try:
@@ -1079,7 +1084,7 @@ class BaseBedrockTranslator:
 
             # Gathering sources from the response
             sources = []
-            urls = re.findall(r'(?:https?://|www\.)(?:[a-zA-Z0-9\-]+\.)+[a-zA-Z]{{2,}}(?:/[^/\s]*)*', response_content)
+            urls = re.findall({repr(url_pattern)}, response_content)
             source_tags = re.findall(r"<source>(.*?)</source>", response_content)
             sources.extend(urls)
             sources.extend(source_tags)

bedrock_agentcore_starter_toolkit/services/runtime.py

@@ -29,7 +29,6 @@ def _handle_http_response(response) -> dict:
     if "text/event-stream" in response.headers.get("content-type", ""):
         return _handle_streaming_response(response)
     else:
-        # Check if response has content
         if not response.content:
             raise ValueError("Empty response from agent endpoint")
 
@@ -43,6 +42,15 @@ def _handle_aws_response(response) -> dict:
         try:
             events = []
             for event in response.get("response", []):
+                if isinstance(event, bytes):
+                    try:
+                        decoded = event.decode("utf-8")
+                        if decoded.startswith('"') and decoded.endswith('"'):
+                            event = json.loads(decoded)
+                        else:
+                            event = decoded
+                    except (UnicodeDecodeError, json.JSONDecodeError):
+                        pass
                 events.append(event)
         except Exception as e:
             events = [f"Error reading EventStream: {e}"]
@@ -64,11 +72,11 @@ def _handle_streaming_response(response) -> Dict[str, Any]:
                         text_chunk = parsed_chunk
                     else:
                         text_chunk = json.dumps(parsed_chunk, ensure_ascii=False)
-                        text_chunk += "\n"
-                    console.print(text_chunk, end="", style="bold cyan")
+                        text_chunk += "\n\n"
+                    console.print(text_chunk, end="")
                     complete_text += text_chunk
                 except json.JSONDecodeError:
-                    console.print(json_chunk, style="bold cyan")
+                    console.print(json_chunk)
                     continue
     console.print()
     return {}
@@ -375,6 +383,28 @@ class BedrockAgentCoreClient:
             endpointName=endpoint_name,
         )
 
+    def delete_agent_runtime_endpoint(self, agent_id: str, endpoint_name: str = "DEFAULT") -> Dict:
+        """Delete agent runtime endpoint.
+
+        Args:
+            agent_id: Agent ID to delete endpoint for
+            endpoint_name: Endpoint name, defaults to "DEFAULT"
+
+        Returns:
+            Response containing the deletion status
+        """
+        self.logger.info("Deleting agent runtime endpoint '%s' for agent ID: %s", endpoint_name, agent_id)
+        try:
+            response = self.client.delete_agent_runtime_endpoint(
+                agentRuntimeId=agent_id,
+                endpointName=endpoint_name,
+            )
+            self.logger.info("Successfully initiated deletion of endpoint '%s' for agent ID: %s", endpoint_name, agent_id)
+            return response
+        except Exception as e:
+            self.logger.error("Failed to delete endpoint '%s' for agent ID '%s': %s", endpoint_name, agent_id, str(e))
+            raise
+
     def invoke_endpoint(
         self,
         agent_arn: str,

bedrock_agentcore_starter_toolkit/utils/runtime/logs.py

@@ -1,5 +1,6 @@
 """Utility functions for agent log information."""
 
+from datetime import datetime, timezone
 from typing import Optional, Tuple
 
 
@@ -14,8 +15,11 @@ def get_agent_log_paths(agent_id: str, endpoint_name: Optional[str] = None) -> T
         Tuple of (runtime_log_group, otel_log_group)
     """
     endpoint_name = endpoint_name or "DEFAULT"
-    runtime_log_group = f"/aws/bedrock-agentcore/runtimes/{agent_id}-{endpoint_name}"
-    otel_log_group = f"/aws/bedrock-agentcore/runtimes/{agent_id}-{endpoint_name} --log-stream-names otel-rt-logs"
+    runtime_log_group = (
+        f"/aws/bedrock-agentcore/runtimes/{agent_id}-{endpoint_name} "
+        f'--log-stream-name-prefix "{datetime.now(timezone.utc).strftime("%Y/%m/%d")}/\\[runtime-logs]"'
+    )
+    otel_log_group = f'/aws/bedrock-agentcore/runtimes/{agent_id}-{endpoint_name} --log-stream-names "otel-rt-logs"'
     return runtime_log_group, otel_log_group
 
 

{bedrock_agentcore_starter_toolkit-0.1.8.dist-info → bedrock_agentcore_starter_toolkit-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bedrock-agentcore-starter-toolkit
-Version: 0.1.8
+Version: 0.1.10
 Summary: A starter toolkit for using Bedrock AgentCore
 Project-URL: Homepage, https://github.com/aws/bedrock-agentcore-starter-toolkit
 Project-URL: Bug Tracker, https://github.com/aws/bedrock-agentcore-starter-toolkit/issues
@@ -22,7 +22,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Requires-Dist: autopep8>=2.3.2
-Requires-Dist: bedrock-agentcore>=0.1.2
+Requires-Dist: bedrock-agentcore>=0.1.3
 Requires-Dist: boto3>=1.39.7
 Requires-Dist: botocore>=1.39.7
 Requires-Dist: docstring-parser<1.0,>=0.15
@@ -38,6 +38,7 @@ Requires-Dist: questionary>=2.1.0
 Requires-Dist: requests>=2.25.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: ruamel-yaml>=0.18.14
+Requires-Dist: strands-agents>=1.7.1
 Requires-Dist: toml>=0.10.2
 Requires-Dist: typer>=0.16.0
 Requires-Dist: typing-extensions<5.0.0,>=4.13.2
@@ -87,6 +88,7 @@ AgentCore Runtime is a secure, serverless runtime purpose-built for deploying an
 ## 🧠 Amazon Bedrock AgentCore Memory
 AgentCore Memory makes it easy for developers to build context aware agents by eliminating complex memory infrastructure management while providing full control over what the AI agent remembers. Memory provides industry-leading accuracy along with support for both short-term memory for multi-turn conversations and long-term memory that can be shared across agents and sessions.
 
+
 **[Memory Quick Start](https://aws.github.io/bedrock-agentcore-starter-toolkit/user-guide/memory/quickstart.html)**
 
 ## 🔗 Amazon Bedrock AgentCore Gateway
@@ -99,7 +101,6 @@ AgentCore Code Interpreter tool enables agents to securely execute code in isola
 
 **[Code Interpreter Quick Start](https://aws.github.io/bedrock-agentcore-starter-toolkit/user-guide/builtin-tools/quickstart-code-interpreter.html)**
 
-
 ## 🌐 Amazon Bedrock AgentCore Browser
 AgentCore Browser tool provides a fast, secure, cloud-based browser runtime to enable AI agents to interact with websites at scale. It provides enterprise-grade security, comprehensive observability features, and automatically scales— all without infrastructure management overhead.
 

{bedrock_agentcore_starter_toolkit-0.1.8.dist-info → bedrock_agentcore_starter_toolkit-0.1.10.dist-info}/RECORD

@@ -1,6 +1,6 @@
 bedrock_agentcore_starter_toolkit/__init__.py,sha256=tN3-JWKvxk4ZSJQJIHQ4mMsDtt8J_cDCz-drcGU9USA,120
 bedrock_agentcore_starter_toolkit/cli/__init__.py,sha256=WuIWfHtJKD9gQA-mE49bq8OHGb0Ugt8upRGaOp6-H90,62
-bedrock_agentcore_starter_toolkit/cli/cli.py,sha256=rmPi14B8WDbbJbrcUIKjXHtbFyfV2cJ7K8yG2_C1kuM,981
+bedrock_agentcore_starter_toolkit/cli/cli.py,sha256=QcwVMLtiTcvgW166tgOmhmI56UQJez-P7b45eujr8wc,1064
 bedrock_agentcore_starter_toolkit/cli/common.py,sha256=R9ZRKmW9gq7u7gkTiCUOTCQTno2lXnsplnq2x0-k2v8,1311
 bedrock_agentcore_starter_toolkit/cli/gateway/__init__.py,sha256=8IZ0kSe6Kz5s2j-SBsoc6qy04MbK85RMTQwZCiR2wmo,68
 bedrock_agentcore_starter_toolkit/cli/gateway/commands.py,sha256=3DuXCvqXlmHU2cmjGzDruyc2Fkrpgoi158myj0vc3nA,3265
@@ -9,7 +9,7 @@ bedrock_agentcore_starter_toolkit/cli/import_agent/__init__.py,sha256=tyM3dXM3Tc
 bedrock_agentcore_starter_toolkit/cli/import_agent/agent_info.py,sha256=V0fZEbV76_H3gmkA17yscyJ8UdcMAquzNqENQ9DXdHA,9477
 bedrock_agentcore_starter_toolkit/cli/import_agent/commands.py,sha256=vpA66fGwKf6cxBelT3Yb_Af5Ow3AMquGLIDe-tBLjmc,22419
 bedrock_agentcore_starter_toolkit/cli/runtime/__init__.py,sha256=0zKPPoYThoDIr3DZhIQJavq5nVTKRsgcE1s9--wx118,60
-bedrock_agentcore_starter_toolkit/cli/runtime/commands.py,sha256=4Mk4ZjlgzYA55Zt--wCDxC84pgso7hpDhdLIUn1f3CI,33141
+bedrock_agentcore_starter_toolkit/cli/runtime/commands.py,sha256=bo6SXsYPDH-X0pxBwmBASz1hS5DrkJ-Wip5JTpU-7r4,44208
 bedrock_agentcore_starter_toolkit/cli/runtime/configuration_manager.py,sha256=5TJK80uzA1ARh263smLfthw1t5Ona3bAtdO1pE7OfNo,5808
 bedrock_agentcore_starter_toolkit/notebook/__init__.py,sha256=wH1ZzIVKsKT_P0qX2kIIoyVxeHj8K40odfR1YI3McHw,129
 bedrock_agentcore_starter_toolkit/notebook/runtime/__init__.py,sha256=DoGfB_uGFLANJVE9rSZtTH3ymw76WBWmD9vORvBIdXs,66
@@ -19,19 +19,20 @@ bedrock_agentcore_starter_toolkit/operations/gateway/__init__.py,sha256=5Qo1GeN-
 bedrock_agentcore_starter_toolkit/operations/gateway/client.py,sha256=8WP2E_u6h1-mPfz4TgNMloD6bcXfoXQDL3UCrbyhT7g,20068
 bedrock_agentcore_starter_toolkit/operations/gateway/constants.py,sha256=0_4J6BN4VAE4-XTQhPTEACkhilRrFqu_iKiuHSm2pYk,4610
 bedrock_agentcore_starter_toolkit/operations/gateway/create_lambda.py,sha256=MQsBJfUj26zBh7LqYWLoekHuvbAHIJE8qcXwrmJOhM0,2875
-bedrock_agentcore_starter_toolkit/operations/gateway/create_role.py,sha256=a38JE28PbdRabskTgLXsiqHqWS1vt06jxEEGneYPO_g,3145
+bedrock_agentcore_starter_toolkit/operations/gateway/create_role.py,sha256=Oma9s6K4T9EFgwPAUDungDL9fGa3W0LsHBknIXUmSpI,4474
 bedrock_agentcore_starter_toolkit/operations/gateway/exceptions.py,sha256=WjsrE7lT2708CJniM_ISMzkfNX4IUteGgvOxleD9JZY,272
-bedrock_agentcore_starter_toolkit/operations/runtime/__init__.py,sha256=7ucAtIbabrDmEaHOfGqHtEr2CkQlEsb5O2tkHirXCqU,673
+bedrock_agentcore_starter_toolkit/operations/runtime/__init__.py,sha256=0jRUuwSoqh4R_WqzfP4XAXngrgyFK5uH8JGXUVars6Y,793
 bedrock_agentcore_starter_toolkit/operations/runtime/configure.py,sha256=7xjNN6fn1U2cv20W2tmHdlbSjc-RU953tnbOaH5iXPQ,9056
 bedrock_agentcore_starter_toolkit/operations/runtime/create_role.py,sha256=o3rimy-9SDOS0r-DKHctKSS6dAVIGelhn1zUhrSeolY,15952
-bedrock_agentcore_starter_toolkit/operations/runtime/invoke.py,sha256=ZcPXI7Zx7NO7G7fM2xY6JKVoo2CR2sdyBlBJ63Pd6MI,4559
+bedrock_agentcore_starter_toolkit/operations/runtime/destroy.py,sha256=a-sAYJScn2Ut2i6NE8AC0-uLtSzjMo204P5nTiGil6M,23920
+bedrock_agentcore_starter_toolkit/operations/runtime/invoke.py,sha256=Wx8NFIEdDmnZDuo0JW0inS6iUQVuGQuxxYBjMHUTwvw,4560
 bedrock_agentcore_starter_toolkit/operations/runtime/launch.py,sha256=RbQt76m1a3TZCWDuMBAP91AA4eH35nW3jk7He8Yqlf8,19260
-bedrock_agentcore_starter_toolkit/operations/runtime/models.py,sha256=Qr45hc73a25cqmiSErq9-degckPXuVhnENyDSshU_mU,3673
+bedrock_agentcore_starter_toolkit/operations/runtime/models.py,sha256=iSpD6Zc-FyS4Fu_oRzXzre4N9uYuzBMDkvZZx4MhBGE,4220
 bedrock_agentcore_starter_toolkit/operations/runtime/status.py,sha256=tpOtzAq1S3z_7baxR_WOT8Avo3MtWKGpelVOOfb-uMA,2516
 bedrock_agentcore_starter_toolkit/services/__init__.py,sha256=s7QtYYFCCX2ff0gZHUdDxCDI3zhUq0fPsjevZbF9xdA,66
 bedrock_agentcore_starter_toolkit/services/codebuild.py,sha256=77aagVvjSMQxBfNmwYTO6lwE_2izXGqgc6kWp9Bf98U,13618
 bedrock_agentcore_starter_toolkit/services/ecr.py,sha256=nW9wIZcXI6amZeLVSmM9F6awVBQP1-zrFXTozSNEDjo,2805
-bedrock_agentcore_starter_toolkit/services/runtime.py,sha256=zGC-p4fagXHWemtdncg1K1h0MFkqd9CsvI3Px_tmAX8,18601
+bedrock_agentcore_starter_toolkit/services/runtime.py,sha256=YXz_1XW6yN6H1_2JNwHpkazYIMQ8bv4pBmt2M8Xawhs,19921
 bedrock_agentcore_starter_toolkit/services/import_agent/__init__.py,sha256=ig-xanZqA3oJdRTucYz9xF9_2yi31ADRVIKFsnIw_l4,68
 bedrock_agentcore_starter_toolkit/services/import_agent/utils.py,sha256=ErY-J0hClJbt5D-pQ99ma-1Fll35tHjqOttM-h0bKKw,15675
 bedrock_agentcore_starter_toolkit/services/import_agent/assets/memory_manager_template.py,sha256=RKrtTxir5XxyMg6cim4CAEbguaxb4mg1Qb31pd7D_kY,7892
@@ -39,7 +40,7 @@ bedrock_agentcore_starter_toolkit/services/import_agent/assets/requirements_lang
 bedrock_agentcore_starter_toolkit/services/import_agent/assets/requirements_strands.j2,sha256=ZAWzCFwdj8Mbq8aBGO3Jnxz0G8XJZPNpIsPf_2Jetfk,100
 bedrock_agentcore_starter_toolkit/services/import_agent/assets/template_fixtures_merged.json,sha256=Xem7jS8n96QEyHBmnPAvWHM4AyTLyma7Nq9sCVHuXJA,175239
 bedrock_agentcore_starter_toolkit/services/import_agent/scripts/__init__.py,sha256=UmhcnsfIo2P1Z9VAJ6Ua2mSkxs4BOeTxgFMcgQXQWCQ,79
-bedrock_agentcore_starter_toolkit/services/import_agent/scripts/base_bedrock_translate.py,sha256=xAqf0Rrhb4QWE_6WBk98NkQdHbCyy2uFzKbiw_9cWA4,72520
+bedrock_agentcore_starter_toolkit/services/import_agent/scripts/base_bedrock_translate.py,sha256=bgiPQAFScuiDGQW2v5Y-qDKwPY8v_GtUcrLnoqjTNM0,72721
 bedrock_agentcore_starter_toolkit/services/import_agent/scripts/bedrock_to_langchain.py,sha256=xwoZcmZ27CfYohwSfLIgbFxa2ubiBFhvufgQEWZkmLk,14950
 bedrock_agentcore_starter_toolkit/services/import_agent/scripts/bedrock_to_strands.py,sha256=eC3v2Ts7B_RUGl4bkUim2uFYYuZZYP9syYBlylJ2gXs,14642
 bedrock_agentcore_starter_toolkit/utils/endpoints.py,sha256=1gIDRd1oO1fymYpiedVit7m6zl5k6N8Ns9N-2ix7ZaE,1153
@@ -47,16 +48,16 @@ bedrock_agentcore_starter_toolkit/utils/logging_config.py,sha256=NtZDyndNKCAbz7j
 bedrock_agentcore_starter_toolkit/utils/runtime/config.py,sha256=qRQid59rD3zJ0d0hcBY4-8R52PNIWEIUt9iR3biuz_c,4495
 bedrock_agentcore_starter_toolkit/utils/runtime/container.py,sha256=wGJLxwT2mhTGPxSoFb6x9lW7Tyz37YghIvSs4jBnO8A,15679
 bedrock_agentcore_starter_toolkit/utils/runtime/entrypoint.py,sha256=d-XjEwvQOdpRHvBGLdIBCs1fgUwNwB0EL_WkcdQXwXQ,5893
-bedrock_agentcore_starter_toolkit/utils/runtime/logs.py,sha256=zRNYiNtEEYatZSRbc5VrUNgWhyP3Pl6p-hMl8AEUx_o,1101
+bedrock_agentcore_starter_toolkit/utils/runtime/logs.py,sha256=-8Nipxg13VJTPuztewvZh04fy1w-pxlR_ALriFr9FIY,1265
 bedrock_agentcore_starter_toolkit/utils/runtime/policy_template.py,sha256=CgER7YXPh0BpR6JcTcumDL_k8bhmfLSEok1sf09-31I,2054
 bedrock_agentcore_starter_toolkit/utils/runtime/schema.py,sha256=gZ0zPvry-ZkXwqUEAy6Izz1RJvmZaXA6a2twnSdk1AY,6418
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/Dockerfile.j2,sha256=0KmVBMP3ZeCrt7dQ5cR8WTvI8ZVt_SFgHYRxn6DYMfA,1261
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/dockerignore.template,sha256=b_70sBi0MwkTuA9TqxPqFcWK7TcmpaXBJ6M2Ipox65Q,691
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/execution_role_policy.json.j2,sha256=-0AXT46IYVQr4fwueXTQ0FVXcCshZFNx7-2mViR34Co,4941
 bedrock_agentcore_starter_toolkit/utils/runtime/templates/execution_role_trust_policy.json.j2,sha256=PPJF6Ofq70W5DUE0NlbmnZjw5RkgepPgkskxEgEG28o,473
-bedrock_agentcore_starter_toolkit-0.1.8.dist-info/METADATA,sha256=T-Y9VXQUjDahbwI5rWcjOYWCgrGU7avljQfd6DBtgLw,9706
-bedrock_agentcore_starter_toolkit-0.1.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bedrock_agentcore_starter_toolkit-0.1.8.dist-info/entry_points.txt,sha256=Tf94DkUf2Tp8P7p8MEXLxre7A7Pp_XNukteiz0wHnk8,77
-bedrock_agentcore_starter_toolkit-0.1.8.dist-info/licenses/LICENSE.txt,sha256=nNPOMinitYdtfbhdQgsPgz1UowBznU6QVN3Xs0pSTKU,10768
-bedrock_agentcore_starter_toolkit-0.1.8.dist-info/licenses/NOTICE.txt,sha256=rkBsg8DbKqfIoQbveqX9foR4uJPUVAokbkr02pRPilE,8674
-bedrock_agentcore_starter_toolkit-0.1.8.dist-info/RECORD,,
+bedrock_agentcore_starter_toolkit-0.1.10.dist-info/METADATA,sha256=Jj6DyqfwRMHvN38lQ4rpB6dzheNwtFLgyLiGpXFpBIU,9744
+bedrock_agentcore_starter_toolkit-0.1.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bedrock_agentcore_starter_toolkit-0.1.10.dist-info/entry_points.txt,sha256=Tf94DkUf2Tp8P7p8MEXLxre7A7Pp_XNukteiz0wHnk8,77
+bedrock_agentcore_starter_toolkit-0.1.10.dist-info/licenses/LICENSE.txt,sha256=nNPOMinitYdtfbhdQgsPgz1UowBznU6QVN3Xs0pSTKU,10768
+bedrock_agentcore_starter_toolkit-0.1.10.dist-info/licenses/NOTICE.txt,sha256=rkBsg8DbKqfIoQbveqX9foR4uJPUVAokbkr02pRPilE,8674
+bedrock_agentcore_starter_toolkit-0.1.10.dist-info/RECORD,,