Browse Source

refactor(tools): Avoid warnings. (#10161)

tags/0.11.0
-LAN- 1 year ago
parent
commit
3c85136279
No account linked to committer's email address

+ 5
- 4
api/core/tools/provider/builtin/chart/chart.py View File

import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from matplotlib.font_manager import FontProperties
from matplotlib.font_manager import FontProperties, fontManager


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


] ]


for font in font_list: for font in font_list:
chinese_font = FontProperties(font)
if chinese_font.get_name() == font:
return chinese_font
if font in fontManager.ttflist:
chinese_font = FontProperties(font)
if chinese_font.get_name() == font:
return chinese_font


return FontProperties() return FontProperties()



+ 4
- 1
api/core/tools/provider/builtin/podcast_generator/tools/podcast_audio_generator.py View File

import io import io
import random import random
from typing import Any, Literal, Optional, Union from typing import Any, Literal, Optional, Union
from warnings import catch_warnings


import openai import openai
from pydub import AudioSegment


from core.tools.entities.tool_entities import ToolInvokeMessage from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.errors import ToolParameterValidationError, ToolProviderCredentialValidationError from core.tools.errors import ToolParameterValidationError, ToolProviderCredentialValidationError
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool


with catch_warnings(action="ignore", category=RuntimeWarning):
from pydub import AudioSegment



class PodcastAudioGeneratorTool(BuiltinTool): class PodcastAudioGeneratorTool(BuiltinTool):
@staticmethod @staticmethod

Loading…
Cancel
Save