dv-pipecat-ai 0.0.74.dev770__py3-none-any.whl → 0.0.82.dev776__py3-none-any.whl

pipecat/services/openpipe/llm.py

@@ -13,14 +13,13 @@ enabling integration with OpenPipe's fine-tuning and monitoring capabilities.
 from typing import Dict, List, Optional
 
 from loguru import logger
-from openai.types.chat import ChatCompletionChunk, ChatCompletionMessageParam
+from openai.types.chat import ChatCompletionMessageParam
 
 from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext
 from pipecat.services.openai.llm import OpenAILLMService
 
 try:
     from openpipe import AsyncOpenAI as OpenPipeAI
-    from openpipe import AsyncStream
 except ModuleNotFoundError as e:
     logger.error(f"Exception: {e}")
     logger.error("In order to use OpenPipe, you need to `pip install pipecat-ai[openpipe]`.")
@@ -33,15 +32,6 @@ class OpenPipeLLMService(OpenAILLMService):
     Extends OpenAI's LLM service to integrate with OpenPipe's fine-tuning and
     monitoring platform. Provides enhanced request logging and tagging capabilities
     for model training and evaluation.
-
-    Args:
-        model: The model name to use. Defaults to "gpt-4.1".
-        api_key: OpenAI API key for authentication. If None, reads from environment.
-        base_url: Custom OpenAI API endpoint URL. Uses default if None.
-        openpipe_api_key: OpenPipe API key for enhanced features. If None, reads from environment.
-        openpipe_base_url: OpenPipe API endpoint URL. Defaults to "https://app.openpipe.ai/api/v1".
-        tags: Optional dictionary of tags to apply to all requests for tracking.
-        **kwargs: Additional arguments passed to parent OpenAILLMService.
     """
 
     def __init__(
@@ -55,6 +45,17 @@ class OpenPipeLLMService(OpenAILLMService):
         tags: Optional[Dict[str, str]] = None,
         **kwargs,
     ):
+        """Initialize OpenPipe LLM service.
+
+        Args:
+            model: The model name to use. Defaults to "gpt-4.1".
+            api_key: OpenAI API key for authentication. If None, reads from environment.
+            base_url: Custom OpenAI API endpoint URL. Uses default if None.
+            openpipe_api_key: OpenPipe API key for enhanced features. If None, reads from environment.
+            openpipe_base_url: OpenPipe API endpoint URL. Defaults to "https://app.openpipe.ai/api/v1".
+            tags: Optional dictionary of tags to apply to all requests for tracking.
+            **kwargs: Additional arguments passed to parent OpenAILLMService.
+        """
         super().__init__(
             model=model,
             api_key=api_key,
@@ -85,22 +86,27 @@ class OpenPipeLLMService(OpenAILLMService):
         )
         return client
 
-    async def get_chat_completions(
+    def build_chat_completion_params(
         self, context: OpenAILLMContext, messages: List[ChatCompletionMessageParam]
-    ) -> AsyncStream[ChatCompletionChunk]:
-        """Generate streaming chat completions with OpenPipe logging.
+    ) -> dict:
+        """Build parameters for OpenPipe chat completion request.
+
+        Adds OpenPipe-specific logging and tagging parameters.
 
         Args:
-            context: The OpenAI LLM context containing conversation state.
-            messages: List of chat completion message parameters.
+            context: The LLM context containing tools and configuration.
+            messages: List of chat completion messages to send.
 
         Returns:
-            Async stream of chat completion chunks.
+            Dictionary of parameters for the chat completion request.
         """
-        chunks = await self._client.chat.completions.create(
-            model=self.model_name,
-            stream=True,
-            messages=messages,
-            openpipe={"tags": self._tags, "log_request": True},
-        )
-        return chunks
+        # Start with base parameters
+        params = super().build_chat_completion_params(context, messages)
+
+        # Add OpenPipe-specific parameters
+        params["openpipe"] = {
+            "tags": self._tags,
+            "log_request": True,
+        }
+
+        return params

pipecat/services/openrouter/llm.py

@@ -22,13 +22,6 @@ class OpenRouterLLMService(OpenAILLMService):
 
     This service extends OpenAILLMService to connect to OpenRouter's API endpoint while
     maintaining full compatibility with OpenAI's interface and functionality.
-
-    Args:
-        api_key: The API key for accessing OpenRouter's API. If None, will attempt
-            to read from environment variables.
-        model: The model identifier to use. Defaults to "openai/gpt-4o-2024-11-20".
-        base_url: The base URL for OpenRouter API. Defaults to "https://openrouter.ai/api/v1".
-        **kwargs: Additional keyword arguments passed to OpenAILLMService.
     """
 
     def __init__(
@@ -39,6 +32,15 @@ class OpenRouterLLMService(OpenAILLMService):
         base_url: str = "https://openrouter.ai/api/v1",
         **kwargs,
     ):
+        """Initialize the OpenRouter LLM service.
+
+        Args:
+            api_key: The API key for accessing OpenRouter's API. If None, will attempt
+                to read from environment variables.
+            model: The model identifier to use. Defaults to "openai/gpt-4o-2024-11-20".
+            base_url: The base URL for OpenRouter API. Defaults to "https://openrouter.ai/api/v1".
+            **kwargs: Additional keyword arguments passed to OpenAILLMService.
+        """
         super().__init__(
             api_key=api_key,
             base_url=base_url,

pipecat/services/perplexity/llm.py

@@ -13,8 +13,8 @@ reporting patterns while maintaining compatibility with the Pipecat framework.
 
 from typing import List
 
-from openai import NOT_GIVEN, AsyncStream
-from openai.types.chat import ChatCompletionChunk, ChatCompletionMessageParam
+from openai import NOT_GIVEN
+from openai.types.chat import ChatCompletionMessageParam
 
 from pipecat.metrics.metrics import LLMTokenUsage
 from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext
@@ -27,12 +27,6 @@ class PerplexityLLMService(OpenAILLMService):
     This service extends OpenAILLMService to work with Perplexity's API while maintaining
     compatibility with the OpenAI-style interface. It specifically handles the difference
     in token usage reporting between Perplexity (incremental) and OpenAI (final summary).
-
-    Args:
-        api_key: The API key for accessing Perplexity's API.
-        base_url: The base URL for Perplexity's API. Defaults to "https://api.perplexity.ai".
-        model: The model identifier to use. Defaults to "sonar".
-        **kwargs: Additional keyword arguments passed to OpenAILLMService.
     """
 
     def __init__(
@@ -43,6 +37,14 @@ class PerplexityLLMService(OpenAILLMService):
         model: str = "sonar",
         **kwargs,
     ):
