Ver Fonte

fix(tools): suppress RuntimeWarnings in podcast audio generator (#10182)

-LAN- há 5 meses atrás
pai
commit
86739bea8b

+ 3 - 2
api/core/tools/provider/builtin/podcast_generator/tools/podcast_audio_generator.py

@@ -1,8 +1,8 @@
 import concurrent.futures
 import io
 import random
+import warnings
 from typing import Any, Literal, Optional, Union
-from warnings import catch_warnings
 
 import openai
 
@@ -10,7 +10,8 @@ from core.tools.entities.tool_entities import ToolInvokeMessage
 from core.tools.errors import ToolParameterValidationError, ToolProviderCredentialValidationError
 from core.tools.tool.builtin_tool import BuiltinTool
 
-with catch_warnings(action="ignore", category=RuntimeWarning):
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
     from pydub import AudioSegment