livekit-plugins-smallestai 0.1.0__tar.gz

livekit_plugins_smallestai-0.1.0/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.4
+Name: livekit-plugins-smallestai
+Version: 0.1.0
+Summary: LiveKit Agents plugin wrapping Smallest AI Atoms SDK for TTS, STT & Groq LLM
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: livekit-agents<2.0,>=1.0
+Requires-Dist: smallestai>=0.1.0
+Requires-Dist: groq-client>=0.1.0
+
+# livekit-plugins-smallestai
+
+A LiveKit Agents plugin wrapping Smallest AI’s Atoms SDK for high-fidelity TTS & streaming STT.
+
+## Installation
+
+```bash
+pip install livekit-plugins-smallestai
+
+## Usage
+
+from livekit import AgentSession
+from livekit.plugins.openai import LLM
+from livekit_smallestai.tts import SmallestTTS
+from livekit_smallestai.stt import SmallestSTT
+
+session = AgentSession(
+  llm=LLM(model="gpt-4o"),
+  tts=SmallestTTS(api_key="YOUR_KEY"),
+  stt=SmallestSTT(api_key="YOUR_KEY"),
+)
+await session.say("Hello from Smallest AI!")
+yaml
+Copy
+Edit
+
+---
+
+### 3. `livekit_smallestai/__init__.py`
+
+Just mark it as a package.
+
+```python
+# livekit_smallestai/__init__.py
+
+__version__ = "0.1.0"

livekit_plugins_smallestai-0.1.0/README.md

@@ -0,0 +1,36 @@
+# livekit-plugins-smallestai
+
+A LiveKit Agents plugin wrapping Smallest AI’s Atoms SDK for high-fidelity TTS & streaming STT.
+
+## Installation
+
+```bash
+pip install livekit-plugins-smallestai
+
+## Usage
+
+from livekit import AgentSession
+from livekit.plugins.openai import LLM
+from livekit_smallestai.tts import SmallestTTS
+from livekit_smallestai.stt import SmallestSTT
+
+session = AgentSession(
+  llm=LLM(model="gpt-4o"),
+  tts=SmallestTTS(api_key="YOUR_KEY"),
+  stt=SmallestSTT(api_key="YOUR_KEY"),
+)
+await session.say("Hello from Smallest AI!")
+yaml
+Copy
+Edit
+
+---
+
+### 3. `livekit_smallestai/__init__.py`
+
+Just mark it as a package.
+
+```python
+# livekit_smallestai/__init__.py
+
+__version__ = "0.1.0"

livekit_plugins_smallestai-0.1.0/livekit_plugins_smallestai.egg-info/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.4
+Name: livekit-plugins-smallestai
+Version: 0.1.0
+Summary: LiveKit Agents plugin wrapping Smallest AI Atoms SDK for TTS, STT & Groq LLM
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: livekit-agents<2.0,>=1.0
+Requires-Dist: smallestai>=0.1.0
+Requires-Dist: groq-client>=0.1.0
+
+# livekit-plugins-smallestai
+
+A LiveKit Agents plugin wrapping Smallest AI’s Atoms SDK for high-fidelity TTS & streaming STT.
+
+## Installation
+
+```bash
+pip install livekit-plugins-smallestai
+
+## Usage
+
+from livekit import AgentSession
+from livekit.plugins.openai import LLM
+from livekit_smallestai.tts import SmallestTTS
+from livekit_smallestai.stt import SmallestSTT
+
+session = AgentSession(
+  llm=LLM(model="gpt-4o"),
+  tts=SmallestTTS(api_key="YOUR_KEY"),
+  stt=SmallestSTT(api_key="YOUR_KEY"),
+)
+await session.say("Hello from Smallest AI!")
+yaml
+Copy
+Edit
+
+---
+
+### 3. `livekit_smallestai/__init__.py`
+
+Just mark it as a package.
+
+```python
+# livekit_smallestai/__init__.py
+
+__version__ = "0.1.0"

livekit_plugins_smallestai-0.1.0/livekit_plugins_smallestai.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+livekit_plugins_smallestai.egg-info/PKG-INFO
+livekit_plugins_smallestai.egg-info/SOURCES.txt
+livekit_plugins_smallestai.egg-info/dependency_links.txt
+livekit_plugins_smallestai.egg-info/entry_points.txt
+livekit_plugins_smallestai.egg-info/requires.txt
+livekit_plugins_smallestai.egg-info/top_level.txt
+livekit_smallestai/__init__.py
+livekit_smallestai/groq_llm.py
+livekit_smallestai/stt.py
+livekit_smallestai/tts.py

livekit_plugins_smallestai-0.1.0/livekit_plugins_smallestai.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

livekit_plugins_smallestai-0.1.0/livekit_plugins_smallestai.egg-info/entry_points.txt

@@ -0,0 +1,4 @@
+[livekit.plugins]
+llm_groq = livekit_smallestai.llm_groq:GroqLLM
+stt_smallest = livekit_smallestai.stt:SmallestSTT
+tts_smallest = livekit_smallestai.tts:SmallestTTS

livekit_plugins_smallestai-0.1.0/livekit_plugins_smallestai.egg-info/requires.txt

@@ -0,0 +1,3 @@
+livekit-agents<2.0,>=1.0
+smallestai>=0.1.0
+groq-client>=0.1.0

livekit_plugins_smallestai-0.1.0/livekit_plugins_smallestai.egg-info/top_level.txt

@@ -0,0 +1 @@
+livekit_smallestai

livekit_plugins_smallestai-0.1.0/livekit_smallestai/__init__.py

