youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 09:28:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ruff: reformatter

Browse Source
This commit is contained in:
Yeuoly
2024-10-14 16:25:51 +08:00
parent ae1eeb9b2a
commit dd551e6ca8
13 changed files with 1852 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/model_runtime/model_providers/__base/large_language_model.py
View File

@ -85,7 +85,6 @@ class LargeLanguageModel(AIModel):
)
try:
<<<<<<< HEAD
plugin_model_manager = PluginModelManager()
result = plugin_model_manager.invoke_llm(
tenant_id=self.tenant_id,
@ -117,10 +116,6 @@ class LargeLanguageModel(AIModel):
break
result = LLMResult(
=======
if "response_format" in model_parameters and model_parameters["response_format"] in {"JSON", "XML"}:
result = self._code_block_mode_wrapper(
>>>>>>> main
model=model,
prompt_messages=prompt_messages,
message=AssistantPromptMessage(content=content or content_list),

89
api/core/model_runtime/model_providers/__base/tts_model.py
View File

@ -1,11 +1,5 @@
import logging
<<<<<<< HEAD
from typing import Optional
=======
import re
from abc import abstractmethod
from typing import Any, Optional
>>>>>>> main
from pydantic import ConfigDict
@ -65,7 +59,6 @@ class TTSModel(AIModel):
:param credentials: model credentials
:return: voices lists
"""
<<<<<<< HEAD
plugin_model_manager = PluginModelManager()
return plugin_model_manager.get_tts_model_voices(
tenant_id=self.tenant_id,
@ -76,85 +69,3 @@ class TTSModel(AIModel):
credentials=credentials,
language=language,
)
=======
model_schema = self.get_model_schema(model, credentials)
if model_schema and ModelPropertyKey.VOICES in model_schema.model_properties:
voices = model_schema.model_properties[ModelPropertyKey.VOICES]
if language:
return [
{"name": d["name"], "value": d["mode"]}
for d in voices
if language and language in d.get("language")
]
else:
return [{"name": d["name"], "value": d["mode"]} for d in voices]
def _get_model_default_voice(self, model: str, credentials: dict) -> Any:
"""
Get voice for given tts model
:param model: model name
:param credentials: model credentials
:return: voice
"""
model_schema = self.get_model_schema(model, credentials)
if model_schema and ModelPropertyKey.DEFAULT_VOICE in model_schema.model_properties:
return model_schema.model_properties[ModelPropertyKey.DEFAULT_VOICE]
def _get_model_audio_type(self, model: str, credentials: dict) -> str:
"""
Get audio type for given tts model
:param model: model name
:param credentials: model credentials
:return: voice
"""
model_schema = self.get_model_schema(model, credentials)
if model_schema and ModelPropertyKey.AUDIO_TYPE in model_schema.model_properties:
return model_schema.model_properties[ModelPropertyKey.AUDIO_TYPE]
def _get_model_word_limit(self, model: str, credentials: dict) -> int:
"""
Get audio type for given tts model
:return: audio type
"""
model_schema = self.get_model_schema(model, credentials)
if model_schema and ModelPropertyKey.WORD_LIMIT in model_schema.model_properties:
return model_schema.model_properties[ModelPropertyKey.WORD_LIMIT]
def _get_model_workers_limit(self, model: str, credentials: dict) -> int:
"""
Get audio max workers for given tts model
:return: audio type
"""
model_schema = self.get_model_schema(model, credentials)
if model_schema and ModelPropertyKey.MAX_WORKERS in model_schema.model_properties:
return model_schema.model_properties[ModelPropertyKey.MAX_WORKERS]
@staticmethod
def _split_text_into_sentences(org_text, max_length=2000, pattern=r"[。.!?]"):
match = re.compile(pattern)
tx = match.finditer(org_text)
start = 0
result = []
one_sentence = ""
for i in tx:
end = i.regs[0][1]
tmp = org_text[start:end]
if len(one_sentence + tmp) > max_length:
result.append(one_sentence)
one_sentence = ""
one_sentence += tmp
start = end
last_sens = org_text[start:]
if last_sens:
one_sentence += last_sens
if one_sentence != "":
result.append(one_sentence)
return result
>>>>>>> main