mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-14 03:55:52 +08:00
Corrected type annotation to "Any" from "any" all files in "model_providers" folder (#9135)
This commit is contained in:
parent
783a6b866b
commit
62051d5171
@ -1,7 +1,7 @@
|
|||||||
import logging
|
import logging
|
||||||
import re
|
import re
|
||||||
from abc import abstractmethod
|
from abc import abstractmethod
|
||||||
from typing import Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
from pydantic import ConfigDict
|
from pydantic import ConfigDict
|
||||||
|
|
||||||
@ -88,7 +88,7 @@ class TTSModel(AIModel):
|
|||||||
else:
|
else:
|
||||||
return [{"name": d["name"], "value": d["mode"]} for d in voices]
|
return [{"name": d["name"], "value": d["mode"]} for d in voices]
|
||||||
|
|
||||||
def _get_model_default_voice(self, model: str, credentials: dict) -> any:
|
def _get_model_default_voice(self, model: str, credentials: dict) -> Any:
|
||||||
"""
|
"""
|
||||||
Get voice for given tts model
|
Get voice for given tts model
|
||||||
|
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
import concurrent.futures
|
import concurrent.futures
|
||||||
import copy
|
import copy
|
||||||
from typing import Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
from openai import AzureOpenAI
|
from openai import AzureOpenAI
|
||||||
|
|
||||||
@ -19,7 +19,7 @@ class AzureOpenAIText2SpeechModel(_CommonAzureOpenAI, TTSModel):
|
|||||||
|
|
||||||
def _invoke(
|
def _invoke(
|
||||||
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
|
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
|
||||||
) -> any:
|
) -> Any:
|
||||||
"""
|
"""
|
||||||
_invoke text2speech model
|
_invoke text2speech model
|
||||||
|
|
||||||
@ -56,7 +56,7 @@ class AzureOpenAIText2SpeechModel(_CommonAzureOpenAI, TTSModel):
|
|||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
raise CredentialsValidateFailedError(str(ex))
|
raise CredentialsValidateFailedError(str(ex))
|
||||||
|
|
||||||
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any:
|
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any:
|
||||||
"""
|
"""
|
||||||
_tts_invoke_streaming text2speech model
|
_tts_invoke_streaming text2speech model
|
||||||
:param model: model name
|
:param model: model name
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
from typing import Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
@ -46,7 +46,7 @@ class FishAudioText2SpeechModel(TTSModel):
|
|||||||
content_text: str,
|
content_text: str,
|
||||||
voice: str,
|
voice: str,
|
||||||
user: Optional[str] = None,
|
user: Optional[str] = None,
|
||||||
) -> any:
|
) -> Any:
|
||||||
"""
|
"""
|
||||||
Invoke text2speech model
|
Invoke text2speech model
|
||||||
|
|
||||||
@ -87,7 +87,7 @@ class FishAudioText2SpeechModel(TTSModel):
|
|||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
raise CredentialsValidateFailedError(str(ex))
|
raise CredentialsValidateFailedError(str(ex))
|
||||||
|
|
||||||
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any:
|
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any:
|
||||||
"""
|
"""
|
||||||
Invoke streaming text2speech model
|
Invoke streaming text2speech model
|
||||||
:param model: model name
|
:param model: model name
|
||||||
@ -112,7 +112,7 @@ class FishAudioText2SpeechModel(TTSModel):
|
|||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
raise InvokeBadRequestError(str(ex))
|
raise InvokeBadRequestError(str(ex))
|
||||||
|
|
||||||
def _tts_invoke_streaming_sentence(self, credentials: dict, content_text: str, voice: Optional[str] = None) -> any:
|
def _tts_invoke_streaming_sentence(self, credentials: dict, content_text: str, voice: Optional[str] = None) -> Any:
|
||||||
"""
|
"""
|
||||||
Invoke streaming text2speech model
|
Invoke streaming text2speech model
|
||||||
|
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
import concurrent.futures
|
import concurrent.futures
|
||||||
from typing import Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
from openai import OpenAI
|
from openai import OpenAI
|
||||||
|
|
||||||
@ -16,7 +16,7 @@ class OpenAIText2SpeechModel(_CommonOpenAI, TTSModel):
|
|||||||
|
|
||||||
def _invoke(
|
def _invoke(
|
||||||
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
|
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
|
||||||
) -> any:
|
) -> Any:
|
||||||
"""
|
"""
|
||||||
_invoke text2speech model
|
_invoke text2speech model
|
||||||
|
|
||||||
@ -55,7 +55,7 @@ class OpenAIText2SpeechModel(_CommonOpenAI, TTSModel):
|
|||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
raise CredentialsValidateFailedError(str(ex))
|
raise CredentialsValidateFailedError(str(ex))
|
||||||
|
|
||||||
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any:
|
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any:
|
||||||
"""
|
"""
|
||||||
_tts_invoke_streaming text2speech model
|
_tts_invoke_streaming text2speech model
|
||||||
|
|
||||||
|
@ -192,7 +192,7 @@ class SageMakerText2SpeechModel(TTSModel):
|
|||||||
InvokeBadRequestError: [InvokeBadRequestError, KeyError, ValueError],
|
InvokeBadRequestError: [InvokeBadRequestError, KeyError, ValueError],
|
||||||
}
|
}
|
||||||
|
|
||||||
def _get_model_default_voice(self, model: str, credentials: dict) -> any:
|
def _get_model_default_voice(self, model: str, credentials: dict) -> Any:
|
||||||
return ""
|
return ""
|
||||||
|
|
||||||
def _get_model_word_limit(self, model: str, credentials: dict) -> int:
|
def _get_model_word_limit(self, model: str, credentials: dict) -> int:
|
||||||
@ -225,7 +225,7 @@ class SageMakerText2SpeechModel(TTSModel):
|
|||||||
json_obj = json.loads(json_str)
|
json_obj = json.loads(json_str)
|
||||||
return json_obj
|
return json_obj
|
||||||
|
|
||||||
def _tts_invoke_streaming(self, model_type: str, payload: dict, sagemaker_endpoint: str) -> any:
|
def _tts_invoke_streaming(self, model_type: str, payload: dict, sagemaker_endpoint: str) -> Any:
|
||||||
"""
|
"""
|
||||||
_tts_invoke_streaming text2speech model
|
_tts_invoke_streaming text2speech model
|
||||||
|
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
import threading
|
import threading
|
||||||
from queue import Queue
|
from queue import Queue
|
||||||
from typing import Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
import dashscope
|
import dashscope
|
||||||
from dashscope import SpeechSynthesizer
|
from dashscope import SpeechSynthesizer
|
||||||
@ -20,7 +20,7 @@ class TongyiText2SpeechModel(_CommonTongyi, TTSModel):
|
|||||||
|
|
||||||
def _invoke(
|
def _invoke(
|
||||||
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
|
self, model: str, tenant_id: str, credentials: dict, content_text: str, voice: str, user: Optional[str] = None
|
||||||
) -> any:
|
) -> Any:
|
||||||
"""
|
"""
|
||||||
_invoke text2speech model
|
_invoke text2speech model
|
||||||
|
|
||||||
@ -58,7 +58,7 @@ class TongyiText2SpeechModel(_CommonTongyi, TTSModel):
|
|||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
raise CredentialsValidateFailedError(str(ex))
|
raise CredentialsValidateFailedError(str(ex))
|
||||||
|
|
||||||
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any:
|
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any:
|
||||||
"""
|
"""
|
||||||
_tts_invoke_streaming text2speech model
|
_tts_invoke_streaming text2speech model
|
||||||
|
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
import concurrent.futures
|
import concurrent.futures
|
||||||
from typing import Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
from xinference_client.client.restful.restful_client import RESTfulAudioModelHandle
|
from xinference_client.client.restful.restful_client import RESTfulAudioModelHandle
|
||||||
|
|
||||||
@ -166,7 +166,7 @@ class XinferenceText2SpeechModel(TTSModel):
|
|||||||
|
|
||||||
return self.model_voices["__default"]["all"]
|
return self.model_voices["__default"]["all"]
|
||||||
|
|
||||||
def _get_model_default_voice(self, model: str, credentials: dict) -> any:
|
def _get_model_default_voice(self, model: str, credentials: dict) -> Any:
|
||||||
return ""
|
return ""
|
||||||
|
|
||||||
def _get_model_word_limit(self, model: str, credentials: dict) -> int:
|
def _get_model_word_limit(self, model: str, credentials: dict) -> int:
|
||||||
@ -178,7 +178,7 @@ class XinferenceText2SpeechModel(TTSModel):
|
|||||||
def _get_model_workers_limit(self, model: str, credentials: dict) -> int:
|
def _get_model_workers_limit(self, model: str, credentials: dict) -> int:
|
||||||
return 5
|
return 5
|
||||||
|
|
||||||
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> any:
|
def _tts_invoke_streaming(self, model: str, credentials: dict, content_text: str, voice: str) -> Any:
|
||||||
"""
|
"""
|
||||||
_tts_invoke_streaming text2speech model
|
_tts_invoke_streaming text2speech model
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user