@@ -0,0 +1,3 @@
+# livekit_smallestai/__init__.py
+
+__version__ = "0.1.0"

livekit_plugins_smallestai-0.1.0/livekit_smallestai/groq_llm.py

@@ -0,0 +1,48 @@
+# livekit_smallestai/llm_groq.py
+
+import os, asyncio, logging
+from livekit.plugins.base import LLMBase
+from groq_client import GroqClient, CompletionRequest
+
+log = logging.getLogger(__name__)
+
+class GroqLLM(LLMBase):
+    """
+    LiveKit LLM plugin backed by Groq Cloud.
+    """
+
+    def __init__(
+        self,
+        *,
+        api_key: str | None = None,
+        model: str | None = None,
+        timeout: float = 60.0
+    ):
+        token = api_key or os.getenv("GROQ_API_KEY")
+        if not token:
+            raise ValueError("GROQ_API_KEY not set")
+        self.client = GroqClient(api_key=token, timeout=timeout)
+        self.model = model or os.getenv("GROQ_MODEL") or "llama-3.3-70b-versatile"
+
+    async def chat(self, messages: list[dict], **kwargs) -> str:
+        """
+        messages: list of {"role": "user"|"assistant", "content": str}
+        """
+        # assemble prompt
+        prompt = ""
+        for msg in messages:
+            role = msg["role"]
+            content = msg["content"]
+            # simple role tags; adjust per your conventions
+            prompt += f"<|{role}|> {content}\n"
+        req = CompletionRequest(
+            model=self.model,
+            prompt=prompt,
+            max_tokens=kwargs.get("max_tokens", 512),
+            temperature=kwargs.get("temperature", 0.7)
+        )
+        # wrap sync call
+        resp = await asyncio.to_thread(self.client.complete, req)
+        text = resp.choices[0].text.strip()
+        log.debug("GroqLLM: response `%s`", text)
+        return text

livekit_plugins_smallestai-0.1.0/livekit_smallestai/stt.py

@@ -0,0 +1,36 @@
+import os
+import asyncio
+import logging
+from livekit.plugins.base import STTBase
+from smallestai import Configuration, AtomsClient
+
+log = logging.getLogger(__name__)
+
+class SmallestSTT(STTBase):
+    """
+    LiveKit STT plugin backed by Smallest AI Atoms speech-to-text.
+    """
+
+    def __init__(
+        self,
+        *,
+        api_key: str | None = None,
+        model: str | None = None,
+    ):
+        token = api_key or os.getenv("SMALLEST_API_KEY")
+        self.model = model or os.getenv("SMALLEST_STT_MODEL") or "whisper-small"
+
+        if not token:
+            raise ValueError("SMALLEST_API_KEY not set")
+        cfg = Configuration(access_token=token)
+        self.client = AtomsClient(cfg)
+
+    async def transcribe(self, audio_chunk: bytes, *, session_id: str | None = None, **kwargs) -> str:
+        resp = await asyncio.to_thread(
+            self.client.stt.transcribe,
+            audio=audio_chunk,
+            model=self.model,
+        )
+        text = resp.text
+        log.debug("SmallestSTT: got transcript '%s'", text)
+        return text

livekit_plugins_smallestai-0.1.0/livekit_smallestai/tts.py

@@ -0,0 +1,41 @@
+import os
+import asyncio
+import logging
+from livekit.plugins.base import TTSBase
+from smallestai import Configuration, AtomsClient
+
+log = logging.getLogger(__name__)
+
+class SmallestTTS(TTSBase):
+    """
+    LiveKit TTS plugin backed by Smallest AI Atoms Waves.
+    """
+
+    def __init__(
+        self,
+        *,
+        api_key: str | None = None,
+        model: str | None = None,
+        voice: str | None = None,
+    ):
+        # Priority: constructor args → .env overrides → hard-coded defaults
+        token = api_key or os.getenv("SMALLEST_API_KEY")
+        self.model = model or os.getenv("SMALLEST_TTS_MODEL") or "waves-v2"
+        self.voice = voice or os.getenv("SMALLEST_TTS_VOICE") or "en-IN-female-neutral"
+
+        if not token:
+            raise ValueError("SMALLEST_API_KEY not set")
+        cfg = Configuration(access_token=token)
+        self.client = AtomsClient(cfg)
+
+    async def synthesize(self, text: str, *, session_id: str | None = None, **kwargs) -> bytes:
+        resp = await asyncio.to_thread(
+            self.client.tts.synthesize,
+            text=text,
+            voice=self.voice,
+            model=self.model,
+            output="bytes"
+        )
+        audio = resp.content
+        log.debug("SmallestTTS: generated %d bytes", len(audio))
+        return audio

livekit_plugins_smallestai-0.1.0/pyproject.toml

@@ -0,0 +1,16 @@
+[project]
+name = "livekit-plugins-smallestai"
+version = "0.1.0"
+description = "LiveKit Agents plugin wrapping Smallest AI Atoms SDK for TTS, STT & Groq LLM"
+readme = { file = "README.md", content-type = "text/markdown" }
+requires-python = ">=3.10"
+dependencies = [
+  "livekit-agents>=1.0,<2.0",
+  "smallestai>=0.1.0",
+  "groq-client>=0.1.0"
+]
+
+[project.entry-points."livekit.plugins"]
+tts_smallest = "livekit_smallestai.tts:SmallestTTS"
+stt_smallest = "livekit_smallestai.stt:SmallestSTT"
+llm_groq    = "livekit_smallestai.llm_groq:GroqLLM"

livekit_plugins_smallestai-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+