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

OhYee · claude · OhYee · commit e9b7d2f75d54 · 2026-05-22T11:26:06.000+08:00
Change-Id: If7b4e36b9005765d075e7ac2e535a10c5beb4f37
Co-developed-by: Claude &lt;noreply@anthropic.com&gt;
Co-Authored-By: Claude Opus 4.6 (1M context) &lt;noreply@anthropic.com&gt;
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/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
@@ -36,13 +36,8 @@
     ModelServiceCreateInput,
     ModelServiceListInput,
 )
-from agentrun.sandbox.model import (
-    PageableInput as SandboxPageableInput,
-)
-from agentrun.sandbox.model import (
-    TemplateInput,
-    TemplateType,
-)
+from agentrun.sandbox.model import PageableInput as SandboxPageableInput
+from agentrun.sandbox.model import TemplateInput, TemplateType
 from agentrun.utils.model import BaseModel
 
 WORKSPACE_ID = "ws-test-12345"