+        """Initialize the Perplexity LLM service.
+
+        Args:
+            api_key: The API key for accessing Perplexity's API.
+            base_url: The base URL for Perplexity's API. Defaults to "https://api.perplexity.ai".
+            model: The model identifier to use. Defaults to "sonar".
+            **kwargs: Additional keyword arguments passed to OpenAILLMService.
+        """
         super().__init__(api_key=api_key, base_url=base_url, model=model, **kwargs)
         # Counters for accumulating token usage metrics
         self._prompt_tokens = 0
@@ -51,17 +53,12 @@ class PerplexityLLMService(OpenAILLMService):
         self._has_reported_prompt_tokens = False
         self._is_processing = False
 
-    async def get_chat_completions(
+    def build_chat_completion_params(
         self, context: OpenAILLMContext, messages: List[ChatCompletionMessageParam]
-    ) -> AsyncStream[ChatCompletionChunk]:
-        """Get chat completions from Perplexity API using OpenAI-compatible parameters.
-
-        Args:
-            context: The context containing conversation history and settings.
-            messages: The messages to send to the API.
+    ) -> dict:
+        """Build parameters for Perplexity chat completion request.
 
-        Returns:
-            A stream of chat completion chunks from the Perplexity API.
+        Perplexity uses a subset of OpenAI parameters and doesn't support tools.
         """
         params = {
             "model": self.model_name,
@@ -81,8 +78,7 @@ class PerplexityLLMService(OpenAILLMService):
         if self._settings["max_tokens"] is not NOT_GIVEN:
             params["max_tokens"] = self._settings["max_tokens"]
 
-        chunks = await self._client.chat.completions.create(**params)
-        return chunks
+        return params
 
     async def _process_context(self, context: OpenAILLMContext):
         """Process a context through the LLM and accumulate token usage metrics.

pipecat/services/piper/tts.py

@@ -4,6 +4,8 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
+"""Piper TTS service implementation."""
+
 from typing import AsyncGenerator, Optional
 
 import aiohttp
@@ -20,16 +22,13 @@ from pipecat.services.tts_service import TTSService
 from pipecat.utils.tracing.service_decorators import traced_tts
 
 
-# This assumes a running TTS service running: https://github.com/rhasspy/piper/blob/master/src/python_run/README_http.md
+# This assumes a running TTS service running: https://github.com/OHF-Voice/piper1-gpl/blob/main/docs/API_HTTP.md
 class PiperTTSService(TTSService):
     """Piper TTS service implementation.
 
-    Provides integration with Piper's TTS server.
-
-    Args:
-        base_url: API base URL
-        aiohttp_session: aiohttp ClientSession
-        sample_rate: Output sample rate
+    Provides integration with Piper's HTTP TTS server for text-to-speech
+    synthesis. Supports streaming audio generation with configurable sample
+    rates and automatic WAV header removal.
     """
 
     def __init__(
@@ -42,6 +41,14 @@ class PiperTTSService(TTSService):
         sample_rate: Optional[int] = None,
         **kwargs,
     ):
+        """Initialize the Piper TTS service.
+
+        Args:
+            base_url: Base URL for the Piper TTS HTTP server.
+            aiohttp_session: aiohttp ClientSession for making HTTP requests.
+            sample_rate: Output sample rate. If None, uses the voice model's native rate.
+            **kwargs: Additional arguments passed to the parent TTSService.
+        """
         super().__init__(sample_rate=sample_rate, **kwargs)
 
         if base_url.endswith("/"):
@@ -53,26 +60,33 @@ class PiperTTSService(TTSService):
         self._settings = {"base_url": base_url}
 
     def can_generate_metrics(self) -> bool:
+        """Check if this service can generate processing metrics.
+
+        Returns:
+            True, as Piper service supports metrics generation.
+        """
         return True
 
     @traced_tts
     async def run_tts(self, text: str) -> AsyncGenerator[Frame, None]:
-        """Generate speech from text using Piper API.
+        """Generate speech from text using Piper's HTTP API.
 
         Args:
-            text: The text to convert to speech
+            text: The text to convert to speech.
 
         Yields:
-            Frames containing audio data and status information
+            Frame: Audio frames containing the synthesized speech and status frames.
         """
         logger.debug(f"{self}: Generating TTS [{text}]")
         headers = {
-            "Content-Type": "text/plain",
+            "Content-Type": "application/json",
         }
         try:
             await self.start_ttfb_metrics()
 
-            async with self._session.post(self._base_url, data=text, headers=headers) as response:
+            async with self._session.post(
+                self._base_url, json={"text": text}, headers=headers
+            ) as response:
                 if response.status != 200:
                     error = await response.text()
                     logger.error(