style(credential, knowledgebase, memory_collection, model, tool, toolset): format code for better readability

OhYee · OhYee · commit 59a0730fb764 · 2026-05-22T10:33:43.000+08:00
Refactored multiple files to improve code formatting by breaking long lines into multiple lines for better readability. This change enhances maintainability and aligns with coding standards.

---
格式化多个文件以提高可读性
通过将长行拆分为多行来重构代码格式。此更改增强了可维护性，并符合编码标准。

Change-Id: Idf8af571313e0b0342f412dc5fc1efa1d10e1ed3
Signed-off-by: OhYee &lt;oyohyee@oyohyee.com&gt;
diff --git a/agentrun/credential/__credential_async_template.py b/agentrun/credential/__credential_async_template.py
@@ -62,7 +62,9 @@ async def create_async(
         Returns:
             Credential: 创建的凭证对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     async def delete_by_name_async(
diff --git a/agentrun/credential/credential.py b/agentrun/credential/credential.py
@@ -72,7 +72,9 @@ async def create_async(
         Returns:
             Credential: 创建的凭证对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     def create(
@@ -113,7 +115,9 @@ def delete_by_name(
             credential_name: 凭证名称
             config: 配置
         """
-        return cls.__get_client(config=config).delete(credential_name, config=config)
+        return cls.__get_client(config=config).delete(
+            credential_name, config=config
+        )
 
     @classmethod
     async def update_by_name_async(
@@ -185,7 +189,9 @@ def get_by_name(cls, credential_name: str, config: Optional[Config] = None):
         Returns:
             Credential: 凭证对象
         """
-        return cls.__get_client(config=config).get(credential_name, config=config)
+        return cls.__get_client(config=config).get(
+            credential_name, config=config
+        )
 
     @classmethod
     async def _list_page_async(
diff --git a/agentrun/knowledgebase/__knowledgebase_async_template.py b/agentrun/knowledgebase/__knowledgebase_async_template.py
@@ -81,7 +81,9 @@ async def create_async(
         Returns:
             KnowledgeBase: 创建的知识库对象 / Created knowledge base object
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     async def delete_by_name_async(
diff --git a/agentrun/knowledgebase/knowledgebase.py b/agentrun/knowledgebase/knowledgebase.py
@@ -91,7 +91,9 @@ async def create_async(
         Returns:
             KnowledgeBase: 创建的知识库对象 / Created knowledge base object
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     def create(
@@ -208,7 +210,9 @@ def get_by_name(
         Returns:
             KnowledgeBase: 知识库对象 / KnowledgeBase object
         """
-        return cls.__get_client(config=config).get(knowledge_base_name, config=config)
+        return cls.__get_client(config=config).get(
+            knowledge_base_name, config=config
+        )
 
     @classmethod
     async def _list_page_async(
diff --git a/agentrun/memory_collection/__memory_collection_async_template.py b/agentrun/memory_collection/__memory_collection_async_template.py
@@ -60,7 +60,9 @@ async def create_async(
         Returns:
             MemoryCollection: 创建的记忆集合对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     async def delete_by_name_async(
diff --git a/agentrun/memory_collection/memory_collection.py b/agentrun/memory_collection/memory_collection.py
@@ -70,7 +70,9 @@ async def create_async(
         Returns:
             MemoryCollection: 创建的记忆集合对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     def create(
diff --git a/agentrun/model/__model_proxy_async_template.py b/agentrun/model/__model_proxy_async_template.py
@@ -57,7 +57,9 @@ async def create_async(
         Returns:
             ModelProxy: 创建的模型服务对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     async def delete_by_name_async(
diff --git a/agentrun/model/__model_service_async_template.py b/agentrun/model/__model_service_async_template.py
@@ -52,7 +52,9 @@ async def create_async(
         Returns:
             ModelService: 创建的模型服务对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     async def delete_by_name_async(
diff --git a/agentrun/model/model_proxy.py b/agentrun/model/model_proxy.py
@@ -67,7 +67,9 @@ async def create_async(
         Returns:
             ModelProxy: 创建的模型服务对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     def create(
diff --git a/agentrun/model/model_service.py b/agentrun/model/model_service.py
@@ -62,7 +62,9 @@ async def create_async(
         Returns:
             ModelService: 创建的模型服务对象
         """
-        return await cls.__get_client(config=config).create_async(input, config=config)
+        return await cls.__get_client(config=config).create_async(
+            input, config=config
+        )
 
     @classmethod
     def create(
diff --git a/agentrun/sandbox/client.py b/agentrun/sandbox/client.py
@@ -748,8 +748,7 @@ async def delete_sandbox_async(
                 raise ClientError(
                     status_code=0,
                     message=(
-                        "Failed to stop sandbox:"
-                        f" {message or 'Unknown error'}"
+                        f"Failed to stop sandbox: {message or 'Unknown error'}"
                     ),
                 )
 
@@ -805,8 +804,7 @@ def delete_sandbox(
                 raise ClientError(
                     status_code=0,
                     message=(
-                        "Failed to stop sandbox:"
-                        f" {message or 'Unknown error'}"
+                        f"Failed to stop sandbox: {message or 'Unknown error'}"
                     ),
                 )
 
diff --git a/agentrun/sandbox/template.py b/agentrun/sandbox/template.py
@@ -118,9 +118,7 @@ async def create_async(
         )
 
     @classmethod
-    def create(
-        cls, input: TemplateInput, config: Optional[Config] = None
-    ):
+    def create(cls, input: TemplateInput, config: Optional[Config] = None):
         return cls.__get_client(config=config).create_template(
             input, config=config
         )
@@ -172,9 +170,7 @@ async def get_by_name_async(
         )
 
     @classmethod
-    def get_by_name(
-        cls, template_name: str, config: Optional[Config] = None
-    ):
+    def get_by_name(cls, template_name: str, config: Optional[Config] = None):
         return cls.__get_client(config=config).get_template(
             template_name=template_name, config=config
         )
diff --git a/agentrun/tool/model.py b/agentrun/tool/model.py
@@ -5,7 +5,7 @@
 """
 
 from enum import Enum
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 from agentrun.utils.model import BaseModel
 
@@ -194,8 +194,8 @@ class ToolSchema(BaseModel):
     required: Optional[List[str]] = None
     """必填字段 / Required fields"""
 
-    additional_properties: Optional[bool] = None
-    """是否允许额外属性 / Whether additional properties are allowed"""
+    additional_properties: Optional[Union[bool, "ToolSchema"]] = None
+    """额外属性约束 / Additional properties constraint"""
 
     items: Optional["ToolSchema"] = None
     """数组元素类型 / Array item type"""
@@ -291,13 +291,24 @@ def from_any_openapi_schema(cls, schema: Any) -> "ToolSchema":
             else None
         )
 
+        additional_properties_raw = pydash_get(schema, "additionalProperties")
+        additional_properties = (
+            cls()
+            if additional_properties_raw == {}
+            else (
+                cls.from_any_openapi_schema(additional_properties_raw)
+                if isinstance(additional_properties_raw, dict)
+                else additional_properties_raw
+            )
+        )
+
         return cls(
             type=pydash_get(schema, "type"),
             description=pydash_get(schema, "description"),
             title=pydash_get(schema, "title"),
             properties=properties,
             required=pydash_get(schema, "required"),
-            additional_properties=pydash_get(schema, "additionalProperties"),
+            additional_properties=additional_properties,
             items=items,
             min_items=pydash_get(schema, "minItems"),
             max_items=pydash_get(schema, "maxItems"),
@@ -334,7 +345,11 @@ def to_json_schema(self) -> Dict[str, Any]:
         if self.required:
             result["required"] = self.required
         if self.additional_properties is not None:
-            result["additionalProperties"] = self.additional_properties
+            result["additionalProperties"] = (
+                self.additional_properties.to_json_schema()
+                if isinstance(self.additional_properties, ToolSchema)
+                else self.additional_properties
+            )
 
         if self.items:
             result["items"] = self.items.to_json_schema()
diff --git a/agentrun/toolset/model.py b/agentrun/toolset/model.py
@@ -5,7 +5,7 @@
 """
 
 from enum import Enum
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 from agentrun.utils.model import BaseModel, Field, PageableInput
 
@@ -103,7 +103,7 @@ class ToolSchema(BaseModel):
     # 对象类型字段
     properties: Optional[Dict[str, "ToolSchema"]] = None
     required: Optional[List[str]] = None
-    additional_properties: Optional[bool] = None
+    additional_properties: Optional[Union[bool, "ToolSchema"]] = None
 
     # 数组类型字段
     items: Optional["ToolSchema"] = None
@@ -179,6 +179,17 @@ def from_any_openapi_schema(cls, schema: Any) -> "ToolSchema":
             else None
         )
 
+        additional_properties_raw = pg(schema, "additionalProperties")
+        additional_properties = (
+            cls()
+            if additional_properties_raw == {}
+            else (
+                cls.from_any_openapi_schema(additional_properties_raw)
+                if isinstance(additional_properties_raw, dict)
+                else additional_properties_raw
+            )
+        )
+
         return cls(
             # 基本字段
             type=pg(schema, "type"),
@@ -187,7 +198,7 @@ def from_any_openapi_schema(cls, schema: Any) -> "ToolSchema":
             # 对象类型
             properties=properties,
             required=pg(schema, "required"),
-            additional_properties=pg(schema, "additionalProperties"),
+            additional_properties=additional_properties,
             # 数组类型
             items=items,
             min_items=pg(schema, "minItems"),
@@ -231,7 +242,11 @@ def to_json_schema(self) -> Dict[str, Any]:
         if self.required:
             result["required"] = self.required
         if self.additional_properties is not None:
-            result["additionalProperties"] = self.additional_properties
+            result["additionalProperties"] = (
+                self.additional_properties.to_json_schema()
+                if isinstance(self.additional_properties, ToolSchema)
+                else self.additional_properties
+            )
 
         # 数组类型
         if self.items:
diff --git a/tests/e2e/test_workspace_id.py b/tests/e2e/test_workspace_id.py
@@ -32,18 +32,16 @@
     CredentialListInput,
 )
 from agentrun.sandbox import Template
-from agentrun.sandbox.model import (
-    PageableInput,
-    TemplateInput,
-    TemplateType,
-)
+from agentrun.sandbox.model import PageableInput, TemplateInput, TemplateType
 from agentrun.utils.exception import ResourceNotExistError
 
 WORKSPACE_ID = os.getenv("AGENTRUN_TEST_WORKSPACE_ID")
 
 pytestmark = pytest.mark.skipif(
     not WORKSPACE_ID,
-    reason="AGENTRUN_TEST_WORKSPACE_ID not configured; skipping workspace_id E2E",
+    reason=(
+        "AGENTRUN_TEST_WORKSPACE_ID not configured; skipping workspace_id E2E"
+    ),
 )
 
 
@@ -98,7 +96,8 @@ async def test_credential_with_workspace_id_async(
             )
             names = [item.credential_name for item in list_results]
             assert credential_name in names, (
-                f"list(workspace_id={ws!r}) 未返回刚创建的凭证 {credential_name!r}，"
+                f"list(workspace_id={ws!r}) 未返回刚创建的凭证"
+                f" {credential_name!r}，"
                 f"实际返回 {names!r}"
             )
             # 列表项的 workspace_id 也应该是同一个
@@ -112,9 +111,7 @@ async def test_credential_with_workspace_id_async(
                 except ResourceNotExistError:
                     pass
 
-    def test_credential_with_workspace_id(
-        self, credential_name: str
-    ):
+    def test_credential_with_workspace_id(self, credential_name: str):
         """凭证创建时指定 workspace_id，回读与列举均能拿到该 workspace_id"""
         client = CredentialClient()
         ws = WORKSPACE_ID  # type: ignore[assignment]
@@ -139,20 +136,17 @@ def test_credential_with_workspace_id(
             ), f"create 返回的 workspace_id 不匹配: {cred.workspace_id!r}"
 
             # 2. get 接口回读 workspace_id
-            cred_fetched = client.get(
-                credential_name=credential_name
-            )
+            cred_fetched = client.get(credential_name=credential_name)
             assert (
                 cred_fetched.workspace_id == ws
             ), f"get 返回的 workspace_id 不匹配: {cred_fetched.workspace_id!r}"
 
             # 3. list 接口按 workspace_id 过滤，本次创建的资源应在结果中
-            list_results = client.list(
-                CredentialListInput(workspace_id=ws)
-            )
+            list_results = client.list(CredentialListInput(workspace_id=ws))
             names = [item.credential_name for item in list_results]
             assert credential_name in names, (
-                f"list(workspace_id={ws!r}) 未返回刚创建的凭证 {credential_name!r}，"
+                f"list(workspace_id={ws!r}) 未返回刚创建的凭证"
+                f" {credential_name!r}，"
                 f"实际返回 {names!r}"
             )
             # 列表项的 workspace_id 也应该是同一个
diff --git a/tests/unittests/memory_collection/test_memory_collection.py b/tests/unittests/memory_collection/test_memory_collection.py
@@ -630,7 +630,9 @@ def test_build_mem0_config_with_mysql_sync(self, mock_get_credential):
         assert vs_config["port"] == 3307
         assert vs_config["embedding_model_dims"] == 1024
 
-    @patch("agentrun.memory_collection.memory_collection.MemoryCollection._resolve_model_service_config")
+    @patch(
+        "agentrun.memory_collection.memory_collection.MemoryCollection._resolve_model_service_config"
+    )
     @patch("agentrun.credential.Credential.get_by_name")
     def test_build_mem0_config_mysql_embedder_dims_sync(
         self, mock_get_credential, mock_resolve
@@ -660,10 +662,14 @@ def test_build_mem0_config_mysql_embedder_dims_sync(
                 config=EmbedderConfigConfig(model="text-embedding-v3"),
             ),
         )
-        config = MemoryCollection._build_mem0_config(memory_collection, None, None)
+        config = MemoryCollection._build_mem0_config(
+            memory_collection, None, None
+        )
         assert config["embedder"]["config"]["embedding_dims"] == 1024
 
-    @patch("agentrun.memory_collection.memory_collection.MemoryCollection._resolve_model_service_config_async")
+    @patch(
+        "agentrun.memory_collection.memory_collection.MemoryCollection._resolve_model_service_config_async"
+    )
     @patch("agentrun.credential.Credential.get_by_name_async")
     @pytest.mark.asyncio
     async def test_build_mem0_config_mysql_embedder_dims_async(
diff --git a/tests/unittests/test_workspace_id.py b/tests/unittests/test_workspace_id.py
diff --git a/tests/unittests/tool/test_model.py b/tests/unittests/tool/test_model.py
diff --git a/tests/unittests/toolset/test_model.py b/tests/unittests/toolset/test_model.py
