소스 검색

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

-LAN- 5 달 전
부모
커밋
86739bea8b
1개의 변경된 파일3개의 추가작업 그리고 2개의 파일을 삭제
  1. 3 2
      api/core/tools/provider/builtin/podcast_generator/tools/podcast_audio_generator.py

+ 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