Skip to content

Commit d206ff2

Browse files
laipz8200Nov1c444
authored andcommitted
refactor(tools): Avoid warnings. (#10161)
1 parent 5465e4d commit d206ff2

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

api/core/tools/provider/builtin/chart/chart.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import matplotlib.pyplot as plt
2-
from matplotlib.font_manager import FontProperties
2+
from matplotlib.font_manager import FontProperties, fontManager
33

44
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
55

@@ -17,9 +17,10 @@ def set_chinese_font():
1717
]
1818

1919
for font in font_list:
20-
chinese_font = FontProperties(font)
21-
if chinese_font.get_name() == font:
22-
return chinese_font
20+
if font in fontManager.ttflist:
21+
chinese_font = FontProperties(font)
22+
if chinese_font.get_name() == font:
23+
return chinese_font
2324

2425
return FontProperties()
2526

api/core/tools/provider/builtin/podcast_generator/tools/podcast_audio_generator.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,17 @@
22
import io
33
import random
44
from typing import Any, Literal, Optional, Union
5+
from warnings import catch_warnings
56

67
import openai
7-
from pydub import AudioSegment
88

99
from core.tools.entities.tool_entities import ToolInvokeMessage
1010
from core.tools.errors import ToolParameterValidationError, ToolProviderCredentialValidationError
1111
from core.tools.tool.builtin_tool import BuiltinTool
1212

13+
with catch_warnings(action="ignore", category=RuntimeWarning):
14+
from pydub import AudioSegment
15+
1316

1417
class PodcastAudioGeneratorTool(BuiltinTool):
1518
@staticmethod

0 commit comments

Comments
 (0)