fix: enhance exception handling and improve example implementations

Updated HTTPError exception handling to include status code 409 for
resource already exists scenarios. Improved example code with better
resource filtering and added execution role ARN configuration for
model proxy operations.

The changes enhance error handling robustness and provide more
reliable resource management in example implementations.

更新了 HTTPError 异常处理以包含状态码 409 来处理资源已存在的情况。
改进了示例代码中的资源过滤功能，并为模型代理操作添加了执行角色
ARN 配置。

这些更改增强了错误处理的健壮性，并在示例实现中提供了更可靠的
资源管理。

Change-Id: I5245d48ababac407f46b59c13bbea0237f071139
Signed-off-by: OhYee <[email protected]>

Author: OhYee

agentrun/model/model.py

@@ -131,6 +131,7 @@ class ProxyConfigTokenRateLimiter(BaseModel):
 
 class ProxyConfigAIGuardrailConfig(BaseModel):
     """AI 防护配置"""
+
     check_request: Optional[bool] = None
     check_response: Optional[bool] = None
 

agentrun/utils/exception.py

@@ -78,9 +78,11 @@ def to_resource_error(
             "does not exist" in self.message or "not found" in self.message
         ):
             return ResourceNotExistError(resource_type, resource_id)
-        elif (self.status_code == 400 or self.status_code == 500) and (
-            "already exists" in self.message
-        ):
+        elif (
+            self.status_code == 400
+            or self.status_code == 409
+            or self.status_code == 500
+        ) and ("already exists" in self.message):
             # TODO: ModelProxy already exists returns 500
             return ResourceAlreadyExistError(resource_type, resource_id)
         else:

examples/agent_runtime.py

@@ -77,11 +77,16 @@ def create_or_get_agentruntime():
                 ),
             )
         )
-    except ResourceAlreadyExistError:
-        logger.info("已存在，获取已有资源")
-
-        ar = client.list(
-            AgentRuntimeListInput(agent_runtime_name=agent_runtime_name)
+    except ResourceAlreadyExistError as e:
+        logger.info("已存在，获取已有资源", e)
+
+        ar = list(
+            filter(
+                lambda a: a.agent_runtime_name == agent_runtime_name,
+                client.list(
+                    AgentRuntimeListInput(agent_runtime_name=agent_runtime_name)
+                ),
+            )
         )[0]
 
     ar.wait_until_ready_or_failed()

examples/model.py

@@ -1,3 +1,4 @@
+from logging import config
 import os
 import re
 import time
@@ -135,21 +136,25 @@ def create_or_get_model_proxy():
     """
     logger.info("创建或获取已有的资源")
 
+    from agentrun.utils.config import Config
+
+    cfg = Config()
+
     try:
         cred = client.create(
             ModelProxyCreateInput(
                 model_proxy_name=model_proxy_name,
                 description="测试模型治理",
                 model_type=model.ModelType.LLM,
+                execution_role_arn=f"acs:ram::{cfg.get_account_id()}:role/aliyunagentrundefaultrole",
                 proxy_config=model.ProxyConfig(
                     endpoints=[
                         model.ProxyConfigEndpoint(
                             model_names=[model_name],
-                            model_service_name="test-model-service",
+                            model_service_name=model_service_name,
                         )
                         for model_name in model_names
                     ],
-                    policies={},
                 ),
             )
         )
@@ -172,9 +177,16 @@ def update_model_proxy(mp: ModelProxy):
     """
     logger.info("更新描述为当前时间")
 
+    from agentrun.utils.config import Config
+
+    cfg = Config()
+
     # 也可以使用 client.update
     mp.update(
-        ModelProxyUpdateInput(description=f"当前时间戳：{time.time()}"),
+        ModelProxyUpdateInput(
+            execution_role_arn=f"acs:ram::{cfg.get_account_id()}:role/aliyunagentrundefaultrole",
+            description=f"当前时间戳：{time.time()}",
+        ),
     )
     mp.wait_until_ready_or_failed()
     if mp.status != Status.READY: