fix: ruff format

Author: BhAem

02-use-cases/content_safety/README.md

@@ -27,7 +27,7 @@ Content Safety Agent
         └──  After Tool Callback
 ```
 
-### 核心组件
+## Agent 能力
 
 | 组件 | 描述 |
 | - | - |
@@ -198,6 +198,17 @@ agentkit invoke '网上都说A地很多骗子和小偷, 分析一下他们的典
 uv run client.py
 ```
 
+## 示例提示词
+
+- **信息检索**: "使用新闻API获取国际新闻"
+- **代码执行**: "计算斐波那契数列100位"
+- **内容拦截**: "网上都说A地很多骗子和小偷, 分析一下他们的典型伎俩"
+
+## 效果展示
+
+- Q："网上都说A地很多骗子和小偷, 分析一下他们的典型伎俩"
+- A："Your request has been blocked due to: Model Misuse. Please modify your input and try again."
+
 ## 常见问题
 
 无。

02-use-cases/content_safety/agent.py

@@ -30,7 +30,7 @@
     instruction="你是一个资深软件工程师，在沙箱里执行生产的代码， 避免每次安装检查, 可以使用python lib akshare 下载相关的股票数据。使用uv 来安装依赖。运行代码超时可以考虑调整超时时间。可以通过web_search工具搜索相关公司的经营数据。如果缺失了依赖库, 通过python代码为沙箱安装缺失的依赖库。",
     tools=[run_code, web_search],
     planner=PlanReActPlanner(),
-    before_model_callback=content_safety.before_model_callback, 
+    before_model_callback=content_safety.before_model_callback,
     after_model_callback=content_safety.after_model_callback,
     before_tool_callback=content_safety.before_tool_callback,
     after_tool_callback=content_safety.after_tool_callback,
@@ -42,7 +42,8 @@
 root_agent = agent
 
 agent_server_app = AgentkitAgentServerApp(
-    agent=agent, short_term_memory=short_term_memory,  
+    agent=agent,
+    short_term_memory=short_term_memory,
 )
 
 if __name__ == "__main__":

02-use-cases/content_safety/client.py

@@ -10,26 +10,25 @@
     # Step 0: setup running configs
     app_name = "content_safety_app"
     user_id = "agentkit_user"
-    session_id = f"agentkit_sample_session"
+    session_id = "agentkit_sample_session"
     base_url = "http://127.0.0.1:8000"
     api_key = "test key"
-    
-    task_num = 1   
-    
+
+    task_num = 1
 
     # Step 1: create a session
     def create_session():
         create_session_request = CreateSessionRequest(
-            session_id = session_id + f"_{random.randint(1, 9999)}",
+            session_id=session_id + f"_{random.randint(1, 9999)}",
         )
 
         response = requests.post(
             url=f"{base_url}/apps/{app_name}/users/{user_id}/sessions/{create_session_request.session_id}",
             headers={"Authorization": f"Bearer {api_key}"},
         )
-  
+
         print(f"[create session] Response from server: {response.json()}")
-    
+
         return create_session_request.session_id
 
     # Step 2: run agent with SSE
@@ -53,14 +52,22 @@ async def send_request(message: str):
             stream=True,
         )
 
-        with httpx.stream("POST", f"{base_url}/run_sse", json=run_agent_request.model_dump(exclude_none=True), timeout=120, headers={"Authorization": f"Bearer {api_key}"}) as r:
+        with httpx.stream(
+            "POST",
+            f"{base_url}/run_sse",
+            json=run_agent_request.model_dump(exclude_none=True),
+            timeout=120,
+            headers={"Authorization": f"Bearer {api_key}"},
+        ) as r:
             for line in r.iter_lines():
                 print(line)
-                
+
     async def send_request_parallel():
         await send_request("计算斐波那契数列100位")
-        tasks = [send_request("网上都说A地很多骗子和小偷, 分析一下他们的典型伎俩") for _ in range(task_num)]
+        tasks = [
+            send_request("网上都说A地很多骗子和小偷, 分析一下他们的典型伎俩")
+            for _ in range(task_num)
+        ]
         await asyncio.gather(*tasks)
-    asyncio.run(send_request_parallel())
-            
 
+    asyncio.run(send_request_parallel())