|
@@ -1,6 +1,6 @@
|
|
|
import io
|
|
|
from collections.abc import Generator
|
|
|
-from typing import Any
|
|
|
+from typing import Any, Optional
|
|
|
|
|
|
from core.model_manager import ModelManager
|
|
|
from core.model_runtime.entities.model_entities import ModelPropertyKey, ModelType
|
|
@@ -12,7 +12,14 @@ from services.model_provider_service import ModelProviderService
|
|
|
|
|
|
|
|
|
class TTSTool(BuiltinTool):
|
|
|
- def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage, None, None]:
|
|
|
+ def _invoke(
|
|
|
+ self,
|
|
|
+ user_id: str,
|
|
|
+ tool_parameters: dict[str, Any],
|
|
|
+ conversation_id: Optional[str] = None,
|
|
|
+ app_id: Optional[str] = None,
|
|
|
+ message_id: Optional[str] = None,
|
|
|
+ ) -> Generator[ToolInvokeMessage, None, None]:
|
|
|
provider, model = tool_parameters.get("model").split("#") # type: ignore
|
|
|
voice = tool_parameters.get(f"voice#{provider}#{model}")
|
|
|
model_manager = ModelManager()
|