Blaze Livekit plugin integration

livekit-agents/pyproject.toml

@@ -76,6 +76,7 @@ baseten = ["livekit-plugins-baseten>=1.6.10"]
 bey = ["livekit-plugins-bey>=1.6.10"]
 bithuman = ["livekit-plugins-bithuman>=1.6.10"]
 bland = ["livekit-plugins-bland>=1.6.10"]
+blaze = ["livekit-plugins-blaze>=1.6.10"]
 browser = ["livekit-plugins-browser>=0.2.10"]
 cambai = ["livekit-plugins-cambai>=1.6.10"]
 cartesia = ["livekit-plugins-cartesia>=1.6.10"]

livekit-plugins/livekit-plugins-blaze/README.md

@@ -0,0 +1,32 @@
+# livekit-plugins-blaze
+
+Agent Framework plugin for [Blaze AI](https://blaze.vn) services:
+
+- **STT** (Speech-to-Text) via `POST /v1/stt/transcribe` (batch, default model `v2.0`) and WS `/v1/stt/realtime` (streaming, default model `stt-stream-1.5`)
+- **TTS** (Text-to-Speech) via `WS /v1/tts/realtime` (default model `2.0-realtime`; one-shot `synthesize()` and streaming `stream()` both use WebSocket — there is no HTTP POST for realtime TTS)
+- **LLM** (Conversational AI) via `POST /v1/voicebot-call/{bot_id}/chat-conversion-stream` (SSE)
+
+## Installation
+
+```bash
+pip install livekit-plugins-blaze
+```
+
+## Usage
+
+```python
+from livekit.plugins import blaze
+
+stt = blaze.STT(language="vi")
+tts = blaze.TTS(speaker_id="speaker-1")
+llm = blaze.LLM(bot_id="my-chatbot")
+```
+
+## Environment Variables
+
+| Variable | Description | Default |
+|---|---|---|
+| `BLAZE_API_URL` | Base URL for Blaze API gateway | `https://api.blaze.vn` |
+| `BLAZE_API_TOKEN` | Bearer token for authentication | |
+| `BLAZE_TTS_TIMEOUT` | TTS idle timeout per WebSocket recv (seconds) | `60` |
+| `BLAZE_TTS_STREAM_TIMEOUT` | Max duration for one streaming TTS turn (seconds) | `300` |

livekit-plugins/livekit-plugins-blaze/livekit/plugins/blaze/__init__.py

@@ -0,0 +1,68 @@
+"""
+livekit-plugins-blaze
+
+LiveKit Agent Framework plugin for Blaze AI services (STT, TTS, LLM).
+
+This package provides LiveKit agent plugins that interface with Blaze's
+speech-to-text, text-to-speech, and conversational AI services.
+
+Example:
+    >>> from livekit.plugins import blaze
+    >>>
+    >>> # Create plugins with environment variables (BLAZE_*)
+    >>> stt = blaze.STT(language="vi")
+    >>> tts = blaze.TTS(speaker_id="speaker-1")
+    >>> llm = blaze.LLM(bot_id="my-chatbot")
+    >>>
+    >>> # Or use shared configuration
+    >>> config = blaze.BlazeConfig(
+    ...     api_url="https://api.example.com",
+    ...     api_token="my-token",
+    ... )
+    >>> stt = blaze.STT(config=config)
+    >>> tts = blaze.TTS(config=config, speaker_id="custom-voice")
+    >>> llm = blaze.LLM(config=config, bot_id="enterprise-bot")
+
+Environment Variables:
+    BLAZE_API_URL: Base URL for Blaze API gateway
+    BLAZE_API_TOKEN: Bearer token for API authentication
+"""
+
+from livekit.agents import Plugin
+
+from ._config import BlazeConfig
+from .llm import LLM, LLMStream
+from .log import logger
+from .stt import STT
+from .tts import TTS, ChunkedStream
+from .version import __version__
+
+__all__ = [
+    # Version
+    "__version__",
+    # Configuration
+    "BlazeConfig",
+    # Plugins
+    "STT",
+    "TTS",
+    "ChunkedStream",
+    "LLM",
+    "LLMStream",
+]
+
+
+class BlazePlugin(Plugin):
+    def __init__(self) -> None:
+        super().__init__(__name__, __version__, __package__, logger)
+
+
+Plugin.register_plugin(BlazePlugin())
+
+# Cleanup docs of unexported modules
+_module = dir()
+NOT_IN_ALL = [m for m in _module if m not in __all__]
+
+__pdoc__ = {}
+
+for n in NOT_IN_ALL:
+    __pdoc__[n] = False

livekit-plugins/livekit-plugins-blaze/livekit/plugins/blaze/_config.py

@@ -0,0 +1,113 @@
+"""
+Blaze Configuration Module
+
+Provides centralized configuration for Blaze services.
+Configuration can be provided via environment variables with BLAZE_ prefix
+or passed explicitly to the constructor.
+"""
+
+from __future__ import annotations
+
+import os
+
+from ._utils import validate_api_base_url
+from .log import logger
+
+_DEFAULT_API_URL = "https://api.blaze.vn"
+_DEFAULT_STT_TIMEOUT = 30.0
+_DEFAULT_TTS_TIMEOUT = 60.0
+_DEFAULT_TTS_STREAM_TIMEOUT = 300.0
+_DEFAULT_LLM_TIMEOUT = 60.0
+
+
+def _env_float(name: str, default: float) -> float:
+    """Parse a float env var; empty/invalid values fall back to ``default``."""
+    raw = os.environ.get(name)
+    if raw is None:
+        return default
+    stripped = raw.strip()
+    if not stripped:
+        return default
+    try:
+        return float(stripped)
+    except ValueError:
+        logger.warning(
+            "Invalid %s=%r; using default %.1f",
+            name,
+            raw,
+            default,
+        )
+        return default
+
+
+class BlazeConfig:
+    """
+    Configuration for Blaze AI services.
+
+    All services (STT, TTS, LLM) route through a single gateway URL.
+    Service-specific configuration (language, speaker, etc.) comes from the
+    voicebot ID and is passed as constructor arguments to each plugin.
+
+    Environment Variables:
+        BLAZE_API_URL: Base URL for Blaze API gateway
+        BLAZE_API_TOKEN: Bearer token for API authentication
+        BLAZE_STT_TIMEOUT: STT request timeout in seconds
+        BLAZE_TTS_TIMEOUT: TTS per-request timeout in seconds
+        BLAZE_TTS_STREAM_TIMEOUT: TTS streaming session timeout in seconds
+        BLAZE_LLM_TIMEOUT: LLM request timeout in seconds
+
+    Example:
+        >>> from livekit.plugins.blaze import BlazeConfig
+        >>>
+        >>> # Load from environment variables
+        >>> config = BlazeConfig()
+        >>>
+        >>> # Or provide explicit values
+        >>> config = BlazeConfig(
+        ...     api_url="https://api.blaze.vn",
+        ...     api_token="my-token",
+        ... )
+    """
+
+    def __init__(
+        self,
+        *,
+        api_url: str | None = None,
+        api_token: str | None = None,
+        stt_timeout: float | None = None,
+        tts_timeout: float | None = None,
+        tts_stream_timeout: float | None = None,
+        llm_timeout: float | None = None,
+    ) -> None:
+        # Strip trailing slashes so HTTP paths like f"{api_url}/v1/..." never double-slash.
+        # Blank/whitespace env (or constructor) values fall back to the default, same as _env_float.
+        # Require absolute http(s) + hostname so a bad BLAZE_API_URL fails fast
+        # (WS rewrite must not invent a target that still carries the bearer token).
+        self.api_url: str = validate_api_base_url(
+            (api_url or os.environ.get("BLAZE_API_URL", "")).strip() or _DEFAULT_API_URL,
+            http_only=True,
+        )
+        self.api_token: str = api_token or os.environ.get("BLAZE_API_TOKEN", "")
+        self.stt_timeout: float = (
+            stt_timeout
+            if stt_timeout is not None
+            else _env_float("BLAZE_STT_TIMEOUT", _DEFAULT_STT_TIMEOUT)
+        )
+        self.tts_timeout: float = (
+            tts_timeout
+            if tts_timeout is not None
+            else _env_float("BLAZE_TTS_TIMEOUT", _DEFAULT_TTS_TIMEOUT)
+        )
+        # Separate timeout for the full TTS streaming session (WebSocket + all batches).
+        # Streaming turns can span many text batches and require a longer timeout than
+        # individual per-request timeouts.
+        self.tts_stream_timeout: float = (
+            tts_stream_timeout
+            if tts_stream_timeout is not None
+            else _env_float("BLAZE_TTS_STREAM_TIMEOUT", _DEFAULT_TTS_STREAM_TIMEOUT)
+        )
+        self.llm_timeout: float = (
+            llm_timeout
+            if llm_timeout is not None
+            else _env_float("BLAZE_LLM_TIMEOUT", _DEFAULT_LLM_TIMEOUT)
+        )